diff --git a/propka/protonate.py b/propka/protonate.py index fe96f3a..412d9a3 100644 --- a/propka/protonate.py +++ b/propka/protonate.py @@ -31,7 +31,7 @@ def __init__(self, verbose=False): 'H': 1, 'He': 2, 'Li': 1, 'Be': 2, 'B': 3, 'C': 4, 'N': 5, 'O': 6, 'F': 7, 'Ne': 8, 'Na': 1, 'Mg': 2, 'Al': 3, 'Si': 4, 'P': 5, 'S': 6, 'Cl': 7, 'Ar': 8, 'K': 1, 'Ca': 2, 'Sc': 2, - 'Ti': 2, 'Va': 2, 'Cr': 1, 'Mn': 2, 'Fe': 2, 'Co': 2, 'Ni': 2, + 'Ti': 2, 'V': 2, 'Cr': 1, 'Mn': 2, 'Fe': 2, 'Co': 2, 'Ni': 2, 'Cu': 1, 'Zn': 2, 'Ga': 3, 'Ge': 4, 'As': 5, 'Se': 6, 'Br': 7, 'Kr': 8, 'I': 7} # TODO - consider putting charges in a configuration file diff --git a/tests/test_protonate.py b/tests/test_protonate.py new file mode 100644 index 0000000..3d06a2a --- /dev/null +++ b/tests/test_protonate.py @@ -0,0 +1,13 @@ +import propka.atom +import propka.protonate + + +def test_protonate_atom(): + atom = propka.atom.Atom( + "HETATM 4479 V VO4 A1578 -19.097 16.967 0.500 1.00 17.21 V " + ) + assert not atom.is_protonated + p = propka.protonate.Protonate() + p.protonate_atom(atom) + assert atom.is_protonated + assert atom.number_of_protons_to_add == 6