diff --git a/powersimdata/data_access/tests/test_execute_csv.py b/powersimdata/data_access/tests/test_execute_csv.py index 618ad71ee..9f8809116 100644 --- a/powersimdata/data_access/tests/test_execute_csv.py +++ b/powersimdata/data_access/tests/test_execute_csv.py @@ -37,5 +37,6 @@ def test_get_execute_file_from_server_type(execute_table): @pytest.mark.integration @pytest.mark.ssh def test_get_execute_file_from_server_header(execute_table): - header = ["id", "status"] + header = ["status"] assert_array_equal(execute_table.columns, header) + assert "id" == execute_table.index.name diff --git a/powersimdata/data_access/tests/test_scenario_csv.py b/powersimdata/data_access/tests/test_scenario_csv.py index b0363f524..18f0b6ae9 100644 --- a/powersimdata/data_access/tests/test_scenario_csv.py +++ b/powersimdata/data_access/tests/test_scenario_csv.py @@ -30,7 +30,6 @@ def test_get_scenario_file_from_server_type(data_access, scenario_table): @pytest.mark.ssh def test_get_scenario_file_from_server_header(data_access, scenario_table): header = [ - "id", "plan", "name", "state", @@ -48,6 +47,7 @@ def test_get_scenario_file_from_server_header(data_access, scenario_table): "infeasibilities", ] assert_array_equal(scenario_table.columns, header) + assert "id" == scenario_table.index.name @pytest.mark.integration