def execute_test(exec_args):
result = subprocess.Popen(" ".join(exec_args),
stdout=subprocess.PIPE,
shell=True)
generated_code = result.stdout.read().decode("utf-8")
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)
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)
