Fixing tests

pyiron · Sep 13, 2020 · 9cc760f · 9cc760f
1 parent 1ea385b
commit 9cc760f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/vasp/test_outcar.py b/tests/vasp/test_outcar.py
@@ -553,7 +553,7 @@ def test_get_dipole_moments(self):
             output = self.outcar_parser.get_dipole_moments(filename)
             self.assertTrue(output)
             self.assertIsInstance(output, list)
-            if len(output) > 1:
+            if len(output[0]) > 1:
                 self.assertIsInstance(output[-1], np.ndarray)
                 self.assertIsInstance(output[-1][-1], np.ndarray)
                 self.assertEqual(len(output[-1][-1]), 3)
@@ -690,7 +690,7 @@ def test_get_kpoints_irreducible_reciprocal(self):
                 self.assertEqual(output_all.__str__(), output.__str__())
 
     def test_get_nelect(self):
-        n_elect_list = [40.0, 16.0, 16.0, 16.0, 16.0, 16.0, 224.0, 358.0]
+        n_elect_list = [40.0, 16.0, 16.0, 16.0, 16.0, 16.0, 224.0, 358.0, 8]
         for filename in self.file_list:
             i = int(filename.split("_")[-1]) - 1
             self.assertEqual(n_elect_list[i], self.outcar_parser.get_nelect(filename))