class TestPlane(unittest.TestCase): def setUp(self): self.test_plane_5_10 = Plane(x_size=5, y_size=10) def test_boundaries(self): self.assert_(self.test_plane_5_10.get_node(Location(4, 9)) is plane.CONST_EMPTY) self.assertRaises(IndexError, self.test_plane_5_10.get_node, **{'location':Location(5, 9)}) self.assertRaises(IndexError, self.test_plane_5_10.get_node, **{'location':Location(1, 10)}) self.assert_(self.test_plane_5_10.get_node(Location(1, 9)) is plane.CONST_EMPTY) self.assert_(self.test_plane_5_10.get_node(Location(0, 0)) is plane.CONST_EMPTY) def test_node_insertion_normal(self): self.test_plane_5_10.insert_node(Node(node_id=1), Location(0, 0)) self.assert_(self.test_plane_5_10.get_node(Location(0, 0))) self.assertFalse(self.test_plane_5_10.get_node(Location(0, 1))) self.test_plane_5_10.insert_node(Node(node_id=1), Location(0, 1)) self.assert_(self.test_plane_5_10.get_node(Location(0, 1))) def test_node_insertion_already_ocupied(self): self.test_plane_5_10.insert_node(Node(node_id=1), Location(0, 0)) self.assert_(self.test_plane_5_10.get_node(Location(0, 0))) self.assertRaises(LocationOcuppied, self.test_plane_5_10.insert_node, Node(2), Location(0, 0)) def test_remove_node_existing(self): """ Tests the removal of an existing node """ node = Node(node_id=1) self.test_plane_5_10.insert_node(node, Location(0, 0)) self.assert_(self.test_plane_5_10.get_node(Location(0, 0))) self.test_plane_5_10.remove_node(node) self.assertFalse(self.test_plane_5_10.get_node(Location(0, 0))) def test_plane_insertion(self): """ Tests the insertion of ids in plane """ my_plane = Plane(x_size=10, y_size=10) my_plane.insert_node(Node(node_id=1), Location(3, 4)) my_plane.insert_node(Node(node_id=2), Location(3, 5)) my_plane.insert_node(Node(node_id=3), Location(1, 2)) self.assert_("1" in my_plane.all_node_ids) self.assert_("2" in my_plane.all_node_ids) self.assert_("3" in my_plane.all_node_ids)