Fix failing dataframe tests (no more row id column)

rerun-io · Oct 7, 2024 · 73902c7 · 73902c7
1 parent c230971
commit 73902c7
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/rerun_py/tests/unit/test_dataframe.py b/rerun_py/tests/unit/test_dataframe.py
@@ -26,8 +26,8 @@ def test_full_view(self) -> None:
         batches = view.select()
         table = pa.Table.from_batches(batches, batches.schema)
 
-        # row, log_time, log_tick, indicator, points, colors
-        assert table.num_columns == 6
+        # log_time, log_tick, indicator, points, colors
+        assert table.num_columns == 5
         assert table.num_rows == 2
 
     def test_select_column(self) -> None:
@@ -72,8 +72,8 @@ def test_view_syntax(self) -> None:
             batches = view.select()
             table = pa.Table.from_batches(batches, batches.schema)
 
-            # row, log_time, log_tick, points
-            assert table.num_columns == 4
+            # log_time, log_tick, points
+            assert table.num_columns == 3
             assert table.num_rows == 2
 
         bad_content_expressions = [
@@ -85,7 +85,7 @@ def test_view_syntax(self) -> None:
             view = self.recording.view(index="log_time", contents=expr)
             batches = view.select()
 
-            # row, log_time, log_tick
+            # log_time, log_tick
             table = pa.Table.from_batches(batches, batches.schema)
-            assert table.num_columns == 3
+            assert table.num_columns == 2
             assert table.num_rows == 0