Skip to content

Commit

Permalink
fix tests for name tweak
Browse files Browse the repository at this point in the history
  • Loading branch information
TomDonoghue committed Mar 5, 2024
1 parent fc12bfd commit cafb078
Showing 1 changed file with 13 additions and 13 deletions.
26 changes: 13 additions & 13 deletions neurodsp/tests/plts/test_combined.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,20 +13,20 @@
@plot_test
def test_plot_timeseries_and_spectrum(tsig, tsig_comb, tsig_burst):

plot_timeseries_and_spectrum(tsig_comb, FS,
save_fig=True, file_path=TEST_PLOTS_PATH,
file_name='test_plot_combined_ts_psd.png')
plot_timeseries_and_spectra(tsig_comb, FS,
save_fig=True, file_path=TEST_PLOTS_PATH,
file_name='test_plot_combined_ts_psd.png')

# Test customizations
plot_timeseries_and_spectrum(tsig_comb, FS,
f_range=(3, 50), color='blue',
spectrum_kwargs={'nperseg' : 500},
ts_kwargs={'xlim' : [0, 5]},
psd_kwargs={'lw' : 2},
save_fig=True, file_path=TEST_PLOTS_PATH,
file_name='test_plot_combined_ts_psd2.png')
plot_timeseries_and_spectra(tsig_comb, FS,
f_range=(3, 50), color='blue',
spectrum_kwargs={'nperseg' : 500},
ts_kwargs={'xlim' : [0, 5]},
psd_kwargs={'lw' : 2},
save_fig=True, file_path=TEST_PLOTS_PATH,
file_name='test_plot_combined_ts_psd2.png')

# Test multi-signal input
plot_timeseries_and_spectrum(np.array([tsig, tsig_comb, tsig_burst]), FS,
save_fig=True, file_path=TEST_PLOTS_PATH,
file_name='test_plot_combined_ts_psd_multi.png')
plot_timeseries_and_spectra(np.array([tsig, tsig_comb, tsig_burst]), FS,
save_fig=True, file_path=TEST_PLOTS_PATH,
file_name='test_plot_combined_ts_psd_multi.png')

0 comments on commit cafb078

Please sign in to comment.