fix unit test

test/unit/test_input_.py

@@ -50,7 +50,6 @@ def test_nuclide_basic():
         ),
         speed=np.array([1.0, 2.0, 3.0, 4.0]),
         decay=np.array([1.0, 2.0, 3.0]),
-        sensitivity=True,
     )
 
     assert n1.tag == "Nuclide"
@@ -80,8 +79,6 @@ def test_nuclide_basic():
     assert (n1.chi_s == np.ones([4, 4]) * 0.25).all()
     assert (n1.chi_p == np.ones([4, 4]) * 0.25).all()
     assert (n1.chi_d == np.ones([3, 4]) * 0.25).all()
-    assert n1.sensitivity == True
-    assert n1.sensitivity_ID == 1
 
 
 def test_nuclide_default():
@@ -109,8 +106,6 @@ def test_nuclide_default():
     assert (n1.chi_s == np.zeros((5, 5))).all()
     assert (n1.chi_p == np.zeros((5, 5))).all()
     assert (n1.chi_d == np.zeros((0, 5))).all()
-    assert n1.sensitivity == False
-    assert n1.sensitivity_ID == 0
 
     assert (n2.capture == np.zeros(5)).all()
     assert (n2.scatter == np.ones(5) * 5.0).all()
@@ -122,8 +117,6 @@ def test_nuclide_default():
     assert (n2.chi_s == np.ones((5, 5)) * 0.2).all()
     assert (n2.chi_p == np.zeros((5, 5))).all()
     assert (n2.chi_d == np.zeros((0, 5))).all()
-    assert n2.sensitivity == False
-    assert n2.sensitivity_ID == 0
 
     assert (n3.capture == np.zeros(5)).all()
     assert (n3.scatter == np.zeros(5)).all()
@@ -135,45 +128,6 @@ def test_nuclide_default():
     assert (n3.chi_s == np.zeros((5, 5))).all()
     assert (n3.chi_p == np.ones((5, 5)) * 0.2).all()
     assert (n3.chi_d == np.zeros((0, 5))).all()
-    assert n3.sensitivity == False
-    assert n3.sensitivity_ID == 0
-
-
-def test_nuclide_IDs_sensitivity():
-    # Start fresh
-    mcdc.reset()
-
-    # Create nuclides with various sensitivity tags
-    n1 = mcdc.nuclide(capture=np.array([1.0]), sensitivity=True)
-    n2 = mcdc.nuclide(capture=np.array([1.0]))
-    n3 = mcdc.nuclide(capture=np.array([1.0]), sensitivity=True)
-    n4 = mcdc.nuclide(capture=np.array([1.0]), sensitivity=True)
-    n5 = mcdc.nuclide(capture=np.array([1.0]))
-    n6 = mcdc.nuclide(capture=np.array([1.0]))
-    n7 = mcdc.nuclide(capture=np.array([1.0]), sensitivity=True)
-
-    # Checks
-    assert n1.ID == 0
-    assert n2.ID == 1
-    assert n3.ID == 2
-    assert n4.ID == 3
-    assert n5.ID == 4
-    assert n6.ID == 5
-    assert n7.ID == 6
-    assert n1.sensitivity
-    assert not n2.sensitivity
-    assert n3.sensitivity
-    assert n4.sensitivity
-    assert not n5.sensitivity
-    assert not n6.sensitivity
-    assert n7.sensitivity
-    assert n1.sensitivity_ID == 1
-    assert n2.sensitivity_ID == 0
-    assert n3.sensitivity_ID == 2
-    assert n4.sensitivity_ID == 3
-    assert n5.sensitivity_ID == 0
-    assert n6.sensitivity_ID == 0
-    assert n7.sensitivity_ID == 4
 
 
 # ======================================================================================
@@ -224,7 +178,6 @@ def test_material_single():
         ),
         speed=np.array([1.0, 2.0, 3.0, 4.0]),
         decay=np.array([1.0, 2.0, 3.0]),
-        sensitivity=True,
     )
 
     # Checks
@@ -256,7 +209,6 @@ def test_material_single():
     assert (m1.nu_f == np.array([4.0, 8.0, 12.0, 16.0])).all()
     assert (m1.chi_s == np.ones([4, 4]) * 0.25).all()
     assert (m1.chi_p == np.ones([4, 4]) * 0.25).all()
-    assert m1.sensitivity == True
 
     # Check if the nuclide was registered
     n2 = mcdc.nuclide(capture=np.ones(5))
@@ -269,7 +221,7 @@ def test_material_multi():
 
     # Create a multi-nuclide material
     n1 = mcdc.nuclide(capture=np.ones(5), speed=np.ones(5) * 1)
-    n2 = mcdc.nuclide(scatter=np.ones((5, 5)), speed=np.ones(5) * 2, sensitivity=True)
+    n2 = mcdc.nuclide(scatter=np.ones((5, 5)), speed=np.ones(5) * 2)
     n3 = mcdc.nuclide(
         fission=np.ones(5), nu_p=np.ones(5), chi_p=np.ones((5, 5)), speed=np.ones(5) * 3
     )
@@ -285,37 +237,6 @@ def test_material_multi():
     assert m1.J == 0
     assert (m1.total == np.ones(5) * 14).all()
     assert (m1.speed == np.ones(5) * 30 / 14).all()
-    assert m1.sensitivity == True
-
-
-def test_material_IDs_sensitivity():
-    # Start fresh
-    mcdc.reset()
-
-    # Create materials with various sensitivity tags
-    m1 = mcdc.material(capture=np.array([1.0]), sensitivity=True)
-    m2 = mcdc.material(capture=np.array([1.0]))
-    m3 = mcdc.material(capture=np.array([1.0]), sensitivity=True)
-    m4 = mcdc.material(capture=np.array([1.0]), sensitivity=True)
-    m5 = mcdc.material(capture=np.array([1.0]))
-    m6 = mcdc.material(capture=np.array([1.0]))
-    m7 = mcdc.material(capture=np.array([1.0]), sensitivity=True)
-
-    # Checks
-    assert m1.ID == 0
-    assert m2.ID == 1
-    assert m3.ID == 2
-    assert m4.ID == 3
-    assert m5.ID == 4
-    assert m6.ID == 5
-    assert m7.ID == 6
-    assert m1.sensitivity
-    assert not m2.sensitivity
-    assert m3.sensitivity
-    assert m4.sensitivity
-    assert not m5.sensitivity
-    assert not m6.sensitivity
-    assert m7.sensitivity
 
 
 # ======================================================================================