def test_align_no_data(self): agg_data = {'foo': []} agg_timepoints = [] res = image_utils.align_timepoints(agg_data, agg_timepoints) exp = {'foo': np.array([])} self.assertEqual(set(res.keys()), set(exp.keys())) for key in res: np.testing.assert_almost_equal(res[key], exp[key])
def test_align_one_record_no_gaps(self): agg_data = { 'foo': [ np.array([1, 2, 3, 4, 5]), ] } agg_timepoints = [np.array([1, 2, 3, 4, 5])] res = image_utils.align_timepoints(agg_data, agg_timepoints) exp = { 'foo': np.array([ [1, 2, 3, 4, 5], ]), } self.assertEqual(set(res.keys()), set(exp.keys())) for key in res: np.testing.assert_almost_equal(res[key], exp[key])