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)