def sol() -> Solution:
return Solution()
def test_judge_square_sum(self):
inputs = (5, 3, 4, 2, 1)
outs = (True, False, True, True, True)
for inp, out in zip(inputs, outs):
self.assertEqual(out, Solution().judgeSquareSum(inp))