Update test_core.py to reflect change to default oxidation states

WMD-group · Dec 2, 2024 · 59bf75c · 59bf75c
1 parent 7d7459d
commit 59bf75c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/smact/tests/test_core.py b/smact/tests/test_core.py
@@ -47,7 +47,8 @@ def test_element_dictionary(self):
         newlist = ["O", "Rb", "W"]
         dictionary = smact.element_dictionary(newlist, TEST_OX_STATES)
         self.assertEqual(dictionary["O"].crustal_abundance, 461000.0)
-        self.assertEqual(dictionary["Rb"].oxidation_states, [-1, 1])
+        self.assertEqual(dictionary["Rb"].oxidation_states_smact14, [-1, 1])
+        self.assertEqual(dictionary["Rb"].oxidation_states, [1])
         self.assertEqual(dictionary["Rb"].oxidation_states_custom, [-1, 1])
         self.assertEqual(dictionary["W"].name, "Tungsten")
         self.assertTrue("Rn" in smact.element_dictionary())