Update cgsmiles/tests/test_write_cgsmiles.py

Co-authored-by: Peter C Kroon <[email protected]>
gruenewald-lab · Oct 8, 2024 · 116b5e9 · 116b5e9
1 parent 7370d54
commit 116b5e9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cgsmiles/tests/test_write_cgsmiles.py b/cgsmiles/tests/test_write_cgsmiles.py
@@ -62,6 +62,7 @@ def test_write_cgsmiles(input_string):
     out_mol = out_resolver.molecule
     assertEqualGraphs(molecule, out_mol)
     out_fragments = out_resolver.fragment_dicts
+    assert len(fragment_dicts) == len(out_fragments)
     for frag_dict, frag_dict_out in zip(fragment_dicts, out_fragments):
         assert set(frag_dict_out) == set(frag_dict)
         for fragname in frag_dict: