def test_example(self):
grid = [
[1, 0, 0, 1, 1],
[0, 0, 0, 0, 1],
[1, 1, 1, 1, 1]]
self.assertEqual(9, get_num_lights_on(grid))
def test_example_full(self):
grid = [[1 for _ in xrange(1000)] for _ in xrange(1000)]
self.assertEqual(1000000, get_num_lights_on(grid))
def test_all_zeroes(self):
grid = get_1000_by_1000_grid()
self.assertEqual(0, get_num_lights_on(grid))