diff --git a/tests/integ/test_function.py b/tests/integ/test_function.py
index fe666f74d8..d5614a9967 100644
--- a/tests/integ/test_function.py
+++ b/tests/integ/test_function.py
@@ -1112,17 +1112,39 @@ def test_array_min_max_functions(session):
 
 @pytest.mark.xfail(reason="SNOW-844750 Waiting for BCR to complete", strict=False)
 def test_array_flatten(session):
-    df = session.create_dataframe([[[[1, 2, 3], [None], [4, 5]]],], schema=["a"])
+    df = session.create_dataframe(
+        [
+            [[[1, 2, 3], [None], [4, 5]]],
+        ],
+        schema=["a"],
+    )
     df = df.select(array_flatten(df.a).as_("flatten_a"))
-    Utils.check_answer(df, [Row(FLATTEN_A='[\n  1,\n  2,\n  3,\n  null,\n  4,\n  5\n]')], statement_params={"ENABLE_ARRAY_FLATTEN_FUNCTION": True})
+    Utils.check_answer(
+        df,
+        [Row(FLATTEN_A="[\n  1,\n  2,\n  3,\n  null,\n  4,\n  5\n]")],
+        statement_params={"ENABLE_ARRAY_FLATTEN_FUNCTION": True},
+    )
 
-    df = session.create_dataframe([[[[[1, 2], [3]]]],], schema=["a"])
+    df = session.create_dataframe(
+        [
+            [[[[1, 2], [3]]]],
+        ],
+        schema=["a"],
+    )
     df = df.select(array_flatten(df.a).as_("flatten_a"))
-    Utils.check_answer(df, [Row(FLATTEN_A='[\n  [\n    1,\n    2\n  ],\n  [\n    3\n  ]\n]')], statement_params={"ENABLE_ARRAY_FLATTEN_FUNCTION": True})
+    Utils.check_answer(
+        df,
+        [Row(FLATTEN_A="[\n  [\n    1,\n    2\n  ],\n  [\n    3\n  ]\n]")],
+        statement_params={"ENABLE_ARRAY_FLATTEN_FUNCTION": True},
+    )
 
     df = session.sql("select [[1, 2], null, [3]] as A")
     df = df.select(array_flatten(df.a).as_("flatten_a"))
-    Utils.check_answer(df, [Row(FLATTEN_A=None)], statement_params={"ENABLE_ARRAY_FLATTEN_FUNCTION": True})
+    Utils.check_answer(
+        df,
+        [Row(FLATTEN_A=None)],
+        statement_params={"ENABLE_ARRAY_FLATTEN_FUNCTION": True},
+    )
 
 
 @pytest.mark.xfail(reason="SNOW-844750 Waiting for BCR to complete", strict=False)