def test_function_name_csharp_default(): infile = _get_pickled_trained_model() mock_args = _get_mock_args(infile=infile, language="c_sharp") generated_code = cli.generate_code(mock_args).strip() assert 'public static double Score' in generated_code
def test_function_name(): infile = _get_pickled_trained_model() mock_args = _get_mock_args( infile=infile, language="python", function_name="predict") generated_code = cli.generate_code(mock_args).strip() assert generated_code.startswith("def predict")
def test_namespace(): infile = _get_pickled_trained_model() mock_args = _get_mock_args( infile=infile, language="c_sharp", namespace="Tests.ML") generated_code = cli.generate_code(mock_args).strip() assert generated_code.find("namespace Tests.ML {") != -1
def test_module_name(): infile = _get_pickled_trained_model() mock_args = _get_mock_args( infile=infile, language="visual_basic", module_name="TestModule") generated_code = cli.generate_code(mock_args).strip() assert generated_code.startswith("Module TestModule")
def test_package_name(): infile = _get_pickled_trained_model() mock_args = _get_mock_args( infile=infile, language="java", package_name="foo.bar.baz") generated_code = cli.generate_code(mock_args).strip() assert generated_code.startswith("package foo.bar.baz;")
def test_class_name(): infile = _get_pickled_trained_model() mock_args = _get_mock_args( infile=infile, language="java", class_name="TestClassName") generated_code = cli.generate_code(mock_args).strip() assert generated_code.startswith("public class TestClassName")
def test_generate_code(): infile = _get_pickled_trained_model() mock_args = _get_mock_args(infile=infile, language="python") generated_code = cli.generate_code(mock_args) utils.verify_python_model_is_expected( generated_code, [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13], expected_output=-47.62913662138064)
def test_generate_code(): infile = _get_pickled_trained_model() mock_args = _get_mock_args(infile=infile, language="python") generated_code = cli.generate_code(mock_args) verify_python_model_is_expected( generated_code, [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13], expected_output=-44.40540274041321)
def test_unsupported_args_are_ignored(): infile = _get_pickled_trained_model() mock_args = _get_mock_args( infile=infile, language="python", class_name="TestClassName", package_name="foo.bar.baz") generated_code = cli.generate_code(mock_args) utils.verify_python_model_is_expected( generated_code, [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13], expected_output=-47.62913662138064)