Skip to content

Commit

Permalink
TST: Adding DataFrame.to_dict(orient=records) numeric consistency tes…
Browse files Browse the repository at this point in the history
…t [Ref pandas-dev#22620]
  • Loading branch information
sofiane87 committed Nov 2, 2019
1 parent 0de9955 commit b72906a
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions pandas/tests/frame/test_convert_to.py
Original file line number Diff line number Diff line change
Expand Up @@ -612,3 +612,11 @@ def test_to_dict_wide(self):
result = df.to_dict("records")[0]
expected = {"A_{:d}".format(i): i for i in range(256)}
assert result == expected

def test_to_dict_orient_dtype(self):
# https://github.com/pandas-dev/pandas/issues/22620
input_data = {'a': [1, 2, 3], 'b': [1.0, 2.0, 3.0], 'c': ['X', 'Y', 'Z']}
df = DataFrame(input_data)
result = {col: type(value) for col, value in df.to_dict('records')[0].items()}
expected = {col: type(data[0]) for col, data in input_data.items()}
assert result == expected

0 comments on commit b72906a

Please sign in to comment.