o
a @ s d Z ddlmZ ddlmZ ddlZddlZddlZddlZddl m
Z ddl m
Z
ddlmZ dd Zd
d ZG d d
d
ZG dd deZdddZd ddZG dd deZG dd de Z
G dd de
ZdS ) z Helpers for writing unit tests. )Iterable)BytesION)TestCasetobytes) XMLWriterc C s t }d}t| tr|| 7 }n(t| tr |t| d7 }nt| tr.|ddd | D 7 }n tdt| j |d7 }|j
|d |j d S )
aA Parses a snippet of XML.
Input can be either a single string (unicode or UTF-8 bytes), or a
a sequence of strings.
The result is in the same format that would be returned by
XMLReader, but the parser imposes no constraints on the root
element so it can be called on small snippets of TTX files.
s utf-8 c s s | ]}t |d V qdS )r Nr ).0s r :/usr/lib/python3/dist-packages/fontTools/misc/testTools.py s zparseXML..z0expected string or sequence of strings; found %rs r )
TestXMLReader_
isinstancebytesstrr r join TypeErrortype__name__parserZParseroot)
xmlSnippetreaderxmlr r r
parseXML s
r c C sB dd t | D }|D ]
\}}}||||| q
| |S )Nc S s g | ] }t |ts|qS r )r r )r
er r r
* s z parseXmlInto..)r stripZfromXMLZpopulateDefaults)ZfontZ parseIntor Z
parsed_xmlnameattrsZcontentr r r
parseXmlInto) s
r# c @ sf e Zd Zdd Zdd Zdd Zddd Zd
d Zd d
Zdd Z dd Z
dd Z dd Z dd Z
dS )FakeFontc C s* || _ dd t|D | _d| _i | _d S )Nc S s i | ]\}}||qS r r )r
igr r r
4 z%FakeFont.__init__..F) glyphOrder_ enumeratereverseGlyphOrderDict_lazytables)selfZglyphsr r r
__init__2 s
zFakeFont.__init__c C
| j | S Nr- )r. tagr r r
__getitem__8
zFakeFont.__getitem__c C s || j |< d S r1 r2 )r. r3 tabler r r
__setitem__; zFakeFont.__setitem__Nc C s | j ||S r1 )r- get)r. r3 defaultr r r
r9 > r8 z FakeFont.getc C r0 r1 r+ r. r! r r r
getGlyphIDA r5 zFakeFont.getGlyphIDc fdd|D S )Nc g | ]} |qS r )r= r
gidr. r r
r E r( z+FakeFont.getGlyphIDMany..r r. Zlstr rB r
getGlyphIDManyD zFakeFont.getGlyphIDManyc C s |t | jk r | j| S d| S )Nz glyph%.5d)lenr) )r. ZglyphIDr r r
getGlyphNameG s
zFakeFont.getGlyphNamec r> )Nc r? r )rG r@ rB r r
r M r( z-FakeFont.getGlyphNameMany..r rC r rB r
getGlyphNameManyL rE zFakeFont.getGlyphNameManyc C | j S r1 )r) rB r r r
getGlyphOrderO zFakeFont.getGlyphOrderc C rI r1 r; rB r r r
getReverseGlyphMapR rK zFakeFont.getReverseGlyphMapc C s t | S r1 )sortedrJ rB r r r
getGlyphNamesU s zFakeFont.getGlyphNamesr1 )r
__module__ __qualname__r/ r4 r7 r9 r= rD rG rH rJ rL rN r r r r
r$ 1 s
r$ c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
r c C sB ddl m} | | _| j| j_| j| j_| j| j_d | _ g | _
d S )Nr ) ParserCreate) Zxml.parsers.expatrQ r
startElement_ZStartElementHandler endElement_ZEndElementHandleraddCharacterData_ZCharacterDataHandlerr stack)r. rQ r r r
r/ Z s
zTestXMLReader_.__init__c C s< ||g f}| j r| j d d | n|| _| j | d S Nr )rU appendr )r. r! r" elementr r r
rR c s
z TestXMLReader_.startElement_c C s | j d S r1 )rU popr< r r r
rS k r8 zTestXMLReader_.endElement_c C s | j d d | d S rV )rU rX )r. datar r r
rT n s z TestXMLReader_.addCharacterData_N)r rO rP r/ rR rS rT r r r r
r Y s
r
c C s( t t | d}|jd |j |S )N)
newlinestrr )r r fileseektruncate)r] writerr r r
makeXMLWriterr s
rb c C s6 t }| || |j d}|dsJ | S )zCall the passed toXML function and return the written content as a
list of lines (unicode strings).
Result is stripped of XML declaration and OS-specific newline characters.
r r\ )rb r^ getvaluedecodeendswith
splitlines)funcZttFontra r r r r
getXML{ s
rh c @ s8 e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd S )
MockFontzYA font-like object that automatically adds any looked up glyphname
to its glyphOrder.c s4 dg _ G fdddt}|ddi _d _d S )Nz.notdefc s e Zd Z fddZdS )z)MockFont.__init__..AllocatingDictc s j | t| }|| |< |S r1 ) _glyphOrderrX rF )Z reverseDictkeyrA rB r r
__missing__ s z5MockFont.__init__..AllocatingDict.__missing__N)r rO rP rl r rB r r
AllocatingDict s rm r F)rj dict_reverseGlyphOrderr, )r. rm r rB r
r/ s
zMockFont.__init__c C s | j | }|S r1 ro )r. ZglyphrA r r r
r= s
zMockFont.getGlyphIDc C rI r1 rp rB r r r
rL rK zMockFont.getReverseGlyphMapc C r0 r1 rj )r. rA r r r
rG r5 zMockFont.getGlyphNamec C rI r1 rq rB r r r
rJ rK zMockFont.getGlyphOrderN) r rO rP __doc__r/ r= rL rG rJ r r r r
ri s ri c @ s e Zd Zdd ZdS )r c C s& t | | t| ds| j| _d S d S )NassertRaisesRegex) _TestCaser/ hasattrZassertRaisesRegexprs )r. Z
methodNamer r r
r/ s
zTestCase.__init__N)r rO rP r/ r r r r
r s r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S ) DataFilesHandlerc C s d | _ d| _d S )Nr )tempdirZ
num_tempfilesrB r r r
setUp s
zDataFilesHandler.setUpc C s | j r t| j d S d S r1 )rw shutilrmtreerB r r r
tearDown s zDataFilesHandler.tearDownc C s& t jtj| j j}t j|d|S )Nr[ )ospathdirnamesysmodulesrO __file__r )r. ZtestfileZfolderr r r
getpath s zDataFilesHandler.getpathc C s | j s
t | _ d S d S r1 )rw tempfileZmkdtemprB r r r
temp_dir s zDataFilesHandler.temp_dirc C s( | tj| j|}t|| |S r1 )r r| r} r rw ry copy2)r. Z font_path file_nameZtemppathr r r
temp_font s zDataFilesHandler.temp_fontN)r rO rP rx r{ r r r r r r r
rv s rv )r\ r1 )rr Zcollections.abcr ior r| ry r r Zunittestr rt ZfontTools.misc.textToolsr ZfontTools.misc.xmlWriterr r r# r$ objectr rb rh ri rv r r r r
s& (