jddlmZddlZddlmZddlmZerddlmZej e 5ddl m Z dddn #1swxYwYedddddZ dS)) annotationsN) TYPE_CHECKING)unstable) SchemaDict)assert_schema_equal_pyTcheck_column_order check_dtypes left_schemar right_schemar boolr returnNonec\ddlm}t||||||dS)aE Assert that the schema of the left and right frame are equal. Raises a detailed `AssertionError` if the schemas of the frames differ. This function is intended for use in unit tests. .. warning:: This functionality is considered **unstable**. It may be changed at any point without it being considered a breaking change. Parameters ---------- left_schema The first DataFrame or LazyFrame to compare. right_schema The second DataFrame or LazyFrame to compare. check_column_order Requires column order to match. check_dtypes Requires data types to match. Examples -------- >>> import polars as pl >>> from polars.testing import assert_schema_equal >>> df1 = pl.DataFrame({"b": [3, 4], "a": [1, 2]}) >>> df2 = pl.DataFrame({"a": [1, 2], "b": [3, 4]}) >>> assert_schema_equal(df1.schema, df2.schema) Traceback (most recent call last): ... AssertionError: Schemas are different (columns are not in the same order) [left]: ["b", "a"] [right]: ["a", "b"] r)SchemarN)polarsrr)r r r r rs b/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/testing/asserts/schema.pyassert_schema_equalrsXT{|-! ) r rr rr r r r rr) __future__r contextlibtypingrpolars._utils.unstablerpolars._typingrsuppress ImportError polars._plrrrrrrs"""""" ++++++*))))))Z%%33222222333333333333333  $ 00000 000sAA A