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)]