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)
