def test_works_with_hdf5_fibsem(self): train_dataset = get_train_dataset(h5py.File) dataset = train_dataset[0] origin = (105, 100, 100) numpy_dataset, expected_dataset = self.get_datasets(dataset, origin) for key in ['data', 'label', 'mask']: self.assertEqual(numpy_dataset[key].shape, expected_dataset[key].shape) np.testing.assert_almost_equal( numpy_dataset[key], expected_dataset[key], err_msg=str((key, "result", numpy_dataset[key], "desired", expected_dataset[key])), )
def test_works_with_dvid(self): train_dataset = get_train_dataset(DVIDDataInstance) dataset = train_dataset[0] dataset['body_names_to_exclude'] = ['out'] origin = (3000, 3000, 3000) numpy_dataset, expected_dataset = self.get_datasets(dataset, origin) for key in ['data', 'label', 'mask']: self.assertEqual(numpy_dataset[key].shape, expected_dataset[key].shape) np.testing.assert_almost_equal( numpy_dataset[key], expected_dataset[key], err_msg=str((key, "result", numpy_dataset[key], "desired", expected_dataset[key])), )