def test_equals_2(self): self.assertEqual(problem(1), 56)
def test_equals(self): self.assertEqual(problem('hello'), 'Error')