This project is currently only available for insiders, which mean that you can get access to it if you sponsor me. You should then have access to this repository.
pip install git ssh://[email protected]:15r10nk-insiders/inline-snapshot-pandas.git@insidersThis packages provides special assert_(frame|series|index)_equal implementation which accept a snapshot as second argument.
from pandas import DataFrame
from inline_snapshot_pandas import assert_frame_equal
from inline_snapshot import snapshot
def test_assert_equal():
df = DataFrame({"col0": [1, 2]})
assert_frame_equal(df, snapshot())pytest --inline-snapshot=createfrom pandas import DataFrame
from inline_snapshot_pandas import assert_frame_equal
from inline_snapshot import snapshot
def test_assert_equal():
df = DataFrame({"col0": [1, 2]})
assert_frame_equal(
df,
snapshot(DataFrame([{"col0": 1}, {"col0": 2}])),
)Another way to use it is to call setup() in conftest.py, which replaces the implementation which pandas uses.
from inline_snapshot_pandas import setup
setup()You can then use implementation from pandas with snapshots.
from pandas import DataFrame
from pandas.testing import assert_frame_equal
from inline_snapshot import snapshot
def test_assert_equal():
df = DataFrame({"col0": [1, 2], "col1": [1, 5j], "col3": ["a", "b"]})
# the second argument can be a snapshot
assert_frame_equal(
df,
snapshot(
DataFrame(
[
{"col0": 1, "col1": (1 0j), "col3": "a"},
{"col0": 2, "col1": 5j, "col3": "b"},
]
)
),
)The version which is currently available on pip provides functions which can be used by non insiders.
They provides a special snapshot function which is implemented as lambda value: value and can be used with the normal assert_*_equal functions of pandas.
The following code can be executed with the insider an non-insider version:
from pandas import DataFrame
from inline_snapshot_pandas import assert_frame_equal
# importing snapshot from inline_snapshot_pandas is important
from inline_snapshot_pandas import snapshot
def test_assert_equal():
df = DataFrame({"col0": [1, 2]})
assert_frame_equal(
df,
snapshot(DataFrame([{"col0": 1}, {"col0": 2}])),
)If you encounter any problems, please report an issue along with a detailed description.
Distributed under the terms of the MIT license, "inline-snapshot-pandas" is free and open source software.