def test_processes_example_file_correct(tmpdir): meta = source.MetaData( chunk_size=6, recurrence_size=2, recurrence_distance=3, consistency_sample_size=10, ) output_filename = pathlib.Path(tmpdir) / "output.tdms" fix.export_correct_data( tdms_path="tests/assets/example_file.tdms", meta=meta, export_path=output_filename, ) tdms_operator = nptdms.TdmsFile(output_filename) df_result = tdms_operator.as_dataframe() assert len(df_result) == 15 assert np.array_equal( df_result.columns.values, np.array([ "/'Untitled'/'D'", "/'Untitled'/'C'", "/'Untitled'/'B'", "/'Untitled'/'A'", ]), ) assert np.array_equal(df_result["/'Untitled'/'A'"].values, np.arange(1, 16))
def main( recurrence_size: int, recurrence_distance: int, chunk_size: int, consistency_sample_size: int, output_file: str, filename: str, ): meta = source.MetaData( recurrence_distance=recurrence_distance, recurrence_size=recurrence_size, chunk_size=chunk_size, consistency_sample_size=consistency_sample_size, ) tdms_path = pathlib.Path(filename) if output_file == "": name = tdms_path.with_suffix("").name + "_corrected.tdms" export_path = tdms_path.parent.joinpath(name) else: export_path = pathlib.Path(output_file) fix.export_correct_data(tdms_path=tdms_path, meta=meta, export_path=export_path) CLI_CONFIG.update_config( recurrence_distance=recurrence_distance, recurrence_size=recurrence_size, chunk_size=chunk_size, consistency_sample_size=consistency_sample_size, ) CLI_CONFIG.to_yaml(PATH_TO_CONFIG)
def main( recurrence_size: int, recurrence_distance: int, chunk_size: int, consistency_sample_size: int, output_file: str, filename: str, segment_size: int, ): meta = source.MetaData( recurrence_distance=recurrence_distance, recurrence_size=recurrence_size, chunk_size=chunk_size, consistency_sample_size=consistency_sample_size, segment_size=segment_size, ) fix.export_correct_data(filename=filename, meta=meta, output_file=output_file) CLI_CONFIG.update_config( recurrence_distance=recurrence_distance, recurrence_size=recurrence_size, chunk_size=chunk_size, consistency_sample_size=consistency_sample_size, segment_size=segment_size, ) CLI_CONFIG.to_yaml(PATH_TO_CONFIG)
def test_prechecks_recognize_wrong_parameters(tmpdir): meta = source.MetaData( chunk_size=6, recurrence_size=3, # <- 2 would be correct recurrence_distance=3, consistency_sample_size=10, ) output_filename = pathlib.Path(tmpdir) / "output.tdms" with pytest.raises(Exception): fix.export_correct_data( tdms_path="tests/assets/example_file.tdms", meta=meta, export_path=output_filename, )
def tests_fails_for_nonexistent_filepath(): meta = source.MetaData( chunk_size=6, recurrence_size=2, recurrence_distance=3, consistency_sample_size=10, ) output_filename = pathlib.Path("NonexistentPath/tdms_file.tdms") with pytest.raises(Exception): fix.export_correct_data( tdms_path="tests/assets/example_file.tdms", meta=meta, export_path=output_filename, )
def test_fails_wrong_file(tmpdir): meta = source.MetaData( chunk_size=6, recurrence_size=2, recurrence_distance=3, consistency_sample_size=10, ) output_filename = pathlib.Path(tmpdir) / "output" with pytest.raises(Exception): fix.export_correct_data( file_name="tests/assets/example_file.txt", meta=meta, output_file=output_filename, )