@@ -310,22 +310,19 @@ def test_missing_field(self, author_missing_data):
310
310
expected = DataFrame (ex_data )
311
311
tm .assert_frame_equal (result , expected )
312
312
313
- @pytest .mark .parametrize ("max_level, columns" , [
314
- (0 , ['TextField' , 'UserField' , 'CreatedBy' , "Image" ]),
315
- (1 , ['TextField' , 'UserField.Id' ,
316
- 'UserField.Name' , 'CreatedBy' , "Image" ]),
317
- ])
313
+ @pytest .mark .parametrize ("max_level" , [0 , 1 ])
318
314
def test_records_path_with_nested_data (self ,
319
315
data_records_path_with_nested_data ,
320
- max_level , columns ):
316
+ max_level ):
321
317
test_input = data_records_path_with_nested_data
322
318
expected_data = expected_data_test_path_with_nested_data ()[max_level ]
323
319
result = json_normalize (test_input ,
324
320
record_path = ["Lookup" ],
325
321
meta = [["CreatedBy" ], ["Image" ]],
326
322
max_level = max_level )
327
- expected_df = DataFrame (data = expected_data , columns = columns )
328
- tm .assert_frame_equal (expected_df , result )
323
+ expected_df = DataFrame (data = expected_data ,
324
+ columns = result .columns .values )
325
+ tm .assert_equal (expected_df , result )
329
326
330
327
331
328
class TestNestedToRecord (object ):
0 commit comments