def test_average(self): expected = [2, 2] param = [[1, 1], [3, 3]] self.assertListEqual(utils2d.average(param), expected) expected = [0, 0] param = [] self.assertListEqual(utils2d.average(param), expected) self.assertListEqual(utils2d.average(None), expected)
def _get_start_room(self): key_func = lambda r: utils2d.distancesqr([0, 0], utils2d.average(r.tiles)) return sorted(self.rooms, key=key_func)[0]