Skip to content

Commit

Permalink
slim down feature tests
Browse files Browse the repository at this point in the history
  • Loading branch information
gtrevisan committed Apr 19, 2024
1 parent 80741e8 commit a14eacd
Showing 1 changed file with 3 additions and 8 deletions.
11 changes: 3 additions & 8 deletions tests/test_cmod_features.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,10 +10,6 @@

TEST_SHOTS = [
1150805012, # Flattop Disruption
1150805013, # No Disruption
1150805014, # No Disruption
1150805015, # Rampdown Disruption
1150805016, # Rampdown Disruption
]

@pytest.fixture(scope='module')
Expand Down Expand Up @@ -71,13 +67,12 @@ def shot_settings_list():
@pytest.mark.parametrize('multiprocessing', [True, False])
@pytest.mark.parametrize('shot_settings_index', list(range(7)))
def test_features(cmod_handler, shot_settings_list, shot_settings_index, multiprocessing):
list_output, df_output, csv_processed, hdf_processed = cmod_handler.get_shots_data(
df_output, csv_processed, hdf_processed = cmod_handler.get_shots_data(
shot_ids_request=TEST_SHOTS,
shot_settings=shot_settings_list[shot_settings_index],
output_type_request=["list", "dataframe", f"{__file__}.csv", f"{__file__}.hdf5"],
num_processes=4 if multiprocessing else 1
output_type_request=["dataframe", f"{__file__}.csv", f"{__file__}.hdf5"],
num_processes=2 if multiprocessing else 1
)
assert isinstance(list_output, list)
assert isinstance(df_output, pd.DataFrame)
assert csv_processed == hdf_processed == len(TEST_SHOTS)

Expand Down

0 comments on commit a14eacd

Please sign in to comment.