def test_set_position(self): geo = Geometry(0, 0, 100, 200) self.assertEqual(geo.x, 0) self.assertEqual(geo.y, 0) # set postion geo.set_position(10, 10) self.assertEqual(geo.x, 10) self.assertEqual(geo.y, 10) # set position relative to gravity geo.set_position(10, 10, Gravity(0, 0)) self.assertEqual(geo.x, 10) self.assertEqual(geo.y, 10) geo.set_position(110, 210, Gravity(1, 1)) self.assertEqual(geo.x, 10) self.assertEqual(geo.y, 10) geo.set_position(60, 110, Gravity(0.5, 0.5)) self.assertEqual(geo.x, 10) self.assertEqual(geo.y, 10)