def test_symmetry(): s1 = Skeleton("s1") s1.add_nodes(["1", "2", "3", "4", "5", "6"]) s1.add_edge("1", "2") s1.add_edge("3", "4") s1.add_edge("5", "6") s1.add_symmetry("1", "5") s1.add_symmetry("3", "6") assert s1.get_symmetry("1").name == "5" assert s1.get_symmetry("5").name == "1" assert s1.get_symmetry("3").name == "6" # Cannot add more than one symmetry to a node with pytest.raises(ValueError): s1.add_symmetry("1", "6") with pytest.raises(ValueError): s1.add_symmetry("6", "1") s1.delete_symmetry("1", "5") assert s1.get_symmetry("1") is None with pytest.raises(ValueError): s1.delete_symmetry("1", "5")
def test_symmetry(): s1 = Skeleton("s1") s1.add_nodes(["1", "2", "3", "4", "5", "6"]) s1.add_edge("1", "2") s1.add_edge("3", "4") s1.add_edge("5", "6") s1.add_symmetry("1", "5") s1.add_symmetry("3", "6") assert (s1.nodes[0], s1.nodes[4]) in s1.symmetries assert (s1.nodes[2], s1.nodes[5]) in s1.symmetries assert len(s1.symmetries) == 2 assert (0, 4) in s1.symmetric_inds assert (2, 5) in s1.symmetric_inds assert len(s1.symmetric_inds) == 2 assert s1.get_symmetry("1").name == "5" assert s1.get_symmetry("5").name == "1" assert s1.get_symmetry("3").name == "6" # Cannot add more than one symmetry to a node with pytest.raises(ValueError): s1.add_symmetry("1", "6") with pytest.raises(ValueError): s1.add_symmetry("6", "1") s1.delete_symmetry("1", "5") assert s1.get_symmetry("1") is None with pytest.raises(ValueError): s1.delete_symmetry("1", "5")