class TestShape(unittest.TestCase):
def setUp(self):
self.s1 = Shape(None, '', 1, (5, 5), (0, 0, 0))
self.s2 = Shape(None, '', 2, (60, 80), (255, 0, 255))
def testAdd(self):
s3 = self.s1.add(self.s2)
self.assertEqual(s3.scale, 3)
self.assertEqual(s3.color, (0, 0, 0))
def testClone(self):
s3 = self.s1.clone()
s4 = self.s1
self.assertTrue(self.s1 == s4)
self.assertFalse(self.s1 == s3)
def testEquals(self):
s3 = Shape(None, '', 1, (5, 5), (0, 0, 0))
self.assertTrue(self.s1.equals(s3))
def action(s1: Shape, s2: Shape):
return [s1.add(s2)]