def test_repeat_dataset(self): rawframe_dataset = RawframeDataset(self.frame_ann_file, self.frame_pipeline, self.data_prefix) repeat_dataset = RepeatDataset(rawframe_dataset, 5) assert len(repeat_dataset) == 10 result_a = repeat_dataset[0] result_b = repeat_dataset[2] assert set(result_a.keys()) == set(result_b.keys()) for key in result_a: if isinstance(result_a[key], np.ndarray): assert np.equal(result_a[key], result_b[key]).all() elif isinstance(result_a[key], list): assert all( np.array_equal(a, b) for (a, b) in zip(result_a[key], result_b[key])) else: assert result_a[key] == result_b[key]
def test_repeat_dataset(self): dataset_cfg = dict(type='RawframeDataset', ann_file=self.frame_ann_file, pipeline=self.frame_pipeline, data_prefix=self.data_prefix) repeat_dataset = RepeatDataset(dataset_cfg, 5) assert len(repeat_dataset) == 10 result_a = repeat_dataset[0] result_b = repeat_dataset[2] assert set(result_a) == set(result_b) for key in result_a: if isinstance(result_a[key], np.ndarray): assert np.equal(result_a[key], result_b[key]).all() elif isinstance(result_a[key], list): assert all( np.array_equal(a, b) for (a, b) in zip(result_a[key], result_b[key])) else: assert result_a[key] == result_b[key]