def test_number_of_rows_for_3_by_3_square(self):
r = Rectangle(3, 3)
rows = r.row_list()
self.assertEqual(len(rows), 8 + 5 * 9)
def test_tetromino_row_for_3_by_3_square(self):
r = Rectangle(3, 3)
rows = r.row_list()
row1 = [1, 1, 1, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0]
self.assertIn(row1, rows)
