def test1(self): input = [[0,0,0,5],[4,3,1,4],[0,1,1,4],[1,2,1,3],[0,0,1,1]] answer = [[0,0,0,0],[0,0,0,4],[0,0,0,0],[0,0,0,3],[0,0,0,0]] s = Solution() s.setZeroes(input) self.assertEqual(input, answer)