diff --git a/tests/test_table.py b/tests/test_table.py index 40176d20a99..d20a26363d5 100644 --- a/tests/test_table.py +++ b/tests/test_table.py @@ -1137,8 +1137,6 @@ def test_table_iter(pa_table, batch_size, drop_last_batch): subtables = list(table_iter(pa_table, batch_size=batch_size, drop_last_batch=drop_last_batch)) assert len(subtables) == num_batches if drop_last_batch: - if not all(len(subtable) == batch_size for subtable in subtables): - raise ArithmeticError([len(subtable) for subtable in subtables]) assert all(len(subtable) == batch_size for subtable in subtables) else: assert all(len(subtable) == batch_size for subtable in subtables[:-1])