Homec4science
Diffusion pybliographer (master) Browse

R3596/tests/ut_format.py
1b495ebab959master

Last Change
Raw File

ut_format.py
View Options

This document is not UTF8. It was detected as Shift JIS and converted to UTF8 for display.
# -*- coding: latin-1 -*-
import os, pybut, sys
from Pyblio import Store, Attribute
from Pyblio.Format import Person, join, HTML, all, one, A, B, I, BR, DSL, switch, Misc, Pages, Text, i18n, record_key
class TestFormat (pybut.TestCase):
def setUp (self):
self.db = Store.get('file').dbopen(pybut.src('ut_format/sample.bip'))
self.rec = self.db[1]
self.rec ['journal'] = [ ]
return
def _cmp (self, v, s, **args):
stage2 = v(self.db, args)
stage3 = stage2(self.rec)
r = HTML.generate(stage3)
assert r == s, 'expected %s, got %s' % (
repr (s), repr (r))
def testAccessor (self):
""" Check that accessors behave properly """
# Failure modes are identical for both all and one
for f in (all, one):
f ('gronf')
try:
f ('jenexistepas')(self.db)
assert False
except KeyError: pass
try:
f ('journal.zoglu')(self.db)
assert False
except KeyError: pass
# Access modes
assert all ('title')(self.db)(self.rec) == self.rec['title']
assert all ('author')(self.db)(self.rec) == self.rec['author']
assert one ('title')(self.db)(self.rec) == self.rec['title'][0]
assert one ('author')(self.db)(self.rec) == self.rec['author'][0]
self.failUnlessEqual(all('nest.sub')(self.db)(self.rec), self.rec ['nest'] [0].q ['sub'])
self.failUnlessEqual(one('nest.sub')(self.db)(self.rec), self.rec ['nest'] [0].q ['sub'] [0])
def testRecordKey(self):
style = u'Record #' + record_key
self.failUnlessEqual(Text.generate(style(self.db)(self.rec)), u'Record #1')
return
def testOneAdd (self):
""" It is possible to add one fields together or with text """
v = one ('title') + ' ok'
self._cmp (v, 'My title ok')
v = 'ok ' + one ('title')
self._cmp (v, 'ok My title')
v = 'ok ' + one ('title') + ' ok'
self._cmp (v, 'ok My title ok')
# a missing field causes a delayed error
v = 'ok ' + one ('gronf') + ' ok'
try:
v(self.db)(self.rec)
assert False
except DSL.Missing: pass
return
def testAlternate (self):
""" When the left version fails, use the right version """
v = one ('title') | 'success'
self._cmp (v, 'My title')
v = one ('gronf') | 'success'
self._cmp (v, 'success')
v = one ('gronf') | one ('regronf') | 'success'
self._cmp (v, 'success')
return
def testJoin (self):
""" The join function takes lists of items and joins them """
v = join (', ') [ Person.lastFirst (all ('author')) ]
self._cmp (v, u'Gobry, Fr馘駻ic, Fobry, Gr馘駻ic, Dobry, Lr馘駻ic')
v = join (', ') [ 'a', 'b', 'c' ]
self._cmp (v, u'a, b, c')
v = join (', ', last = '; ') [ 'a', 'b', 'c' ]
self._cmp (v, u'a, b; c')
v = join (', ') [ 'a', 'b', 'c' ] + ' ok'
self._cmp (v, u'a, b, c ok')
# join skip missing values
v = join (', ') [ one ('title'), one ('journal'), one ('gronf') ]
self._cmp (v, u'My title')
# join fails when _no_ value is available
v = join (', ') [ one ('gronf'), one ('regronf') ]
phase2 = v(self.db)
try:
phase2(self.rec)
assert False
except DSL.Missing: pass
# Join with a weird tag in the middle.
v = 'a ' + join (BR) [ 'toto', 'tutu' ] + ' b'
self._cmp (v, u'a toto<br>tutu b')
return
def testMarkup (self):
v = I [ one ('title') ]
self._cmp (v, u'<i>My title</i>')
v = A (href = "http://pybliographer.org/") [ one ('title') ]
self._cmp (v, u'<a href="http://pybliographer.org/">My title</a>')
v = I [ 'simple' ]
self._cmp (v, u'<i>simple</i>')
v = 'a ' + I [ 'simple ', B [ one ('title') ] ]
self._cmp (v, u'a <i>simple <b>My title</b></i>')
v = 'a ' + BR + 'in the middle of a phrase'
self._cmp (v, u'a <br>in the middle of a phrase')
return
def testInitials (self):
self.failUnlessEqual(Person.initials(u'Fr馘駻ic'), 'F.')
self.failUnlessEqual(Person.initials(u'Jean-Pierre'), 'J.-P.')
self.failUnlessEqual(Person.initials(u'Jean Pierre'), 'J.P.')
self.failUnlessEqual(Person.initials(u'J.Pierre'), 'J.P.')
self.failUnlessEqual(Person.initials(u'JP'), 'J.P.')
self.failUnlessEqual(Person.initials(u'J'), 'J.')
self.failUnlessEqual(Person.initials(u'J.'), 'J.')
def testMissingPerson (self):
r = Store.Record()
r['author'] = [Attribute.Person (last = 'Gobry', first = u'Fr馘駻ic'),
Attribute.Person (last = 'Fobry') ]
def run(fn):
formatter = fn(all('author'))(self.db)
return formatter(r)
assert run(Person.initialLast) == ['F. Gobry', 'Fobry']
assert run(Person.firstLast) == [u'Fr馘駻ic Gobry', 'Fobry']
assert run(Person.lastFirst) == [u'Gobry, Fr馘駻ic', 'Fobry']
def testPages (self):
assert Pages.pagesLong(one ('singlepage'))(self.db)(self.rec) == u'page\xa0123'
assert Pages.pagesLong(one ('pagerange'))(self.db)(self.rec) == u'pages\xa0123-134'
def testMiscPlural (self):
zm = Misc.plural (all('gronf'),
zero='zero',
more='more')(self.db)
zom = Misc.plural (all('gronf'),
zero='zero',
one='one',
more='more')(self.db)
zotm = Misc.plural (all('gronf'),
zero='zero',
one='one',
two='two',
more='more')(self.db)
def run(format, vals):
r = Store.Record()
r['gronf'] = vals
return Text.generate(format(r))
self.failUnlessEqual(run(zm, []), 'zero')
self.failUnlessEqual(run(zm, [1]), 'more')
self.failUnlessEqual(run(zom, [1]), 'one')
self.failUnlessEqual(run(zom, [1, 2]), 'more')
self.failUnlessEqual(run(zotm, [1, 2]), 'two')
self.failUnlessEqual(run(zotm, [1, 2, 3]), 'more')
def testSwitch(self):
""" Test the 'switch' operator."""
def run(c, r):
f = c(self.db)
return Text.generate(f(r))
def txo(name):
return Attribute.Txo(self.db.schema.txo['type'].byname(name))
# One cannot use switch on a non-txo attribute
citation = switch('title').default(one('title'))
try:
citation(self.db)
assert False, 'should not be accepted'
except TypeError:
pass
# A switch fails when the value to switch on does not exist
# and there is no default case.
citation = switch('type').case(ARTICLE=one('title'))
try:
run(citation, self.rec)
assert False, 'should not succeed'
except DSL.Missing:
pass
# With a default case, it should pass
citation = switch('type').case(ARTICLE=one('singlepage'))
citation = citation.default(one('title'))
self.failUnlessEqual(run(citation, self.rec), 'My title')
# Test with the actual value
self.rec['type'] = [txo('ARTICLE')]
self.failUnlessEqual(run(citation, self.rec), '123')
# Another value will also return the default
self.rec['type'] = [txo('BOOK')]
self.failUnlessEqual(run(citation, self.rec), 'My title')
return
def testI18n(self):
cite = i18n(fr=u'titre ' + one('title'),
en=u'title ' + one('title'),
default = u'default ' + one('title'))
r = []
for ln in ('en', 'fr', 'it'):
f = cite(self.db, props={'ln': ln})
r.append(Text.generate(f(self.rec)))
self.failUnlessEqual(r, [
u'title My title',
u'titre My title',
u'default My title'
])
class TestOutput (pybut.TestCase):
def setUp (self):
self.db = Store.get('file').dbopen(pybut.src('ut_format/sample.bip'))
self.rec = self.db[1]
self.rec ['journal'] = [ ]
self.rec ['title'] = [ Attribute.Text (u'My < title &') ]
return
class TestOutputHTML (TestOutput):
def _cmp (self, v, s):
stage2 = v(self.db)
stage3 = stage2(self.rec)
r = HTML.generate (stage3)
assert r == s, 'expected %s, got %s' % (
repr (s), repr (r))
def testEscape (self):
v = one ('title') + ' ok &'
self._cmp (v, 'My &lt; title &amp; ok &amp;')
class TestOutputText (TestOutput):
def _cmp (self, v, s):
stage2 = v(self.db)
stage3 = stage2(self.rec)
r = Text.generate(stage3)
assert r == s, 'expected %s, got %s' % (
repr (s), repr (r))
def testEscape (self):
v = one ('title') + ' ok &'
self._cmp (v, 'My < title & ok &')
suite = pybut.suite (TestFormat, TestOutputHTML, TestOutputText)
if __name__ == '__main__': pybut.run (suite)
c4science · Help