def test_equals(self): self.assertEqual(square(2), 4)
def test_equals_2(self): self.assertEqual(square(50), 2500)
def test_equals_3(self): self.assertEqual(square(1), 1)