def test_two_three_four(self): self.assertEqual(square(2), 4) self.assertEqual(square(3), 9) self.assertEqual(square(4), 16)
def test_negative(self): self.assertEqual(square(-1), 1)
def test_zero(self): self.assertEqual(square(0), 0)
def test_one(self): self.assertEqual(square(1), 1)
def test_empty_list_recursive(self): self.assertEqual(square([[],[[],[]]]), [[],[[],[]]])
def test_list_recursive(self): self.assertEqual(square([1, [1, 2], 3, [1, 2, [1, 3]]] ), [1, [1, 4], 9, [1, 4, [1, 9]]] )
def test_list(self): self.assertEqual(square([2, -2, 1j]), [4, 4, -1])
def test_empty_list(self): self.assertEqual(square([]), [])
def test_complex(self): self.assertEqual(square(1 - 1j), (1 - 1j)*(1 - 1j))