a ai@shdZddlmZddlZddlZddlmZddlZddlZGdddejj Z Gdddejj Z dS) zTests for samba.graph)print_functionN)graphc@s4eZdZddZddZddZddZd d Zd S) DotFileTestscCs*t||}|dur&|d||f|S)Nz%r did not match /%s/)rematchZfail)selfexpsmr 3/usr/lib/python3/dist-packages/samba/tests/graph.py assertMatch#s zDotFileTests.assertMatchcCs||dd|rd}nd}|||d|d|d}||d||d|d |d |d ||d d ||dddS)Nrz/* generated by samba */z^digraph \w+ {$z ^graph \w+ {$z^label="([\w ]+)";$z^fontsize=10;$$z&node[fontname=Helvetica; fontsize=10];) assertEqualr group)rlinestitledirectedrr r r r assertHeader)szDotFileTests.assertHeadercCs6t||D]&\}}|d|}|||dq dS)Nz ^"(\w+)";$rzipr rr)rrnamesnliner r r r assertVertices7s zDotFileTests.assertVerticesc Cs|rdnd}t||D]n\}}|\}}|d|} || d||| d||| d|| dr|d| dqdS) Nz->z--z.^"(\w+)" ([>-]{2}) "(\w+)" ?(?:\[([^\]])\])?;$rrrrz^[\w ]*$r) rredgesrZ connectorZedger abr r r r assertEdges<s  zDotFileTests.assertEdgescCstd}tt|d}tt|dd|dd}||d|dfg}g}dd|D}d|fd |fd |fd |fd |ffD]~\}}d D]p\} } d|| f} tj||| | d} | d} || | | | | dd|| | t |dd|| qqdS)NZabcdefghrrrcSs,g|]$\}}t|t|Adkr||fqS)r)ord).0r#r$r r r Qs z5DotFileTests.test_basic_dot_files..allr ZringZnoZdisjoint))Tr)FZ undirectedz%s %s)rr ) tuple itertools combinationslistrrZ dot_graphsplitrr!r%len)rZverticesZ all_edgesZ line_edgesZ ring_edgesZno_edgesZdisjoint_edgesnamer"rtagrgrr r r test_basic_dot_filesJs,    z!DotFileTests.test_basic_dot_filesN)__name__ __module__ __qualname__r rr!r%r6r r r r r!s rcs4eZdZfddZddZddZddZZS) DistanceTestscs&tt|ttjddd|_dS)NcSs|pdS)Nrr )r3r r r oz%DistanceTests.setUp..)key)superr:setUpsortedrZ COLOUR_SETSsorted_colour_sets)r __class__r r r?is zDistanceTests.setUpc Csxgd}dddddddd d d d }d D]L}|jD]@}d||f}tjd|||d}|j|||dd||fdq0q&dS)N) antbatcatdog)rEelephant)rJrI)rFrI)frogrJ)rKrH)rFrJ)rJrH)rHrErGu[ destination ╭────── ant │╭───── bat ││╭──── cat │││╭─── dog ││││╭── elephant source │││││╭─ frog ant ·1221- bat 3·211- cat 12·12- dog ---·-- elephant 2311·- frog 23121·u destination ╭────── ant │╭───── bat ││╭──── cat │││╭─── dog ││││╭── elephant source │││││╭─ frog  ant ·1221-  bat 3·211-  cat 12·12-  dog ---·-- elephant 2311·-  frog 23121· u destination ╭────── ant │╭───── bat ││╭──── cat │││╭─── dog ││││╭── elephant source │││││╭─ frog  ant ·1221-  bat 3·211-  cat 12·12-  dog ---·-- elephant 2311·-  frog 23121· u destination ╭────── ant │╭───── bat ││╭──── cat │││╭─── dog ││││╭── elephant source │││││╭─ frog  ant ·1221-  bat 3·211-  cat 12·12-  dog ---·-- elephant 2311·-  frog 23121· u destination ╭────── ant │╭───── bat ││╭──── cat │││╭─── dog ││││╭── elephant source │││││╭─ frog  ant ·1221-  bat 3·211-  cat 12·12-  dog ---·-- elephant 2311·-  frog 23121· a destination ,------ ant |,----- bat ||,---- cat |||,--- dog ||||,-- elephant source |||||,- frog ant 01221- bat 30211- cat 12012- dog ---0-- elephant 23110- frog 231210 au destination ,------ ant |,----- bat ||,---- cat |||,--- dog ||||,-- elephant source |||||,- frog  ant 01221-  bat 30211-  cat 12012-  dog ---0-- elephant 23110-  frog 231210 au destination ,------ ant |,----- bat ||,---- cat |||,--- dog ||||,-- elephant source |||||,- frog  ant 01221-  bat 30211-  cat 12012-  dog ---0-- elephant 23110-  frog 231210 a destination ,------ ant |,----- bat ||,---- cat |||,--- dog ||||,-- elephant source |||||,- frog  ant 01221-  bat 30211-  cat 12012-  dog ---0-- elephant 23110-  frog 231210 a destination ,------ ant |,----- bat ||,---- cat |||,--- dog ||||,-- elephant source |||||,- frog  ant 01221-  bat 30211-  cat 12012-  dog ---0-- elephant 23110-  frog 231210 zutf8 True, colour Nonezutf8 True, colour ansizutf8 True, colour ansi-heatmapz utf8 True, colour xterm-256colorz(utf8 True, colour xterm-256color-heatmapzutf8 False, colour Nonezutf8 False, colour ansizutf8 False, colour ansi-heatmapz!utf8 False, colour xterm-256colorz)utf8 False, colour xterm-256color-heatmapTFutf8 %s, colour %sutf8colourTWrong output: %s %sstripmsgrArZdistance_matrixZassertStringsEqualrr"ZexpectedrPrQkr r r r test_simple_distanceqs.     z"DistanceTests.test_simple_distancec Csxgd}dddddddd d d d }d D]L}|jD]@}d||f}tjd|||d}|j|||dd||fdq0q&dS)N)rD)rHrF)rFrE)rErHu destination ╭─── ant │╭── bat source ││╭─ cat ant ·11 bat 1·2 cat 21· uk destination ╭─── ant │╭── bat source ││╭─ cat  ant ·11  bat 1·2  cat 21· ug destination ╭─── ant │╭── bat source ││╭─ cat  ant ·11  bat 1·2  cat 21· u destination ╭─── ant │╭── bat source ││╭─ cat  ant ·11  bat 1·2  cat 21· u destination ╭─── ant │╭── bat source ││╭─ cat  ant ·11  bat 1·2  cat 21· zj destination ,--- ant |,-- bat source ||,- cat ant 011 bat 102 cat 210 aD destination ,--- ant |,-- bat source ||,- cat  ant 011  bat 102  cat 210 aD destination ,--- ant |,-- bat source ||,- cat  ant 011  bat 102  cat 210 a destination ,--- ant |,-- bat source ||,- cat  ant 011  bat 102  cat 210 a destination ,--- ant |,-- bat source ||,- cat  ant 011  bat 102  cat 210 rLrMrNrOTrRrSrVrWr r r test_simple_distance2s*         \    z#DistanceTests.test_simple_distance2c Csxgd}dddddddd d d d }d D]L}|jD]@}d||f}tjd|||d}|j|||dd||fdq0q&dS)N)rD)rFrHrG)rIrE)rIZeelu destination ╭───── ant │╭──── bat ││╭─── cat │││╭── dog source ││││╭─ eel ant ·1234 bat 3·123 cat 23·12 dog 123·1 eel ----· u destination ╭───── ant │╭──── bat ││╭─── cat │││╭── dog source ││││╭─ eel  ant ·1234  bat 3·123  cat 23·12  dog 123·1  eel ----· u destination ╭───── ant │╭──── bat ││╭─── cat │││╭── dog source ││││╭─ eel  ant ·1234  bat 3·123  cat 23·12  dog 123·1  eel ----· u destination ╭───── ant │╭──── bat ││╭─── cat │││╭── dog source ││││╭─ eel  ant ·1234  bat 3·123  cat 23·12  dog 123·1  eel ----· u destination ╭───── ant │╭──── bat ││╭─── cat │││╭── dog source ││││╭─ eel  ant ·1234  bat 3·123  cat 23·12  dog 123·1  eel ----· z destination ,----- ant |,---- bat ||,--- cat |||,-- dog source ||||,- eel ant 01234 bat 30123 cat 23012 dog 12301 eel ----0 a destination ,----- ant |,---- bat ||,--- cat |||,-- dog source ||||,- eel  ant 01234  bat 30123  cat 23012  dog 12301  eel ----0 a destination ,----- ant |,---- bat ||,--- cat |||,-- dog source ||||,- eel  ant 01234  bat 30123  cat 23012  dog 12301  eel ----0 aS destination ,----- ant |,---- bat ||,--- cat |||,-- dog source ||||,- eel  ant 01234  bat 30123  cat 23012  dog 12301  eel ----0 am destination ,----- ant |,---- bat ||,--- cat |||,-- dog source ||||,- eel  ant 01234  bat 30123  cat 23012  dog 12301  eel ----0 rLrMrNrOTrRrSrVrWr r r test_simple_distance3s,            z#DistanceTests.test_simple_distance3)r7r8r9r?rYrZr[ __classcell__r r rBr r:gs  -ir:) __doc__Z __future__rZsambaZ samba.testsrrr.ZtestsZTestCaseInTempDirrZTestCaser:r r r r s  F