def test_disconnect_nodes(self): n1 = Node((1, 1)) n2 = Node((2, 2)) n1.connect_to(n2, 'E') n2.connect_to(n1, 'W') n1.disconnect_label('E') for container in n1.siblings, n2.siblings, n1.labels, n2.labels: self.assertEqual(len(container), 0)
def test_connect_nodes(self): n1 = Node((1, 1)) n2 = Node((2, 2)) n1.connect_to(n2, 'E') self.assertIn('E', n1.labels) self.assertEqual(n1.labels['E'], n2) self.assertIn(n2, n1.siblings) self.assertEqual(n1.siblings[n2], 'E') self.assertEqual(len(n2.siblings), 0) self.assertEqual(len(n2.labels), 0) n2.connect_to(n1, 'W')