experimental assert_frame_equal with snapshot support

[15r10nk]

This branch contains a assert_frame_equal which can be used with a snapshot argument.
It uses internal pandas.testing.assert_frame_equal and provides the same arguments.

Important

this branch is experimental and will never be merged into this project.
I keep it here to simplify the development and will provide a separate package which adds pandas support for inline-snapshot when this experiment works out.

You can migrate your code by changing the import

from pandas.testing import assert_frame_equal
from inline_snapshot._pandas import assert_frame_equal

You can use it like this:

from inline_snapshot._pandas import assert_frame_equal

from inline_snapshot import snapshot
from pandas import DataFrame

def test_df():
   df = DataFrame({"col0": [1, 2]})

   # the second argument can be a snapshot,
   assert_frame_equal(df, snapshot())

   # and the generated code looks like this.
   assert_frame_equal(df, snapshot(DataFrame({"col0": {0: 1, 1: 2}})))

   # it can also be used without a snapshot
   assert_frame_equal(df, df)

[Galsor]

The code is not so specific to Pandas.

Seems to me, there is an opportunity for a generic wrapper around custom testing function.
Something like
snapshotify(assert_frame_equal)(df, snapshot())

[Galsor]

Could we imagine the import update to be automatically handled by code gen?

A "testing function" registry could store functions covered by inline-snapshot and facilitate import updates handling.

[15r10nk]

Seems to me, there is an opportunity for a generic wrapper around custom testing function.
Something like

The implementation is already generic. But I want to keep the imports, because this allows to adopt snapshot capability by changing the imports and keep the calls the same.

Could we imagine the import update to be automatically handled by code gen?

I think that it is easier to replace the imports in your code with the search and replace function of your IDE.

but we could provide a patch_testing_function() function which patches the functions in pandas.testing with the functions which can handle snapshots. A little bit evil, but could be done. This would require no import changes.