@@ -650,7 +650,7 @@ def test_unnest_without_nulls(nested_df):
650650def test_join_full ():
651651 ctx = SessionContext ()
652652 batch = pa .RecordBatch .from_arrays (
653- [pa .array ([1 , 2 , 3 ]), pa .array (['a' , 'b' , 'c' ])],
653+ [pa .array ([1 , 2 , 3 ]), pa .array (["a" , "b" , "c" ])],
654654 names = ["num" , "name" ],
655655 )
656656 df1 = ctx .create_dataframe ([[batch ]], "l" )
@@ -663,7 +663,11 @@ def test_join_full():
663663
664664 df3 = df1 .join (df2 , on = "num" , how = "full" )
665665
666- expected = {'num' : [1 , 3 , None , 5 ], 'name' : ['a' , 'c' , 'b' , None ], 'value' : [True , True , None , False ]}
666+ expected = {
667+ "num" : [1 , 3 , None , 5 ],
668+ "name" : ["a" , "c" , "b" , None ],
669+ "value" : [True , True , None , False ]
670+ }
667671 assert expected == df3 .to_pydict ()
668672
669673 # To show how user can do post-processing
@@ -672,7 +676,11 @@ def test_join_full():
672676 "l.name" ,
673677 "r.value"
674678 )
675- expected = {'num' : [1 , 3 , 2 , 5 ], 'name' : ['a' , 'c' , 'b' , None ], 'value' : [True , True , None , False ]}
679+ expected = {
680+ "num" : [1 , 3 , 2 , 5 ],
681+ "name" : ["a" , "c" , "b" , None ],
682+ "value" : [True , True , None , False ]
683+ }
676684 assert expected == df4 .to_pydict ()
677685
678686
0 commit comments