def test_barycenter_times(self): '''Test image production.''' scanset = ScanSet('test.hdf5') scanset.barycenter_times() assert 'barytime' in scanset.colnames assert np.all( np.abs(scanset['barytime'] - scanset['time']) < 9 * 60 / 86400)
def test_apply_user_filt(self): '''Test apply user filts.''' def user_fun(scanset): return scanset['barytime'] - np.floor(scanset['barytime']) scanset = ScanSet('test.hdf5') scanset.barycenter_times() phase_in_sec = scanset.apply_user_filter(user_fun, 'Phase_in_sec') assert np.min(phase_in_sec) >= 0 assert np.max(phase_in_sec) <= 1 assert np.all(phase_in_sec == scanset['Phase_in_sec'])