Tests: common.remove_numeric_parentheses()

ReactionMechanismGenerator · Oct 28, 2024 · d676a05 · d676a05
1 parent 3b6dc35
commit d676a05
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_common.py b/tests/test_common.py
@@ -171,3 +171,13 @@ def test_get_o2_stoichiometry():
     assert common.get_o2_stoichiometry(smiles='CCCCCC') == 9.5  # 6 CO2 + 7 H2O
     assert common.get_o2_stoichiometry(smiles='CCO') == 3  # 2 CO2 + 3 H2O - O
     assert common.get_o2_stoichiometry(smiles='NCC') == 7.5 / 2  # 2 CO2 + 3.5 H2O
+
+
+def test_remove_numeric_parentheses():
+    """Test the remove_numeric_parentheses() function"""
+    assert common.remove_numeric_parentheses('C2H5(2)') == 'C2H5'
+    assert common.remove_numeric_parentheses('C2H5') == 'C2H5'
+    assert common.remove_numeric_parentheses('CH2(S)') == 'CH2(S)'
+    assert common.remove_numeric_parentheses('C2H5(547)') == 'C2H5'
+    assert common.remove_numeric_parentheses('C2H5(547)H') == 'C2H5(547)H'
+