diff --git a/rerun_py/tests/unit/test_dataframe.py b/rerun_py/tests/unit/test_dataframe.py index a2e3e29ac13e..b4961debe61f 100644 --- a/rerun_py/tests/unit/test_dataframe.py +++ b/rerun_py/tests/unit/test_dataframe.py @@ -104,36 +104,14 @@ def test_schema_recording(self) -> None: # Color, Points3DIndicator, Position3D, Radius, Text, TextIndicator assert len(schema.component_columns()) == 6 - # TODO(cmc): revert me - # assert schema.index_columns()[0].name == "log_tick" - # assert schema.index_columns()[1].name == "log_time" - # assert schema.index_columns()[2].name == "my_index" - # assert schema.component_columns()[0].entity_path == "/points" - # assert schema.component_columns()[0].component_name == "rerun.components.Points3DIndicator" - # assert schema.component_columns()[0].is_static is False - # assert schema.component_columns()[1].entity_path == "/points" - # assert schema.component_columns()[1].component_name == "rerun.components.Color" - # assert schema.component_columns()[1].is_static is False - # assert schema.component_columns()[2].entity_path == "/points" - # assert schema.component_columns()[2].component_name == "rerun.components.Position3D" - # assert schema.component_columns()[2].is_static is False - # assert schema.component_columns()[3].entity_path == "/points" - # assert schema.component_columns()[3].component_name == "rerun.components.Radius" - # assert schema.component_columns()[3].is_static is False - # assert schema.component_columns()[4].entity_path == "/static_text" - # assert schema.component_columns()[4].component_name == "rerun.components.TextLogIndicator" - # assert schema.component_columns()[4].is_static is True - # assert schema.component_columns()[5].entity_path == "/static_text" - # assert schema.component_columns()[5].component_name == "rerun.components.Text" - # assert schema.component_columns()[5].is_static is True assert schema.index_columns()[0].name == "log_tick" assert schema.index_columns()[1].name == "log_time" assert schema.index_columns()[2].name == "my_index" assert schema.component_columns()[0].entity_path == "/points" - assert schema.component_columns()[0].component_name == "rerun.components.Color" + assert schema.component_columns()[0].component_name == "rerun.components.Points3DIndicator" assert schema.component_columns()[0].is_static is False assert schema.component_columns()[1].entity_path == "/points" - assert schema.component_columns()[1].component_name == "rerun.components.Points3DIndicator" + assert schema.component_columns()[1].component_name == "rerun.components.Color" assert schema.component_columns()[1].is_static is False assert schema.component_columns()[2].entity_path == "/points" assert schema.component_columns()[2].component_name == "rerun.components.Position3D" @@ -142,11 +120,14 @@ def test_schema_recording(self) -> None: assert schema.component_columns()[3].component_name == "rerun.components.Radius" assert schema.component_columns()[3].is_static is False assert schema.component_columns()[4].entity_path == "/static_text" - assert schema.component_columns()[4].component_name == "rerun.components.Text" + assert schema.component_columns()[4].component_name == "rerun.components.TextLogIndicator" assert schema.component_columns()[4].is_static is True assert schema.component_columns()[5].entity_path == "/static_text" - assert schema.component_columns()[5].component_name == "rerun.components.TextLogIndicator" + assert schema.component_columns()[5].component_name == "rerun.components.Text" assert schema.component_columns()[5].is_static is True + assert schema.index_columns()[0].name == "log_tick" + assert schema.index_columns()[1].name == "log_time" + assert schema.index_columns()[2].name == "my_index" def test_schema_view(self) -> None: schema = self.recording.view(index="my_index", contents="points").schema()