o #e@s"dgZGdddeZddZdS)Originc@s^eZdZdZddZeddZddZdd Zd d Z d d Z dddZ ddZ ddZ dS)raGThis represents the origin of some object in some string. For example, if we have an object ``x1_obj`` that was produced by parsing the ``x1`` in the formula ``"y ~ x1:x2"``, then we conventionally keep track of that relationship by doing:: x1_obj.origin = Origin("y ~ x1:x2", 4, 6) Then later if we run into a problem, we can do:: raise PatsyError("invalid factor", x1_obj) and we'll produce a nice error message like:: PatsyError: invalid factor y ~ x1:x2 ^^ Origins are compared by value, and hashable. cCs||_||_||_dSNcodestartend)selfrrrr,lib/python3.10/site-packages/patsy/origin.py__init__#s zOrigin.__init__cCsg}|D]}|durt|ts|j}|durq||q|s!dStdd|D}t|dks2Jtdd|D}tdd|D}||||S)a!Class method for combining a set of Origins into one large Origin that spans them. Example usage: if we wanted to represent the origin of the "x1:x2" term, we could do ``Origin.combine([x1_obj, x2_obj])``. Single argument is an iterable, and each element in the iterable should be either: * An Origin object * ``None`` * An object that has a ``.origin`` attribute which fulfills the above criteria. Returns either an Origin object, or None. NcSg|]}|jqSr)r.0orrr Cz"Origin.combine..cSr r)rr rrr rErcSr r)rr rrr rFr) isinstanceroriginappendsetlenminmaxpop)clsZ origin_objsZoriginsobjZcodesrrrrr combine(s zOrigin.combinecCs|j|j|jS)zgExtracts and returns the span of the original code represented by this Origin. Example: ``x1``.rrrrr relevant_codeIszOrigin.relevant_codecCs.t|to|j|jko|j|jko|j|jkSr)rrrrrrotherrrr __eq__Ns    z Origin.__eq__cCs ||k Srrrrrr __ne__T z Origin.__ne__cCstt|j|j|jfSr)hashrrrrrrrr __hash__WszOrigin.__hash__cCs.dd||jd|d|jd|j|jfS)a3Produces a user-readable two line string indicating the origin of some code. Example:: y ~ x1:x2 ^^ If optional argument 'indent' is given, then both lines will be indented by this much. The returned string does not have a trailing newline. z %s%s %s%s%s ^r)rindentrrr caretizeZs zOrigin.caretizecCs<d|jd|j|j|j|j|j|jd|j|jfS)Nz%s<-%s (%s-%s)>rrrrr __repr__ls zOrigin.__repr__cCstr)NotImplementedErrorrrrr __getstate__tszOrigin.__getstate__N)r&)__name__ __module__ __qualname____doc__r classmethodrrr!r"r%r*r+r-rrrr r s    cCs"tddd}tddd}|dksJ|dksJt||g}|jdks*J|jdks1J|jdks8J|jdddksBJ|tdddksLJGd d d t}t|||dg}||kseJt|||g}||kstJt||gdusJd d lm}|td d d dS)NZ012345z 012345 ^^z 012345 ^)r)z 012345 ^^^c@seZdZdddZdS)z"test_Origin..ObjWithOriginNcSs ||_dSr)r)rrrrr r r#z+test_Origin..ObjWithOrigin.__init__r)r.r/r0r rrrr ObjWithOriginsr6r&)assert_no_pickling) rr*rrrrobjectZ patsy.utilr7)Zo1Zo2Zo3r6Zo4Zo5r7rrr test_Originws$     r:N)__all__r9rr:rrrr s  j