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)