Updated test to make more sense.

idaholab · Nov 30, 2024 · 2c1071a · 2c1071a
1 parent ac7e86c
commit 2c1071a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_integration.py b/tests/test_integration.py
@@ -223,8 +223,9 @@ def test_cell_material_setter(simple_problem):
 
 def test_problem_cells_setter(simple_problem):
     problem = copy.deepcopy(simple_problem)
-    cells = copy.deepcopy(simple_problem.cells)
-    cells.remove(cells[1])
+    #TODO test cells clone
+    cells = simple_problem.cells.clone()
+    cells.remove(cells[4])
     with pytest.raises(TypeError):
         problem.cells = 5
     with pytest.raises(TypeError):
@@ -234,7 +235,7 @@ def test_problem_cells_setter(simple_problem):
     problem.cells = cells
     assert problem.cells.objects == cells.objects
     problem.cells = list(cells)
-    assert problem.cells[2] == cells[2]
+    assert problem.cells[6] == cells[6]
     # test that cell modifiers are still there
     problem.cells._importance.format_for_mcnp_input((6, 2, 0))