def test_using_new_replaces_model(self, mock_remove, mock_exists):
load_options(True, False, False, False, False)
self.assertTrue(mock_remove.called)
def test_missing_options_file_raises_error(self, mock_exists, mock_loads):
with self.assertRaises(FileNotFoundError):
load_options(False, False, False, True, False)
def test_usecudnn_calls_cuda(self, mock_options, mock_cuda,
mock_load_model, mock_load, mock_is_file,
mock_exists):
self.assertEqual(load_options(True, False, False, True, False),
(self.OPTIONS_CUDA, 'Cuda model'))
def test_missing_train_dataset_raises_error(self, mock_exists,
mock_loadss):
with self.assertRaises(FileNotFoundError):
load_options(True, False, False, True, False)