def test_can_get_nearby_residues_and_ligands(self, mock_atoms):
mock_atoms.return_value = self.atoms
atom = Atom("C", 4, 8, 3)
self.assertEqual(atom.nearby_residues(3, ligands=True, a=1),
set(self.residues + ["LIG"]))
mock_atoms.assert_called_with(3, a=1)
def test_can_get_nearby_residues(self, mock_atoms):
mock_atoms.return_value = self.atoms
atom = Atom("C", 4, 8, 3)
self.assertEqual(atom.nearby_residues(3, a=1), set(self.residues))
mock_atoms.assert_called_with(3, a=1)