diff --git a/pyomeca/__init__.py b/pyomeca/__init__.py index acbe1dc..35edb80 100644 --- a/pyomeca/__init__.py +++ b/pyomeca/__init__.py @@ -1,6 +1,6 @@ +from ._version import __version__ from .analogs import Analogs from .angles import Angles from .dataarray_accessor import DataArrayAccessor from .markers import Markers from .rototrans import Rototrans -from ._version import __version__ diff --git a/pyomeca/dataarray_accessor.py b/pyomeca/dataarray_accessor.py index 8c08a1d..a09b164 100644 --- a/pyomeca/dataarray_accessor.py +++ b/pyomeca/dataarray_accessor.py @@ -509,7 +509,7 @@ def band_stop( ``` ![band_stop](/images/api/band_stop.svg) - + !!! note You can also perform a notch filter with this method. A notch filter is a band-stop filter with a narrow bandwidth. diff --git a/pyomeca/io/read.py b/pyomeca/io/read.py index 2cbd695..96e4dea 100644 --- a/pyomeca/io/read.py +++ b/pyomeca/io/read.py @@ -141,7 +141,11 @@ def read_sto_or_mot( end_header = find_end_header_in_opensim_file(filename) data = caller.from_csv( - filename, header=end_header + 1, first_column=0, time_column=0, **kwargs, + filename, + header=end_header + 1, + first_column=0, + time_column=0, + **kwargs, ) data.attrs["rate"] = (1 / (data.time[1] - data.time[0])).round().item() return data @@ -149,7 +153,12 @@ def read_sto_or_mot( def read_trc(caller: Callable, filename: Union[str, Path], **kwargs): data = caller.from_csv( - filename, header=3, first_row=6, first_column=1, time_column=1, **kwargs, + filename, + header=3, + first_row=6, + first_column=1, + time_column=1, + **kwargs, ) data.attrs["rate"] = (1 / (data.time[1] - data.time[0])).round().item() return data diff --git a/pyomeca/processing/interp.py b/pyomeca/processing/interp.py index 568a47c..dff05a1 100644 --- a/pyomeca/processing/interp.py +++ b/pyomeca/processing/interp.py @@ -14,7 +14,9 @@ def time_normalize( if norm_time: first_last_time = (0, 99) array["time"] = np.linspace( - first_last_time[0], first_last_time[1], array["time"].shape[0], + first_last_time[0], + first_last_time[1], + array["time"].shape[0], ) else: first_last_time = (array.time[0], array.time[-1]) diff --git a/tests/test_docstrings.py b/tests/test_docstrings.py index 804874d..2e9d333 100644 --- a/tests/test_docstrings.py +++ b/tests/test_docstrings.py @@ -39,7 +39,8 @@ def test_docstring_example(method): else "" ) code_block = extract_code_blocks_from_md(method.__doc__).replace( - "plt.show()", plt_show_replacer, + "plt.show()", + plt_show_replacer, ) exec(code_block, {}, {}) diff --git a/tests/test_fileio_read_csv_c3d.py b/tests/test_fileio_read_csv_c3d.py index 10c36ee..2135d2c 100644 --- a/tests/test_fileio_read_csv_c3d.py +++ b/tests/test_fileio_read_csv_c3d.py @@ -79,7 +79,8 @@ def test_read_analogs( @pytest.mark.parametrize("usecols", [[20.0]]) @pytest.mark.parametrize("extension", _extensions) def test_read_catch_error( - usecols, extension, + usecols, + extension, ): with pytest.raises(IndexError): Markers.from_csv(MARKERS_CSV) diff --git a/tests/test_processing_filter.py b/tests/test_processing_filter.py index 6507299..54f6b07 100644 --- a/tests/test_processing_filter.py +++ b/tests/test_processing_filter.py @@ -14,10 +14,12 @@ def test_proc_filters(): **EXPECTED_VALUES[32], ) is_expected_array( - ANALOGS_DATA.meca.low_pass(order=order, cutoff=5), **EXPECTED_VALUES[32], + ANALOGS_DATA.meca.low_pass(order=order, cutoff=5), + **EXPECTED_VALUES[32], ) is_expected_array( - ANALOGS_DATA.meca.high_pass(order=order, cutoff=100), **EXPECTED_VALUES[33], + ANALOGS_DATA.meca.high_pass(order=order, cutoff=100), + **EXPECTED_VALUES[33], ) is_expected_array( ANALOGS_DATA.meca.band_pass(order=order, cutoff=[10, 200]), @@ -34,16 +36,20 @@ def test_proc_filters(): **EXPECTED_VALUES[36], ) is_expected_array( - MARKERS_DATA.meca.low_pass(order=order, cutoff=5), **EXPECTED_VALUES[36], + MARKERS_DATA.meca.low_pass(order=order, cutoff=5), + **EXPECTED_VALUES[36], ) is_expected_array( - MARKERS_DATA.meca.high_pass(order=order, cutoff=10), **EXPECTED_VALUES[37], + MARKERS_DATA.meca.high_pass(order=order, cutoff=10), + **EXPECTED_VALUES[37], ) is_expected_array( - MARKERS_DATA.meca.band_pass(order=order, cutoff=[1, 10]), **EXPECTED_VALUES[38], + MARKERS_DATA.meca.band_pass(order=order, cutoff=[1, 10]), + **EXPECTED_VALUES[38], ) is_expected_array( - MARKERS_DATA.meca.band_stop(order=order, cutoff=[5, 6]), **EXPECTED_VALUES[39], + MARKERS_DATA.meca.band_stop(order=order, cutoff=[5, 6]), + **EXPECTED_VALUES[39], ) with pytest.raises(ValueError): diff --git a/tests/test_processing_marker.py b/tests/test_processing_marker.py index 58176ce..0de9059 100644 --- a/tests/test_processing_marker.py +++ b/tests/test_processing_marker.py @@ -18,7 +18,8 @@ def test_rotate(): for marker in range(n_markers): for frame in range(n_frames): expected_rotated_marker[:, marker, frame] = np.dot( - rt.isel(time=frame), markers.isel(channel=marker, time=frame), + rt.isel(time=frame), + markers.isel(channel=marker, time=frame), ) np.testing.assert_array_almost_equal( diff --git a/tests/test_processing_rt.py b/tests/test_processing_rt.py index feeb593..a5156b3 100644 --- a/tests/test_processing_rt.py +++ b/tests/test_processing_rt.py @@ -61,7 +61,8 @@ def test_construct_rt(): np.arange(0, rt_random_angles.time.size / 0.5, 1 / 0.5) rt_with_time = Rototrans( - rt_random_angles, time=np.arange(0, rt_random_angles.time.size / 100, 1 / 100), + rt_random_angles, + time=np.arange(0, rt_random_angles.time.size / 100, 1 / 100), ) assert rt_with_time.time[-1] == 0.09