@@ -1820,8 +1820,9 @@ def test_prepare_emptyroom_bads(bads):
1820
1820
@pytest .mark .parametrize ("set_annot_when" , ("before" , "after" ))
1821
1821
@pytest .mark .parametrize ("raw_meas_date" , ("orig" , None ))
1822
1822
@pytest .mark .parametrize ("raw_er_meas_date" , ("orig" , None ))
1823
+ @pytest .mark .parametrize ("equal_sfreq" , (False , True ))
1823
1824
def test_prepare_emptyroom_annot_first_samp (
1824
- set_annot_when , raw_meas_date , raw_er_meas_date
1825
+ set_annot_when , raw_meas_date , raw_er_meas_date , equal_sfreq
1825
1826
):
1826
1827
"""Test prepare_emptyroom."""
1827
1828
raw = read_raw_fif (raw_fname , allow_maxshield = "yes" , verbose = False )
@@ -1861,12 +1862,15 @@ def test_prepare_emptyroom_annot_first_samp(
1861
1862
assert set_annot_when == "after"
1862
1863
meas_date = "from_raw"
1863
1864
want_date = raw .info ["meas_date" ]
1865
+ if not equal_sfreq :
1866
+ with raw_er .info ._unlock ():
1867
+ raw_er .info ["sfreq" ] -= 100
1864
1868
raw_er_prepared = maxwell_filter_prepare_emptyroom (
1865
1869
raw_er = raw_er , raw = raw , meas_date = meas_date , emit_warning = True
1866
1870
)
1867
1871
assert raw_er .first_samp == raw_er_first_samp_orig
1868
1872
assert raw_er_prepared .info ["meas_date" ] == want_date
1869
- assert raw_er_prepared .first_samp == raw .first_samp
1873
+ assert raw_er_prepared .first_time == raw .first_time
1870
1874
1871
1875
# Ensure (movement) annotations carry over regardless of whether they're
1872
1876
# set before or after preparation
@@ -1878,7 +1882,8 @@ def test_prepare_emptyroom_annot_first_samp(
1878
1882
prop_bad = np .isnan (raw .get_data ([0 ], reject_by_annotation = "nan" )).mean ()
1879
1883
assert 0.3 < prop_bad < 0.4
1880
1884
assert len (raw_er_prepared .annotations ) == want_annot
1881
- prop_bad_er = np .isnan (
1882
- raw_er_prepared .get_data ([0 ], reject_by_annotation = "nan" )
1883
- ).mean ()
1884
- assert_allclose (prop_bad , prop_bad_er )
1885
+ if equal_sfreq :
1886
+ prop_bad_er = np .isnan (
1887
+ raw_er_prepared .get_data ([0 ], reject_by_annotation = "nan" )
1888
+ ).mean ()
1889
+ assert_allclose (prop_bad , prop_bad_er )
0 commit comments