@@ -396,7 +396,7 @@ def test_marquez_dag_with_extractor(mock_get_or_create_marquez_client,
396396 run_id = None
397397 )
398398 assert mock_marquez_client .create_job .mock_calls [0 ]. \
399- kwargs ['context' ].get ('extract' ) == 'extract'
399+ kwargs ['context' ].get ('extract' ) == 'extract'
400400
401401 # run is created
402402 mock_marquez_client .create_job_run .assert_called_once_with (
@@ -455,7 +455,7 @@ def test_marquez_dag_with_extractor(mock_get_or_create_marquez_client,
455455 ])
456456
457457 assert mock_marquez_client .create_job .mock_calls [0 ]. \
458- kwargs ['context' ].get ('extract' ) == 'extract'
458+ kwargs ['context' ].get ('extract' ) == 'extract'
459459
460460 mock_marquez_client .mark_job_run_as_completed .assert_called_once_with (
461461 run_id = run_id ,
@@ -637,7 +637,7 @@ def test_marquez_dag_with_extract_on_complete(
637637 )
638638 ])
639639 assert mock_marquez_client .create_job .mock_calls [0 ]. \
640- kwargs ['context' ].get ('extract_on_complete' ) == 'extract_on_complete'
640+ kwargs ['context' ].get ('extract_on_complete' ) == 'extract_on_complete'
641641
642642 # run is created
643643 mock_marquez_client .create_job_run .assert_called_once_with (
@@ -677,7 +677,7 @@ def test_marquez_dag_with_extract_on_complete(
677677 ])
678678
679679 assert mock_marquez_client .create_job .mock_calls [0 ]. \
680- kwargs ['context' ].get ('extract_on_complete' ) == 'extract_on_complete'
680+ kwargs ['context' ].get ('extract_on_complete' ) == 'extract_on_complete'
681681
682682 mock_marquez_client .mark_job_run_as_completed .assert_called_once_with (
683683 run_id = run_id ,
0 commit comments