class TestData(): def setup(self): self.nr_dims = 10 np.random.seed(1) self.data = SstatsMap('expected_results/test_data') # Generate some data. self.xx = np.random.randn(3, self.nr_dims) self.info = { 'label': 4, 'nr_slices': 3, 'begin_frames': [1, 3, 10], 'end_frames': [2, 9, 15], 'nr_descs_per_slice': [10, 9, 13]} def test_write_data(self): self.data.write('test_1', self.xx[0]) self.data.write('test_2', self.xx[1], info=self.info) self.data.write('test_3', self.xx[2], info=dict(label=10)) def test_read_data(self): # Nose tests in random order. self.data.write('test_1', self.xx[0]) self.data.write('test_2', self.xx[1], info=self.info) test_1_data = self.data.read('test_1') assert_array_almost_equal(self.xx[0], test_1_data) test_2_data = self.data.read('test_2') assert_array_almost_equal(self.xx[1], test_2_data) test_2_info = self.data.read_info('test_2') assert test_2_info == self.info @raises(Exception) def test_read_data_exception(self): self.data.read('file_that_does_not_exist') def test_check_data(self): self.data.write('test_1', self.xx) self.data.write('test_2', []) assert self.data.check(['test_1'], self.nr_dims, verbose=False) == True assert self.data.check( ['test_1'], self.nr_dims + 1, verbose=False) == False assert self.data.check( ['test_1', 'test_2'], self.nr_dims, verbose=False) == False