You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
3.1 KiB
103 lines
3.1 KiB
import pandas |
|
|
|
from .conftest import EXAMPLE_DIR_CSV_WITH_PARAMS, EXAMPLE_DIR_CSV_WO_PARAMS |
|
|
|
|
|
def test_search_params_file_ok(example_dir): |
|
from sensospot_parser.parameters import _search_params_file |
|
|
|
result = _search_params_file(example_dir / EXAMPLE_DIR_CSV_WITH_PARAMS) |
|
|
|
assert result.suffix == ".svexp" |
|
|
|
|
|
def test_search_params_file_no_parameters_folder(example_dir): |
|
from sensospot_parser.parameters import _search_params_file |
|
|
|
result = _search_params_file(example_dir / EXAMPLE_DIR_CSV_WO_PARAMS) |
|
|
|
assert result is None |
|
|
|
|
|
def test_ssearch_measurement_params_file_parameters_file(tmpdir): |
|
from sensospot_parser.parameters import _search_params_file |
|
|
|
params_dir = tmpdir / "Parameters" |
|
params_dir.mkdir() |
|
|
|
result = _search_params_file(tmpdir) |
|
|
|
assert result is None |
|
|
|
|
|
def test_parse_channel_info(example_dir): |
|
from sensospot_parser.parameters import ( |
|
_parse_measurement_params, |
|
_search_params_file, |
|
) |
|
|
|
params = _search_params_file(example_dir / EXAMPLE_DIR_CSV_WITH_PARAMS) |
|
result = _parse_measurement_params(params) |
|
|
|
expected = pandas.DataFrame( |
|
{ |
|
"Exposure.Id": [1, 2, 3], |
|
"Parameters.Channel": ["green", "red", "red"], |
|
"Parameters.Time": [100.0, 150.0, 15.0], |
|
} |
|
) |
|
|
|
assert result.equals(expected) |
|
|
|
|
|
def test_get_measurement_params_file_found(example_dir): |
|
from sensospot_parser.parameters import get_measurement_params |
|
|
|
result = get_measurement_params(example_dir / EXAMPLE_DIR_CSV_WITH_PARAMS) |
|
|
|
expected = pandas.DataFrame( |
|
{ |
|
"Exposure.Id": [1, 2, 3], |
|
"Parameters.Channel": ["green", "red", "red"], |
|
"Parameters.Time": [100.0, 150.0, 15.0], |
|
} |
|
) |
|
|
|
assert result.equals(expected) |
|
|
|
|
|
def test_get_measurement_params_file_not_found(example_dir): |
|
from sensospot_parser.parameters import get_measurement_params |
|
|
|
result = get_measurement_params(example_dir / EXAMPLE_DIR_CSV_WO_PARAMS) |
|
|
|
assert result is None |
|
|
|
|
|
def test_add_measurement_parameters_with_params_file(exposure_df, example_dir): |
|
from sensospot_parser.parameters import add_measurement_parameters |
|
|
|
folder = example_dir / EXAMPLE_DIR_CSV_WITH_PARAMS |
|
exposure_df = add_measurement_parameters(exposure_df, folder) |
|
|
|
expected = [(1, "green", 100), (2, "red", 150), (3, "red", 15)] |
|
for exposure_id, channel, time in expected: |
|
mask = exposure_df["Exposure.Id"] == exposure_id |
|
example_row = exposure_df.loc[mask].iloc[0] |
|
assert example_row["Parameters.Channel"] == channel |
|
assert example_row["Parameters.Time"] == time |
|
|
|
|
|
def test_add_measurement_parameters_without_params_file( |
|
exposure_df, example_dir |
|
): |
|
from sensospot_parser.parameters import add_measurement_parameters |
|
|
|
folder = example_dir / EXAMPLE_DIR_CSV_WO_PARAMS |
|
exposure_df = add_measurement_parameters(exposure_df, folder) |
|
|
|
for exposure_id in range(1, 4): |
|
mask = exposure_df["Exposure.Id"] == exposure_id |
|
one_exposure_data_frame = exposure_df.loc[mask] |
|
assert one_exposure_data_frame["Parameters.Channel"].hasnans |
|
assert one_exposure_data_frame["Parameters.Time"].hasnans
|
|
|