def test_set_location(self):
n1 = NodeData(0, (1, 2, 3))
assert (1, 2, 3) == n1.getLocation()
n1.setLocation((4, 2, 4))
assert (4, 2, 4) == n1.getLocation()
n1.setLocation(None)
assert (4, 2, 4) == n1.getLocation()
def test_get_location(self):
n1 = NodeData(0, (1, 2, 3))
n2 = NodeData(1)
assert (1, 2, 3) == n1.getLocation()
assert None == n2.getLocation()