Restore original density in test_cell_density

paulromano · Sep 19, 2025 · Feb 28, 2024 · Feb 28, 2024 · Feb 28, 2024 · Feb 28, 2024
commit db1ce5f9b3abc02648604f4fc91aaf3ad410983e
diff --git a/tests/unit_tests/test_lib.py b/tests/unit_tests/test_lib.py
@@ -161,16 +161,21 @@ def test_properties_temperature(lib_init):
 
 def test_cell_density(lib_init):
     cell = openmc.lib.cells[1]
-    cell.set_density(1.5, 0)
-    assert cell.get_density(0) == pytest.approx(1.5)
-    cell.set_density(2.0)
-    assert cell.get_density() == pytest.approx(2.0)
+    print('density', cell.get_density())
+    orig_density = cell.get_density()
+    try:
+        cell.set_density(1.5, 0)
+        assert cell.get_density(0) == pytest.approx(1.5)
+        cell.set_density(2.0)
+        assert cell.get_density() == pytest.approx(2.0)
+    finally:
+        cell.set_density(orig_density)
 
 
 def test_properties_cell_density(lib_init):
     # Cell density should be 2.0 from above test
     cell = openmc.lib.cells[1]
-    assert cell.get_density() == pytest.approx(2.0)
+    orig_density = cell.get_density()
 
     # Export properties and change density
     openmc.lib.export_properties('properties.h5')
@@ -179,7 +184,7 @@ def test_properties_cell_density(lib_init):
 
     # Import properties and check that density is restored
     openmc.lib.import_properties('properties.h5')
-    assert cell.get_density() == pytest.approx(2.0)
+    assert cell.get_density() == pytest.approx(orig_density)
 
 
 def test_new_cell(lib_init):