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')
