Skip to content

Commit

Permalink
blacken
Browse files Browse the repository at this point in the history
  • Loading branch information
dhensle committed Feb 8, 2024
1 parent 2fc1c9d commit bb7f79a
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions activitysim/abm/models/util/test/test_vehicle_type_alts.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
from activitysim.abm.models.vehicle_type_choice import (
get_combinatorial_vehicle_alternatives,
construct_model_alternatives,
VehicleTypeChoiceSettings
VehicleTypeChoiceSettings,
)
from activitysim.core import workflow

Expand Down Expand Up @@ -37,7 +37,7 @@ def test_vehicle_type_alts():
data={
"body_type": ["Car", "Car", "Car", "SUV", "SUV"],
"fuel_type": ["Gas", "Gas", "BEV", "Gas", "BEV"],
"age": ['1', '2', '3', '1', '2'],
"age": ["1", "2", "3", "1", "2"],
"dummy_data": [1, 2, 3, 4, 5],
},
index=[0, 1, 2, 3, 4],
Expand All @@ -54,4 +54,6 @@ def test_vehicle_type_alts():
pdt.assert_index_equal(alts_long.index, alts_wide.index)

# columns need to be in correct order for downstream configs
pdt.assert_index_equal(alts_long.columns, pd.Index(["body_type", "age", "fuel_type"]))
pdt.assert_index_equal(
alts_long.columns, pd.Index(["body_type", "age", "fuel_type"])
)

0 comments on commit bb7f79a

Please sign in to comment.