Fix formatting to pass black

apache · May 18, 2021 · 34f9d64 · 34f9d64
1 parent c92f7b0
commit 34f9d64
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/tests/db_engine_specs/hive_tests.py b/tests/db_engine_specs/hive_tests.py
@@ -387,20 +387,16 @@ def is_correct_result(data: List, result: List) -> bool:
         df = pd.DataFrame({"partition": data})
         return HiveEngineSpec._latest_partition_from_df(df) == result
 
-    assert is_correct_result(["ds=01-01-19"],["01-01-19"])
+    assert is_correct_result(["ds=01-01-19"], ["01-01-19"])
     assert is_correct_result(
-        ["ds=01-01-19","ds=01-03-19","ds=01-02-19"],
-        ["01-03-19"]
+        ["ds=01-01-19", "ds=01-03-19", "ds=01-02-19"], ["01-03-19"]
     )
+    assert is_correct_result(["ds=01-01-19/hour=1"], ["01-01-19", "1"])
     assert is_correct_result(
-        ["ds=01-01-19/hour=1"],
-        ["01-01-19","1"]
+        ["ds=01-01-19/hour=1", "ds=01-03-19/hour=1", "ds=01-02-19/hour=1"],
+        ["01-03-19", "1"],
     )
     assert is_correct_result(
-        ["ds=01-01-19/hour=1","ds=01-03-19/hour=1","ds=01-02-19/hour=1"],
-        ["01-03-19","1"]
-    )
-    assert is_correct_result(
-        ["ds=01-01-19/hour=1","ds=01-03-19/hour=1","ds=01-02-19/hour=2"],
-        ["01-03-19","1"]
+        ["ds=01-01-19/hour=1", "ds=01-03-19/hour=1", "ds=01-02-19/hour=2"],
+        ["01-03-19", "1"],
     )