fix test

chemosim-lab · Jul 1, 2024 · cc30735 · cc30735
1 parent 9172025
commit cc30735
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_ifp.py b/tests/test_ifp.py
@@ -37,7 +37,7 @@ def test_interaction_data_iteration(ifp: IFP) -> None:
     data = next(ifp.interactions())
     assert isinstance(data, InteractionData)
     assert data.ligand == ResidueId("LIG", 1, "G")
-    assert data.protein.chain == "A"
+    assert data.protein.chain in {"A", "B"}
     assert data.interaction in {"Hydrophobic", "VdWContact"}
     assert "distance" in data.metadata
     for data in ifp.interactions():