self.assertEqual(test_zero[i][0].get_shape()[1], cell.state_size[i][0])