search

m3dev / gokart

Gokart solves reproducibility, task dependencies, constraints of good code, and ease of use for Machine Learning Pipeline.
https://gokart.readthedocs.io/en/latest/
MIT License
305 stars 57 forks source link

Feature Request: pd.testing.assert_frame_equal wrapper #243

Open vaaaaanquish opened 3 years ago

vaaaaanquish commented 3 years ago

Gokart is based on a DataFrame. But pd.testing.assert_frame_equal is not a simple way to compare df.

This is especially true for sort and index problems.

now

expected = pd.DataFrame(...)
resulted = gokart.build(Hoge())

expected  = expected.sort_values([col1, col2, ...]).reset_index(drop=True)
resulted  = resulted.sort_values([col1, col2, ...]).reset_index(drop=True)

pd.testing.assert_frame_equal(resulted, expected)

feature image

expected = pd.DataFrame(...)
resulted = gokart.build(Hoge())

gokart.testing.assert_frame_equal(resulted, expected)