tests/_test_make.py

   1 #!/usr/bin/python
   2
   3 import os, sys
   4
   5 sys.path.extend(('.', '..'))
   6
   7 import tempfile
   8 from pprint import pprint, pformat
   9 import epub
  10
  11 from lxml.etree import Element
  12 import lxml
  13
  14 try:
  15     import json
  16 except ImportError:
  17     import simplejson as json
  18
  19 def test_ncx():
  20     try:
  21         f = open('tests/firefox-info.json')
  22     except:
  23         f = open('firefox-info.json')
  24     info = json.load(f)
  25     f.close()
  26
  27     metadata = info['metadata']
  28     manifest = info['manifest']
  29     toc = info['TOC']
  30
  31     #to switch from id to filename,
  32     filemap = dict((k, v[0]) for k, v in manifest.iteritems())
  33
  34     pprint(toc)
  35     ncx = epub.make_ncx(toc, metadata, filemap)
  36     #pprint(ncx)
  37     print ncx
  38
  39
  40
  41
  42
  43
  44 if __name__ == '__main__':
  45     test_ncx()
  46