def test_cli_print(): print_pipeline(define_success_pipeline(), full=False, print_fn=lambda *_args, **_kwargs: None) print_pipeline(define_success_pipeline(), full=True, print_fn=lambda *_args, **_kwargs: None)
def test_execute_pipeline(): pipeline = define_success_pipeline() environment = { 'solids': { 'sum_solid': { 'inputs': { 'num': { 'csv': { 'path': script_relative_path('num.csv') } } } } } } result = execute_pipeline(pipeline, environment_dict=environment) assert result.success assert result.result_for_solid('sum_solid').transformed_value().to_dict( 'list') == { 'num1': [1, 3], 'num2': [2, 4], 'sum': [3, 7], } assert result.result_for_solid('sum_sq_solid').transformed_value().to_dict( 'list') == { 'num1': [1, 3], 'num2': [2, 4], 'sum': [3, 7], 'sum_sq': [9, 49], }
def test_cli_execute(): # currently paths in env files have to be relative to where the # script has launched so we have to simulate that cwd = os.getcwd() try: os.chdir(script_relative_path('../..')) do_execute_command( define_success_pipeline(), [script_relative_path('../../dagster_pandas/examples/pandas_hello_world/*.yml')], lambda *_args, **_kwargs: None, ) finally: # restore cwd os.chdir(cwd)
def test_cli_execute(): # currently paths in env files have to be relative to where the # script has launched so we have to simulate that cwd = os.getcwd() try: os.chdir(script_relative_path('../..')) do_execute_command( pipeline=define_success_pipeline(), env_file_list=[ script_relative_path('../../dagster_pandas/examples/pandas_hello_world/*.yml') ], raise_on_error=True, ) finally: # restore cwd os.chdir(cwd)
def test_pipeline_include(): assert define_success_pipeline()