fix valuesview not support index

cnerg · Mar 6, 2019 · 44a31f1 · 44a31f1
1 parent 7dc6709
commit 44a31f1
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/test_mcnp.py b/tests/test_mcnp.py
@@ -551,8 +551,9 @@ def test_read_mcnp():
         list(read_materials[2]._mats.keys())[0].comp.keys())
     for i in range(2):
         assert_almost_equal(
-            list(expected_multimaterial._mats.keys())[0].comp.values()[i],
-            list(read_materials[2]._mats.keys())[0].comp.values()[i])
+            list(list(expected_multimaterial._mats.keys())
+                 [0].comp.values())[i],
+            list(list(read_materials[2]._mats.keys())[0].comp.values())[i])
     assert_almost_equal(
         list(expected_multimaterial._mats.keys())[0].mass,
         list(read_materials[2]._mats.keys())[0].mass)
@@ -570,8 +571,9 @@ def test_read_mcnp():
         list(read_materials[2]._mats.keys())[1].comp.keys())
     for i in range(2):
         assert_almost_equal(
-            list(expected_multimaterial._mats.keys())[1].comp.values()[i],
-            list(read_materials[2]._mats.keys())[1].comp.values()[i])
+            list(list(expected_multimaterial._mats.keys())
+                 [1].comp.values())[i],
+            list(list(read_materials[2]._mats.keys())[1].comp.values())[i])
     assert_equal(
         list(expected_multimaterial._mats.keys())[1].mass,
         list(read_materials[2]._mats.keys())[1].mass)