def test_crop_data_removes_edge_data(self):
go_data = GoData()
array = np.arange(324.0).reshape((1, 18, 18))
cropped_array = go_data.crop_data(array)
assert np.array_equal(cropped_array, array[:, 8:10, 8:10])
def test_crop_data_removes_edge_data(self):
go_data = GoData()
array = np.arange(324.0).reshape((1, 18, 18))
cropped_array = go_data.crop_data(array)
assert np.array_equal(cropped_array, array[:, 8:10, 8:10])
def test_convert_mat_file_to_numpy_file_writes_extracted_numpys_to_files(self, mock_numpy_save, h5py_file_mock):
go_data = GoData()
go_data.convert_mat_data_to_numpy_array = Mock(side_effect=[1, 2])
go_data.crop_data = lambda x: x
go_data.convert_mat_file_to_numpy_file('')
assert mock_numpy_save.call_args_list[0] == ((os.path.join('images_') + '.npy', 1),)
assert mock_numpy_save.call_args_list[1] == ((os.path.join('depths_') + '.npy', 2),)
def test_convert_mat_file_to_numpy_file_reads_the_mat_file(self, h5py_file_mock, mock_numpy_save):
mat_file_name = 'fake name'
go_data = GoData()
go_data.crop_data = Mock()
go_data.convert_mat_data_to_numpy_array = Mock()
go_data.convert_mat_file_to_numpy_file(mat_file_name)
assert h5py_file_mock.call_args == ((mat_file_name, 'r'),)
def test_convert_mat_file_to_numpy_file_writes_extracted_numpys_to_files(self, mock_numpy_save, h5py_file_mock):
go_data = GoData()
go_data.convert_mat_data_to_numpy_array = Mock(side_effect=[1, 2])
go_data.crop_data = lambda x: x
go_data.convert_mat_file_to_numpy_file('')
assert mock_numpy_save.call_args_list[0] == ((os.path.join('images_') + '.npy', 1),)
assert mock_numpy_save.call_args_list[1] == ((os.path.join('depths_') + '.npy', 2),)
def test_convert_mat_file_to_numpy_file_reads_the_mat_file(self, h5py_file_mock, mock_numpy_save):
mat_file_name = 'fake name'
go_data = GoData()
go_data.crop_data = Mock()
go_data.convert_mat_data_to_numpy_array = Mock()
go_data.convert_mat_file_to_numpy_file(mat_file_name)
assert h5py_file_mock.call_args == ((mat_file_name, 'r'),)
def test_convert_mat_file_to_numpy_file_passes_can_be_called_on_a_specific_number_of_images(
self, mock_numpy_save, h5py_file_mock):
go_data = GoData()
mock_convert = Mock()
go_data.convert_mat_data_to_numpy_array = mock_convert
go_data.crop_data = Mock()
go_data.convert_mat_file_to_numpy_file('', number_of_samples=2)
assert mock_convert.call_args[1]['number_of_samples'] == 2
def test_convert_mat_file_to_numpy_file_passes_can_be_called_on_a_specific_number_of_images(self,
mock_numpy_save,
h5py_file_mock):
go_data = GoData()
mock_convert = Mock()
go_data.convert_mat_data_to_numpy_array = mock_convert
go_data.crop_data = Mock()
go_data.convert_mat_file_to_numpy_file('', number_of_samples=2)
assert mock_convert.call_args[1]['number_of_samples'] == 2
def test_convert_mat_file_to_numpy_file_calls_extract_mat_data_to_numpy_array(self, h5py_file_mock,
mock_numpy_save):
h5py_file_mock.return_value = 'fake mat data'
go_data = GoData()
go_data.convert_mat_data_to_numpy_array = Mock()
go_data.crop_data = Mock()
go_data.convert_mat_file_to_numpy_file('')
assert go_data.convert_mat_data_to_numpy_array.call_args_list[1][0] == ('fake mat data', 'depths')
assert go_data.convert_mat_data_to_numpy_array.call_args_list[0][0] == ('fake mat data', 'images')
def test_convert_mat_file_to_numpy_file_calls_extract_mat_data_to_numpy_array(self, h5py_file_mock,
mock_numpy_save):
h5py_file_mock.return_value = 'fake mat data'
go_data = GoData()
go_data.convert_mat_data_to_numpy_array = Mock()
go_data.crop_data = Mock()
go_data.convert_mat_file_to_numpy_file('')
assert go_data.convert_mat_data_to_numpy_array.call_args_list[1][0] == ('fake mat data', 'depths')
assert go_data.convert_mat_data_to_numpy_array.call_args_list[0][0] == ('fake mat data', 'images')