for i in range(4):
        self.assertEqual(len(cell.state_size[i]), 2)