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)