h@+dZddlZddlZddlmZmZddlmZdZ dZ dZ dZ d Z dd Zd Zd Z dd Zy)z,Testing functionality for geopandas objects.N) GeoDataFrame GeoSeries) GeometryDtypecJtj5tjddtt |dr|j cdddSt |dr|j cdddStj |cdddS#1swYyxYw)z) tolerance)geom_equals_exactrrrs r_geom_almost_equals_maskrEsG$ t/?@ ==4== ( * ;t $ &rct|tr(t|tr|j}|j}t||j S)a Test for 'almost' geometric equality. Empty or missing geometries considered equal. This method allows small difference in the coordinates, but this requires coordinates be in the same order for all components of a geometry. Parameters ---------- this, that : arrays of Geo objects (or anything that has an `is_empty` property) Returns ------- bool True if all geometries in left almost equal geometries in right ) isinstancergeometryrrrs rgeom_almost_equalsr#]s?$$ %*T<*H}}}} #D$ / 3 3 55rc >t|t|k(sJt|dt||rvd} t|jts J| j |jt|jts J| j |j|r+t|j t |j sJ|rFt|tsJt|t |sJ|r]|j|jk(sDJt|ts t|}t|tst||j }|j j|j s!Jd|j d|j |rH|j|jk(js!Jd|jd|j|rFt|jj}t|jj}|sFtj5tj ddt"t%|||dddyt%|||y#1swYyxYw) ar Test util for checking that two GeoSeries are equal. Parameters ---------- left, right : two GeoSeries check_dtype : bool, default False If True, check geo dtype [only included so it's a drop-in replacement for assert_series_equal]. check_index_type : bool, default False Check that index types are equal. check_series_type : bool, default True Check that both are same type (*and* are GeoSeries). If False, will attempt to convert both into GeoSeries. check_less_precise : bool, default False If True, use geom_equals_exact with relative error of 0.5e-6. If False, use geom_equals. check_geom_type : bool, default False If True, check that all the geom types are equal. check_crs: bool, default True If `check_series_type` is True, then also check that the crs matches. normalize: bool, default False If True, normalize the geometries before comparing equality. Typically useful with ``check_less_precise=True``, which uses ``geom_equals_exact`` and requires exact coordinate order. z != z(dtype should be a GeometryDtype, got {0})indexzindex: ztype: rz CRS mismatchN)lenr!dtyperformatr%typercrsequals geom_typerarray normalizer r r r _check_equality) leftright check_dtypecheck_index_typecheck_series_typecheck_less_precisecheck_geom_type check_crsr.msgs rassert_geoseries_equalr9vsL t9E "Bs4ykc%j\$BB "8$**m4Lcjj6LL4%++}5Mszz$**7MM5$**d5;;&7888$ ***$U ,,, 88uyy( (($ *T?D%+e4::6E ::  U[[ )RWTZZLU[[M+RR )%//1668 T^^$D(9 : 8--/0%++//12   $ $ & =  # #Hnk J D%); < = = e%78  = =s )JJcHt|}t|dkDr|dddzS|S)zTruncate WKT repr of geom.dNz...)strr&)geomss r_truncated_stringr?s+ D A 1v|#wrc hd}|rd}t||}nd}t||}|js||}||}t|j t |t ||j j|t|jdt|jdy)Nz{0} out of {1} geometries are not {3}equal. Indices where geometries are not {3}equal: {2} The first not {3}equal geometry: Left: {4} Right: {5} zalmost r) rrrAssertionErrorr(r&r%to_listr?iloc)r0r1r5assert_error_messagepreciseequalunequal_left_geomsunequal_right_geomss rr/r/s (u5!$. 99;!5&\#UFm ' '&'D "((002!"4"9"9!"<=!"5":":1"=>     rc F ddlm} m} |rt |t sJt |t|sJ| r|j |jn|j|jvr|j|jvrnj|js |jsnQ|j|jk(s8Jt |t s t |}t |t s t |}|j|jk(s;Jd|jd|jd|jd|j|r|j|}| |j|j|d |jjD].\} }t |tst!|| || | |||| 0|j|jk(sJ|j#d }|j#d }| |||||d y#t$r ddlm} m} YwxYw)a{Check that two GeoDataFrames are equal. Parameters ---------- left, right : two GeoDataFrames check_dtype : bool, default True Whether to check the DataFrame dtype is identical. check_index_type, check_column_type : bool, default 'equiv' Check that index types are equal. check_frame_type : bool, default True Check that both are same type (*and* are GeoDataFrames). If False, will attempt to convert both into GeoDataFrame. check_like : bool, default False If true, ignore the order of rows & columns check_less_precise : bool, default False If True, use geom_equals_exact. if False, use geom_equals. check_geom_type : bool, default False If True, check that all the geom types are equal. check_crs: bool, default True If `check_frame_type` is True, then also check that the crs matches. normalize: bool, default False If True, normalize the geometries before comparing equality. Typically useful with ``check_less_precise=True``, which uses ``geom_equals_exact`` and requires exact coordinate order. r)assert_frame_equalassert_index_equalNz#GeoDataFrame shape mismatch, left: z , right: z. Left columns: z, right columns: zGeoDataFrame.columns)exactobj)r.r2r5r6r7r")excluder)r2r3check_column_typerN)pandas.testingrKrL ImportErrorpandas.util.testingr!rr)_geometry_column_namecolumnsr*shape reindex_likedtypesitemsrr9 select_dtypes)r0r1r2r3rPcheck_frame_type check_liker5r6r7r.rKrLcolr'left2right2s rassert_geodataframe_equalr`sNOI $ ---$U ,,, **2//7**$,,>//u}}DXXeiixx599,,,$ -%D%. 'E :: $ -djj^9U[[OT((9%--9J L $   ' emm+rgsy2-)#$,/$068L9^ Bqr