Skip to content

Commit

Permalink
fix: Updating test to match new fields.
Browse files Browse the repository at this point in the history
  • Loading branch information
pesap committed Sep 25, 2024
1 parent 9c35e68 commit 763162b
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions tests/test_cost_curves.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,13 +15,13 @@ def test_cost_curve():
value_curve=InputOutputCurve(
function_data=LinearFunctionData(proportional_term=1.0, constant_term=2.0)
),
vom_units=InputOutputCurve(
vom_cost=InputOutputCurve(
function_data=LinearFunctionData(proportional_term=2.0, constant_term=1.0)
),
)

assert cost_curve.value_curve.function_data.proportional_term == 1.0
assert cost_curve.vom_units.function_data.proportional_term == 2.0
assert cost_curve.vom_cost.function_data.proportional_term == 2.0


def test_fuel_curve():
Expand All @@ -30,14 +30,14 @@ def test_fuel_curve():
value_curve=InputOutputCurve(
function_data=LinearFunctionData(proportional_term=1.0, constant_term=2.0)
),
vom_units=InputOutputCurve(
vom_cost=InputOutputCurve(
function_data=LinearFunctionData(proportional_term=2.0, constant_term=1.0)
),
fuel_cost=2.5,
)

assert fuel_curve.value_curve.function_data.proportional_term == 1.0
assert fuel_curve.vom_units.function_data.proportional_term == 2.0
assert fuel_curve.vom_cost.function_data.proportional_term == 2.0
assert fuel_curve.fuel_cost == 2.5


Expand All @@ -48,7 +48,7 @@ def test_value_curve_custom_serialization():
value_curve=InputOutputCurve(
function_data=LinearFunctionData(proportional_term=1.0, constant_term=2.0)
),
vom_units=InputOutputCurve(
vom_cost=InputOutputCurve(
function_data=LinearFunctionData(proportional_term=2.0, constant_term=1.0)
),
),
Expand All @@ -73,7 +73,7 @@ def test_value_curve_serialization(tmp_path):
value_curve=InputOutputCurve(
function_data=LinearFunctionData(proportional_term=1.0, constant_term=2.0)
),
vom_units=InputOutputCurve(
vom_cost=InputOutputCurve(
function_data=LinearFunctionData(proportional_term=2.0, constant_term=1.0)
),
),
Expand Down

0 comments on commit 763162b

Please sign in to comment.