:Oc@@sddlmZddlmZddlmZmZmZddlm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZddlmZdd lmZmZmZmZmZmZmZd efd YZ d S( i(tabsolute_import(tdate(t GEOSGeometrytPointt MultiPoint(tCollecttCounttExtenttFtUnion(tGeometry(tmysqltoracletno_mysqlt no_oraclet no_spatialite(tTestCasei(tCitytLocationtDirectoryEntrytParceltBooktAuthortArticletRelatedGeoModelTestcB@seZdZedZeedZedZdZdZ dZ dZ dZ d Z d Zed Zd Zd ZedZeeedZdZdZRS(c C@stjj}tjj}tjjd}dddf}x|||fD]}}xtt||D]c\}}|\}} } } |j||j|j| |j|jt| | |j j qeWqOWdS(s:Testing `select_related` on geographic models (see #7126).tlocationtAuroratTXgƆnaXgtw@@tRoswelltNMg9d!ZgW%}@@t KecksburgtPAgu|Sg67D@N(RRgƆnaXgtw@@(RRg9d!ZgW%}@@(RRgu|Sg67D@( Rtobjectstalltselect_relatedtzipt assertEqualtnametstateRRtpoint( tselftqs1tqs2tqs3tcitiestqstreftctnmtsttlontlat((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest02_select_relateds c@sdfd}dddf}xa|D]Y\}}}ttjjd |j|d d}|t|||djjq.WdS(sHTesting the `transform` GeoQuerySet method on related geographic models.ic@sLj|j|jj|j|jj|j|jdS(N(tassertAlmostEqualtxtyR$tsrid(R.tpnt(R(ttol(sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyt check_pnt+sRis(POINT(1490553.98959621 314792.131023984)Ris(POINT(481902.189077221 868477.766629735)Ris'POINT(2269923.2484839 7069381.28722222)R%t field_nametlocation__pointN(s Kecksburgis(POINT(1490553.98959621 314792.131023984)(sRoswellis(POINT(481902.189077221 868477.766629735)(sAurorais'POINT(2269923.2484839 7069381.28722222)(tlistRR tfiltert transformRRR'(R(R;t transformedR%R8twktR-((R(R:sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest03_transform_related$s *c C@stjjtd}d }d}tjjdd}tjjdd jdd}|d }d }xb||f||f||fgD]?\}} x0t|| D]\} } |j| | |qWqWd S(sITesting the `extent` GeoQuerySet aggregates on related geographic models.R=g9d!Zg|zl=@gu|Sg67D@gƆnaXR<R&Rtlocation__point__extentiN(g9d!Zg|zl=@gu|Sg67D@(gƆnaXg|zl=@gu|Sg67D@(RR t aggregateRtextenttexcludeR#R5( R(taggst all_extentt txpa_extentte1te2te3R:R.tetref_valte_val((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyt test04a_related_extent_aggregate<s! .c C@sFtjjtd}tdd}tdd}tdd}tdd }td d }trt|||||d d }t||d d }n3t|||||d d }t||d d }tjjdd} tjjddjdd} |d} |j || |j || |j || dS(sKTesting the `unionagg` GeoQuerySet aggregates on related geographic models.R=g9d!ZgW%}@@gƆnaXgtw@@gu|Sg67D@gIG9M3XgMqd@@g =Wg|zl=@R8iR<tname__inRtHoustontDallass Fort Worthtlocation__point__unionN(sRoswellRSRTs Fort Worth( RR RER RR RtunionaggRGR$( R(RHtp1tp2tp3tp4tp5tref_u1tref_u2tu1tu2tu3((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest04b_related_union_aggregatePs"! cC@sttjjj}dS(s`Testing that calling select_related on a query over a model with an FK to a model subclass worksN(R>RR R!R"(R(tl((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyt$test05_select_related_fk_to_subclassssc C@stddd}tjjdd}|jj}|jddt}|jd}t jj dd d |d |d |d |d|}|j }|jddt}t jj ddd |d |d |d |d|}t jj dt d }|jdt||jd|djtstt jj dt d }|jdt||jd|djnt jj d t d}|jdt||jd |djts t jj dt d}|jdt||jd |djndS(s*Testing F() expressions on GeometryFields.ssPOLYGON((-97.501205 33.052520,-97.501205 33.052576,-97.501150 33.052576,-97.501150 33.052520,-97.501205 33.052520))R8iR%RitcloneidtP1tcitytcenter1tcenter2tborder1tborder2tP2tcenter1__withiniitcenter2__withintcity__location__pointtborder2__containsN(RRR tgetRR'R@tTruetbufferRtcreatetcentroidR?RR$tlenR%R ( R(tb1tpcitytc1tc2tb2RWRXR-((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest06_f_expressionsxs. 3 3cC@stjj}tjj}tjj}xt|||D]q\}}}|jt|dt|jt|dt|j |j |d|j |j |dq@WdS(s4Testing values() and values_list() and GeoQuerySets.R'iN( RR R!tvaluest values_listR#t assertTruet isinstanceR R$R'(R(tgqstgvqstgvlqstmtdtt((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyt test07_valuess"cC@s[tjj}tjjd}x3t||D]"\}}|j|j|jq1WdS(s0Testing defer() and only() on Geographic models.R'N(RR R!tdeferR#R$R'(R(R-tdef_qstloctdef_loc((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest08_defer_onlyscC@s|d }d }tjjdjdd}xKt|||D]7\}}}|j|d||j|d|q=WdS( sMEnsuring correct primary key column is selected across relations. See #10757.iiiiitidt location__idN(iiiii(iiiii(RR torder_byR|R#R$(R(tcity_idstloc_idstids_qstval_dicttc_idtl_id((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest09_pk_relationss "cC@stjjddjjjd}tjjddjjjd}tjjd|}tjjd|}||B}g|D]}|j^q}|jdt ||j d|k|j d|kdS(s9Testing the combination of two GeoQuerySets. See #10807.R%Rg?Rtlocation__point__withiniN( RR RpRR'RrR?R%R$RuR~(R(tbuf1tbuf2R)R*tcombinedR/tnames((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest10_combines$$ cC@scddl}ddlm}tjj}|j|j}|j|}|j ||j dS(s?Ensuring GeoQuery objects are unpickled correctly. See #10839.iN(tGeoQuery( tpicklet django.contrib.gis.db.models.sqlRRR R!tdumpstquerytloadsR$t __class__(R(RRR-tq_strtq((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest11_geoquery_pickles  cC@sYtjjdd}tjjdtdjd|jj}|jd|j dS(sBTesting `Count` aggregate use with the `GeoManager` on geo-fields.R%RTt num_citiesRfRiN( RR RpRtannotateRRRR$R(R(tdallasR((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyt test12a_counts-cC@stjjdtdjdd}tjjdjdtdjdd}|jdt||jd|dj|jdt||jd|dddS( sRTesting `Count` aggregate use with the `GeoManager` on non geo-fields. See #11087.t num_bookstbookst num_books__gtiR%iiN( RR RRR?R|R$RuR(R(R-tvqs((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyt test12b_counts '0cC@stjjddjdtdjddd}|jdt||jd|dd|jt |ddt d S( s8Testing `Count` aggregate with `.values()`. See #15305.RiRRfR'iiiN( RR R?RRR|R$RuR~RR(R(R-((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyt test13c_counts6cC@sJtjjdd}tjjdjdd}|jd|jdS(sOTesting `select_related` on a nullable ForeignKey via `GeoManager`. See #11381.ttitlesWithout AuthortauthorN(RR RsR"RpR$tNoneR(R(t no_authortb((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest13_select_related_null_fkscC@std}tjjddjdd}tjjddjtdd}x:||fD],}|jdt||j||qbWdS( sATesting the `collect` GeoQuerySet method and `Collect` aggregate.s_MULTIPOINT(-97.516111 33.058333,-96.801611 32.782057,-95.363151 29.763374,-96.801611 32.782057)R&RR<R=tlocation__point__collectiN( RRR R?tcollectRERR$Ru(R(tref_geomRxRytcoll((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest14_collect s !(cC@s%tjjd}t|j}dS(sTTesting doing select_related on the related name manager of a unique FK. See #13934.tauthor__articleN(RR R"tstrR(R(R-tsql((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest15_invalid_select_relatedscC@sggttjjdtdjddD]}|j^q.}|j|jddg|dS(sMEnsure annotated date querysets work if spatial backend is used. See #14648.RRtdobtyeariiN( R>RR RRtdatesRtsortR$(R(tdtt birth_years((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyttest16_annotated_date_queryset's@ (t__name__t __module__R4R RCRRQRaRcR{RRRRRRRRRRRRR(((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyRs* #  0     N(!t __future__RtdatetimeRtdjango.contrib.gis.geosRRRtdjango.contrib.gis.db.modelsRRRRR t#django.contrib.gis.geometry.backendR tdjango.contrib.gis.tests.utilsR R R RRt django.testRtmodelsRRRRRRRR(((sB../Django//lib/python/django/contrib/gis/tests/relatedapp/tests.pyts((4