clean up

src/snowflake/snowpark/dataframe.py

@@ -1254,7 +1254,6 @@ def sort(
                 )
 
         if self._select_statement:
-
             return self._with_plan(self._select_statement.sort(sort_exprs))
         return self._with_plan(Sort(sort_exprs, self._plan))
 

tests/integ/test_column.py

@@ -99,14 +99,14 @@ def test_cast_decimal(session, number_word):
 def test_cast_map_type(session):
     df = session.create_dataframe([['{"key": "1"}']], schema=["a"])
     result = df.select(parse_json(df["a"]).cast("object")).collect()
-    assert json.loads(result[0][0])
+    assert json.loads(result[0][0]) == {"key": "1"}
 
 
 @pytest.mark.localtest
 def test_cast_array_type(session):
     df = session.create_dataframe([["[1,2,3]"]], schema=["a"])
     result = df.select(parse_json(df["a"]).cast("array")).collect()
-    assert json.loads(result[0][0])
+    assert json.loads(result[0][0]) == [1, 2, 3]
 
 
 @pytest.mark.localtest