def test_crawl(self):
self.atomC1.add_connected(self.atomC2, 2)
result = molecule_splitter._crawl(
self.atomC1,
("node", ("ext-atom", ("atom", "C"), [("hydrogen", 2)]),
("next", 2, ("node", ("ext-atom",
("atom", "C"), [("hydrogen", 2)]), None))),
[])
expected = [self.atomC1, self.atomC2]
expected_alt = [self.atomC2, self.atomC1]
self.assertTrue(result == expected or result == expected_alt)
def test_crawl_2(self):
self.atomC1.add_connected(self.atomO, 2)
self.atomC1.add_connected(self.atomN)
result = molecule_splitter._crawl(
self.atomC1,
("node", ("ext-atom", ("atom", "C"), [("next", 2,
("node",
("atom", "O"), None))]),
("next", 1,
("node", ("ext-atom",
("atom", "N"), [("hydrogen", 2)]), None)), None), [])
expected = list(
itertools.permutations([self.atomC1, self.atomN, self.atomO]))
self.assertIn(tuple(result), expected)
def test_crawl(self):
self.atomC1.add_connected(self.atomC2, 2)
result = molecule_splitter._crawl(
self.atomC1,
(
"node",
("ext-atom", ("atom", "C"), [("hydrogen", 2)]),
("next", 2, ("node", ("ext-atom", ("atom", "C"), [("hydrogen", 2)]), None)),
),
[],
)
expected = [self.atomC1, self.atomC2]
expected_alt = [self.atomC2, self.atomC1]
self.assertTrue(result == expected or result == expected_alt)
def test_crawl_2(self):
self.atomC1.add_connected(self.atomO, 2)
self.atomC1.add_connected(self.atomN)
result = molecule_splitter._crawl(
self.atomC1,
(
"node",
("ext-atom", ("atom", "C"), [("next", 2, ("node", ("atom", "O"), None))]),
("next", 1, ("node", ("ext-atom", ("atom", "N"), [("hydrogen", 2)]), None)),
None,
),
[],
)
expected = list(itertools.permutations([self.atomC1, self.atomN, self.atomO]))
self.assertIn(tuple(result), expected)
