class LoadFileWidgetModelTest(unittest.TestCase):
def setUp(self):
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.model = BrowseFileWidgetModel(self.loaded_data, self.context)
def assert_model_empty(self):
self.assertEqual(self.model.loaded_workspaces, [])
self.assertEqual(self.model.loaded_filenames, [])
self.assertEqual(self.model.loaded_runs, [])
# ------------------------------------------------------------------------------------------------------------------
# TESTS
# ------------------------------------------------------------------------------------------------------------------
def test_model_initialized_with_empty_lists_of_loaded_data(self):
self.assert_model_empty()
def test_executing_load_without_filenames_does_nothing(self):
self.model.execute()
self.assert_model_empty()
def test_execute_successfully_loads_given_files(self):
files = ['EMU00019489.nxs', 'emu00006473.nxs', 'emu00006475.nxs']
self.model.loadData(files)
self.model.execute()
six.assertCountEqual(self, [
os.path.split(filename)[-1]
for filename in self.model.loaded_filenames
], files)
six.assertCountEqual(self, self.model.loaded_runs,
[[19489], [6473], [6475]])
def test_model_is_cleared_correctly(self):
files = ['EMU00019489.nxs', 'emu00006473.nxs', 'emu00006475.nxs']
self.model.loadData(files)
self.model.execute()
self.assertEqual(len(self.model.loaded_filenames), 3)
self.model.clear()
self.assert_model_empty()
def test_execute_throws_if_one_file_does_not_load_correctly_but_still_loads_other_files(
self):
files = ['EMU00019489.nxs', 'emu00006473.nxs', 'NonExistent.nxs']
self.model.loadData(files)
with self.assertRaises(ValueError):
self.model.execute()
six.assertCountEqual(self, [
os.path.split(filename)[-1]
for filename in self.model.loaded_filenames
], ['EMU00019489.nxs', 'emu00006473.nxs'])
six.assertCountEqual(self, self.model.loaded_runs, [[19489], [6473]])
class LoadFileWidgetModelTest(unittest.TestCase):
def setUp(self):
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.model = BrowseFileWidgetModel(self.loaded_data, self.context)
def assert_model_empty(self):
self.assertEqual(self.model.loaded_workspaces, [])
self.assertEqual(self.model.loaded_filenames, [])
self.assertEqual(self.model.loaded_runs, [])
# ------------------------------------------------------------------------------------------------------------------
# TESTS
# ------------------------------------------------------------------------------------------------------------------
def test_model_initialized_with_empty_lists_of_loaded_data(self):
self.assert_model_empty()
def test_executing_load_without_filenames_does_nothing(self):
self.model.execute()
self.assert_model_empty()
def test_execute_successfully_loads_given_files(self):
files = ['EMU00019489.nxs', 'emu00006473.nxs', 'emu00006475.nxs']
self.model.loadData(files)
self.model.execute()
# We make the filename all uppercase to deal with some os's capitalising the instrument and some not.
# We just need to check the right files were added to the model.
expected_files = ['EMU00019489.NXS', 'EMU00006473.NXS', 'EMU00006475.NXS']
self.assertCountEqual([os.path.split(filename.upper())[-1] for filename in self.model.loaded_filenames], expected_files)
self.assertCountEqual(self.model.loaded_runs, [[19489], [6473], [6475]])
def test_model_is_cleared_correctly(self):
files = ['EMU00019489.nxs', 'emu00006473.nxs', 'emu00006475.nxs']
self.model.loadData(files)
self.model.execute()
self.assertEqual(len(self.model.loaded_filenames), 3)
self.model.clear()
self.assert_model_empty()
def test_execute_throws_if_one_file_does_not_load_correctly_but_still_loads_other_files(self):
files = ['EMU00019489.nxs', 'emu00006473.nxs', 'NonExistent.nxs']
self.model.loadData(files)
with self.assertRaises(ValueError):
self.model.execute()
# We make the filename all uppercase to deal with some os's capitalising the instrument and some not.
# We just need to check the right files were added to the model.
self.assertCountEqual([os.path.split(filename.upper())[-1] for filename in self.model.loaded_filenames],
['EMU00019489.NXS', 'EMU00006473.NXS'])
self.assertCountEqual(self.model.loaded_runs, [[19489], [6473]])
class LoadFileWidgetModelTest(unittest.TestCase):
def setUp(self):
setup_context_for_tests(self)
self.data_context.instrument = 'EMU'
self.model = BrowseFileWidgetModel(self.loaded_data, self.context)
def assert_model_empty(self):
self.assertEqual(self.model.loaded_workspaces, [])
self.assertEqual(self.model.loaded_filenames, [])
self.assertEqual(self.model.loaded_runs, [])
# ------------------------------------------------------------------------------------------------------------------
# TESTS
# ------------------------------------------------------------------------------------------------------------------
def test_model_initialized_with_empty_lists_of_loaded_data(self):
self.assert_model_empty()
def test_executing_load_without_filenames_does_nothing(self):
self.model.execute()
self.assert_model_empty()
def test_execute_successfully_loads_given_files(self):
files = ['EMU00019489.nxs', 'emu00006473.nxs', 'emu00006475.nxs']
self.model.loadData(files)
self.model.execute()
six.assertCountEqual(self, [os.path.split(filename)[-1] for filename in self.model.loaded_filenames], files)
six.assertCountEqual(self, self.model.loaded_runs, [[19489], [6473], [6475]])
def test_model_is_cleared_correctly(self):
files = ['EMU00019489.nxs', 'emu00006473.nxs', 'emu00006475.nxs']
self.model.loadData(files)
self.model.execute()
self.assertEqual(len(self.model.loaded_filenames), 3)
self.model.clear()
self.assert_model_empty()
def test_execute_throws_if_one_file_does_not_load_correctly_but_still_loads_other_files(self):
files = ['EMU00019489.nxs', 'emu00006473.nxs', 'NonExistent.nxs']
self.model.loadData(files)
with self.assertRaises(ValueError):
self.model.execute()
six.assertCountEqual(self, [os.path.split(filename)[-1] for filename in self.model.loaded_filenames],
['EMU00019489.nxs', 'emu00006473.nxs'])
six.assertCountEqual(self, self.model.loaded_runs, [[19489], [6473]])
class LoadFileWidgetModelTest(unittest.TestCase):
def setUp(self):
self.data = MuonLoadData()
self.model = BrowseFileWidgetModel(self.data)
patcher = mock.patch('Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def mock_load_function(self, files_to_load, load_return_values):
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(side_effect=load_return_values)
self.model.loadData(files_to_load)
def assert_model_empty(self):
self.assertEqual(self.model.loaded_workspaces, [])
self.assertEqual(self.model.loaded_filenames, [])
self.assertEqual(self.model.loaded_runs, [])
# ------------------------------------------------------------------------------------------------------------------
# TESTS
# ------------------------------------------------------------------------------------------------------------------
def test_model_initialized_with_empty_lists_of_loaded_data(self):
self.assert_model_empty()
def test_executing_load_without_filenames_does_nothing(self):
self.model.execute()
self.assert_model_empty()
def test_execute_successfully_loads_given_files(self):
files = ['EMU00019489.nxs', 'EMU00019490.nxs', 'EMU00019491.nxs']
load_return_vals = [([1 + i], 19489 + i, filename) for i, filename in enumerate(files)]
self.mock_load_function(files, load_return_vals)
self.model.execute()
six.assertCountEqual(self, self.model.loaded_workspaces, [[1], [2], [3]])
six.assertCountEqual(self, self.model.loaded_filenames, files)
six.assertCountEqual(self, self.model.loaded_runs, [19489, 19490, 19491])
def test_model_is_cleared_correctly(self):
files = [r'EMU00019489.nxs', r'EMU00019490.nxs', r'EMU00019491.nxs']
load_return_vals = [([1 + i], 19489 + i, filename) for i, filename in enumerate(files)]
self.mock_load_function(files, load_return_vals)
self.model.execute()
self.assertEqual(len(self.model.loaded_filenames), 3)
self.model.clear()
self.assert_model_empty()
def test_execute_throws_if_one_file_does_not_load_correctly_but_still_loads_other_files(self):
files = [r'EMU00019489.nxs', r'EMU00019490.nxs', r'EMU00019491.nxs']
load_return_vals = [([1 + i], 19489 + i, filename) for i, filename in enumerate(files)]
# Mock load to throw on a particular index
self.load_utils_patcher.load_workspace_from_filename = mock.Mock()
self.load_utils_patcher.load_workspace_from_filename.side_effect = iter(IteratorWithException(load_return_vals, [1]))
self.model.loadData(files)
with self.assertRaises(ValueError):
self.model.execute()
six.assertCountEqual(self, self.model.loaded_filenames, [files[i] for i in [0, 2]])
six.assertCountEqual(self, self.model.loaded_runs, [19489, 19491])
class LoadFileWidgetModelTest(unittest.TestCase):
def setUp(self):
self.data = MuonLoadData()
self.model = BrowseFileWidgetModel(self.data)
patcher = mock.patch(
'Muon.GUI.Common.load_file_widget.model.load_utils')
self.addCleanup(patcher.stop)
self.load_utils_patcher = patcher.start()
def mock_load_function(self, files_to_load, load_return_values):
self.load_utils_patcher.load_workspace_from_filename = mock.Mock(
side_effect=load_return_values)
self.model.loadData(files_to_load)
def assert_model_empty(self):
self.assertEqual(self.model.loaded_workspaces, [])
self.assertEqual(self.model.loaded_filenames, [])
self.assertEqual(self.model.loaded_runs, [])
# ------------------------------------------------------------------------------------------------------------------
# TESTS
# ------------------------------------------------------------------------------------------------------------------
def test_model_initialized_with_empty_lists_of_loaded_data(self):
self.assert_model_empty()
def test_executing_load_without_filenames_does_nothing(self):
self.model.execute()
self.assert_model_empty()
def test_execute_successfully_loads_given_files(self):
files = ['EMU00019489.nxs', 'EMU00019490.nxs', 'EMU00019491.nxs']
load_return_vals = [([1 + i], 19489 + i, filename)
for i, filename in enumerate(files)]
self.mock_load_function(files, load_return_vals)
self.model.execute()
six.assertCountEqual(self, self.model.loaded_workspaces,
[[1], [2], [3]])
six.assertCountEqual(self, self.model.loaded_filenames, files)
six.assertCountEqual(self, self.model.loaded_runs,
[19489, 19490, 19491])
def test_model_is_cleared_correctly(self):
files = [r'EMU00019489.nxs', r'EMU00019490.nxs', r'EMU00019491.nxs']
load_return_vals = [([1 + i], 19489 + i, filename)
for i, filename in enumerate(files)]
self.mock_load_function(files, load_return_vals)
self.model.execute()
self.assertEqual(len(self.model.loaded_filenames), 3)
self.model.clear()
self.assert_model_empty()
def test_execute_throws_if_one_file_does_not_load_correctly_but_still_loads_other_files(
self):
files = [r'EMU00019489.nxs', r'EMU00019490.nxs', r'EMU00019491.nxs']
load_return_vals = [([1 + i], 19489 + i, filename)
for i, filename in enumerate(files)]
# Mock load to throw on a particular index
self.load_utils_patcher.load_workspace_from_filename = mock.Mock()
self.load_utils_patcher.load_workspace_from_filename.side_effect = iter(
IteratorWithException(load_return_vals, [1]))
self.model.loadData(files)
with self.assertRaises(ValueError):
self.model.execute()
six.assertCountEqual(self, self.model.loaded_filenames,
[files[i] for i in [0, 2]])
six.assertCountEqual(self, self.model.loaded_runs, [19489, 19491])
