feat: Allow user to specify a format for stdout, format for text resu…

…lt handler (#242)
GoogleCloudPlatform · Jul 30, 2021 · 1259529 · 1259529
1 parent f41759b
commit 1259529
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/unit/result_handlers/test_text.py b/tests/unit/result_handlers/test_text.py
@@ -38,8 +38,9 @@ def test_import(module_under_test):
 
 def test_basic_result_handler(module_under_test):
     """Test basic handler executes """
+    format = "json"
     result_df = DataFrame(SAMPLE_RESULT_DATA)
     result_handler = module_under_test.TextResultHandler()
 
-    handler_output = result_handler.execute(SAMPLE_CONFIG, result_df)
+    handler_output = result_handler.execute(SAMPLE_CONFIG, format, result_df)
     assert handler_output["count"].sum() == result_df["count"].sum()