def test_correct_position_post_collision(self):
from data.Mallet import Mallet
from data.Disc import Disc
mock = Mock()
mock.playerColor = 1
Mallet.load_image = Mock()
Disc.load_image = Mock()
m = Mallet(10, 35, 35, 1, mock, [(0, 100), (0, 100)])
d = Disc(37, 37, 1, 10, [(0, 100), (0, 100)])
m.correct_position_post_collision(d)
distance_vector = m.pos - d.pos
self.assertTrue(distance_vector.length >= m.radius + d.radius)
m = Mallet(10, 35, 35, 1, mock, [(0, 100), (0, 100)])
d = Disc(37, 37, 1, 10, [(0, 100), (0, 100)])
d.correct_position_post_collision(m)
distance_vector = m.pos - d.pos
self.assertTrue(distance_vector.length >= m.radius + d.radius)
