o DDiQt@sddlmZddlZddlmZddlmZddlmZddl m Z ddl m Z Gdd d eZ Gd d d eZGd d d eZGdddeZdS))AnyN)parse_statement) CSTNodeTest) ensure_type) CodeRange) data_providerc@s\eZdZeejeedfdddejeeededfdddejeeededfdddejeeededeeededfdde d d d ejejeedede edd fdd dejejeedede edd ejeedede edd fdddejejeedede edd eeededeeededejede edd fdddejejejededej e de dddej ede de ddej e de dddejedej ede de ddd fe ddde d dd fdeddfd d!Zed"d#d$d%d&d#d'd%d(d#d'd%d)d#d'd%d*d#d+d%d,d#d-d%d.d#d-d%d/d#d0d%fdeddfd1d2ZdS)3ImportCreateTestfoonames import foonodecodebarimport foo.barbazimport foo.bar, foo.bazrrrrexpected_positionasnameimport foo.bar as baz%import foo.bar as baz, foo.baz as barinstagramunittestut:import foo.bar as baz, insta.gram, foo.baz, unittest as ut whitespace_beforewhitespace_afterdot whitespace_before_aswhitespace_after_asrcommar whitespace_after_import.import foo . bar as baz , unittest as ut)r.kwargsreturnNcK|jdi|dSN validate_nodeselfr3r7r7X/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/_nodes/tests/test_import.py test_validszImportCreateTest.test_validcCs tjddS)Nr7r )cstImportr7r7r7r<s zImportCreateTest.at least one ImportAliasget_node expected_recCstjttdfdS)Nr )r>r? ImportAliasNamer7r7r7r<r@szempty name identifierc C(tjtttdtdfdS)NrEblar r>r?rF AttributerGr7r7r7r<r@c CrH)NrIrEr rJr7r7r7r<r@rLcCs0tjtjttdtdtdfdS)Nr rr.r )r>r?rFrKrGCommar7r7r7r<r@sztrailing commac Cs0tjtttdtdftddS)Nr rrEr/)r>r?rFrKrGSimpleWhitespacer7r7r7r<r@szat least one spacecCs4tjtjtdtjtdtdddfdS)Nr rrEr+rr )r>r?rFrGAsNamerOr7r7r7r<r@scCs0tjtjtjtjddtjddddgdS)Nz0.)valueA)rRattr)namer )r>r?rFrKFloatrGr7r7r7r<r@sz-imported name must be a valid qualified name.cKr5r6assert_invalidr:r7r7r< test_invalidsMzImportCreateTest.test_invalid)__name__ __module__ __qualname__rr>r?rFrGrKrrQDotrOrNrr=rYr7r7r7r<rs            Lrc@s(eZdZeejeedfdddejeeededfdddejeeededfdddejejeededej e ddd eeeded fdd dejejeedede ed d fdd dejejeedede ed ej e dddejeeded e edd fdddejejeedede ed ej e dddejeededej e ddd ejeeded ej e ddd ejede edd fdddejejejededej e de dddej ed e de ddej e de dddejedej ede de ddd fe ddddfde ddfddZdS) ImportParseTestr r r r rrr#r&rMrrrrr-rrrr r!r"r$r'r)r*r/r1r3r4NcK|jddddi|dS)NparsercStt|tjjdSNrrrr>SimpleStatementLinebodyrr7r7r<r@n  z,ImportParseTest.test_valid..r7r8r:r7r7r<r=s  zImportParseTest.test_valid)rZr[r\rr>r?rFrGrKrNrOrQr]rr=r7r7r7r<r^s      r^c@seZdZeejedeedfdddejedejedeeddfdddejedeedeedfdd dejedejede d ejede d fdd e d d dejede dde d ddeje fedeedfdddeje e fedeedfdddeje e fdeedfdddejede ejedeeddfedde d ddejej ededdej ededdfedej eddejedejedededdej ededddejedejed ededddfejedd!edededd"d#e d d$df d%ed&dfd'd(Zed)d*d+d,d-d*d.d,d/d*d0d,d1d*d2d,d3d*d4d,d5d*d4d,d6d*d7d,d8d*d9d,d:d*d;d,dd?ZdS)@ImportFromCreateTestr rmoduler from foo import barr rrfrom foo import bar as bazfrom foo import bar, bazrMzfrom foo import bar,baz,rrrfrom foo import *)rrelativerkr from .foo import barfrom ..foo import barNfrom .. import barrklparr rparfrom foo import (bar as baz))rr#r$r_r)r*r-r r!r%rrrkrwr rxwhitespace_after_fromwhitespace_before_importr0=from . . foo import ( bar as baz , unittest as ut ))r=r3r4cKr5r6r8r:r7r7r<r=vs zImportFromCreateTest.test_validcCstjdttdfdS)Nrrj)r> ImportFromrFrGr7r7r7r<r@szImportFromCreateTest.zMust have a module specifiedrBcCstjtdddS)Nr r7rj)r>rrGr7r7r7r<r@ srAcC(tjtdttdftdS)Nr rrkr rw)r>rrGrF LeftParenr7r7r7r<r@ zleft paren without right parencCr)Nr rrkr rx)r>rrGrF RightParenr7r7r7r<r@rzright paren without left parencCtjtdttdS)Nr r)r>rrG ImportStarrr7r7r7r<r@ szcannot have parenscCr)Nr r)r>rrGrrr7r7r7r<r@&s cC*tjtdttdftddS)Nr rrE)rkr r}r>rrGrFrOr7r7r7r<r@. zone space after fromcCr)Nr rrE)rkr r~rr7r7r7r<r@6rzone space before importcCr)Nr rrE)rkr r0rr7r7r7r<r@>rzone space after importc CsNtjtdtjtdtjtjdtftfdtdddfdS) Nr rr)rwrxrErPrrj)r>rrGrFrQrrrOr7r7r7r<r@Fszone space before as keywordcKr5r6rWr:r7r7r<rYsXz!ImportFromCreateTest.test_invalid)rZr[r\rr>rrGrFrQrNrrr]rrrOrr=rYr7r7r7r<rius*         (       Wric@s eZdZeejedeedfdddejedejedeeddfdddejedejedej e d d d eedfdd dejedejedej e d d d ejede d fe e d ddejede dddejefedeedfdddejeefedeedfdddejeefdeedfdddejede ejedeeddfe dddejeje de ddeje de d dfedej e d d ejedejede de ddej e d e dddejedejede de dddfej e d de de de dd d!df d"ed#dfd$d%ZdS)&ImportFromParseTestr rrjrlr rrrmr#r_rMrn)rkr rwrxzfrom foo import (bar, baz,)rorqrsrtNrurvryrEr)r$r*r-r r!r{z r|rr3r4cKr`)NracSrbrcrdrgr7r7r<r@rhz0ImportFromParseTest.test_valid..r7r8r:r7r7r<r=_s  zImportFromParseTest.test_valid)rZr[r\rr>rrGrFrQrNrOrrrr]rr=r7r7r7r<r^s         (r)typingrlibcstr>rlibcst._nodes.tests.baserlibcst.helpersrlibcst.metadatarlibcst.testing.utilsrrr^rirr7r7r7r<s      Wj