def test_area_calculateDistances(self): area = Area(5) area.saveArea() testArea = Area('file') self.assertEqual(testArea.numberOfBlocks, 5) self.assertEqual(testArea.cords.shape, (5, 2)) self.assertEqual(testArea.sizes.shape, (5, )) self.assertEqual(testArea.distances.shape, (5, 5))