Skip to content

Commit

Permalink
linting
Browse files Browse the repository at this point in the history
  • Loading branch information
sabinala committed Jul 30, 2024
1 parent 52e36c7 commit 5b53a4d
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions tests/test_interfaces.py
Original file line number Diff line number Diff line change
Expand Up @@ -739,28 +739,27 @@ def test_errors_for_bad_amrs(
@pytest.mark.parametrize("end_time", END_TIMES)
@pytest.mark.parametrize("logging_step_size", LOGGING_STEP_SIZES)
@pytest.mark.parametrize("num_samples", NUM_SAMPLES)
@pytest.mark.parametrize("start_time", START_TIMES)
@pytest.mark.parametrize("seirhd_npi_intervention", SEIRHD_NPI_STATIC_PARAM_INTERV)
def test_intervention_on_constant_param(
sample_method,
model_fixture,
end_time,
logging_step_size,
num_samples,
start_time,
seirhd_npi_intervention,
):
# Assert that sample returns expected result with intervention on constant parameter
if "SEIRHD_NPI" not in model_fixture.url:
print("skipped")
print(model_fixture.url)
pytest.skip("Only test 'SEIRHD_NPI' models with constant parameter delta")
else:
print("unskipped")
print(model_fixture.url)
processed_result = sample_method(
model_fixture.url,
end_time,
logging_step_size,
num_samples,
start_time=start_time,
static_parameter_interventions=seirhd_npi_intervention,
)["data"]
assert isinstance(processed_result, pd.DataFrame)
Expand Down

0 comments on commit 5b53a4d

Please sign in to comment.