Add unit test for protonate.py

2022-03-02 15:45:11 -05:00
parent f49eaedefb
commit b920f5aeec
1 changed files with 13 additions and 0 deletions
--- a/tests/test_protonate.py
+++ 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