def test_question_answer_demo(self, opt, args, kwargs): sys.argv = [ BIN, 'question_answer', 'train_data', 'validation_data', 'export' ] + opt with patch_qa() as run: cli.main() run.assert_called_once_with('train_data', 'validation_data', 'export', *args, **kwargs)
def test_text_classification_demo(self, opt, args, kwargs): sys.argv = [ BIN, 'text_classification', 'data', 'lite', 'label', 'vocab' ] + opt with patch_text() as run: cli.main() run.assert_called_once_with('data', 'lite', 'label', 'vocab', *args, **kwargs)
def test_init(self, tf_opt, expected_tf): sys.argv = [BIN, 'image_classification', 'data', 'lite', 'label' ] + tf_opt with patch_image() as run, patch_setup() as setup: cli.main() setup.assert_called_once_with(expected_tf) run.assert_called_once_with('data', 'lite', 'label', 'efficientnet_b0')
def test_text_classification_demo_lack_param(self): sys.argv = [BIN, 'text_classification', 'data'] with patch_text() as run: with self.assertRaisesRegex(fire.core.FireExit, '2'): cli.main() run.assert_not_called()
def test_image_classification_demo(self, opt, args, kwargs): sys.argv = [BIN, 'image_classification', 'data', 'export'] + opt with patch_image() as run: cli.main() run.assert_called_once_with('data', 'export', *args, **kwargs)
def test_help(self, opt): sys.argv = opt with self.assertRaisesRegex(fire.core.FireExit, '0'): cli.main()
def test_invalid_command(self): sys.argv = [BIN, 'invalid_command'] with self.assertRaisesRegex(fire.core.FireExit, '2'): cli.main()
def test_question_answer_lack_param(self): sys.argv = [BIN, 'question_answer', 'train_data', 'validation_data'] with patch_qa() as run: with self.assertRaisesRegex(fire.core.FireExit, '2'): cli.main() run.assert_not_called()