test/unit/test_parter.py

   1 import sys
   2 if sys.path[0] != "../..":
   3     sys.path.insert(0, "../..")
   4
   5 import unittest
   6
   7 from pyx import *
   8 from pyx.graph.axis.tick import tick
   9 from pyx.graph.axis.parter import lin, log
  10
  11
  12 class ParterTestCase(unittest.TestCase):
  13
  14     def PartEqual(self, part1, part2):
  15         self.failUnlessEqual(len(part1), len(part2))
  16         for tick1, tick2 in zip(part1, part2):
  17             self.failUnlessEqual(tick1, tick2)
  18             self.failUnlessEqual(tick1.ticklevel, tick2.ticklevel)
  19             self.failUnlessEqual(tick1.labellevel, tick2.labellevel)
  20             self.failUnlessEqual(tick1.label, tick2.label)
  21
  22     def testLinParter(self):
  23         self.PartEqual(lin(tickdist="1", labeldist=()).defaultpart(0, 1, 1, 1),
  24                        [tick((0, 1), 0, None), tick((1, 1), 0, None)])
  25         self.PartEqual(lin(tickdist=("1", "0.5"), labeldist=()).defaultpart(0, 1, 1, 1),
  26                        [tick((0, 1), 0, None), tick((1, 2), 1, None), tick((1, 1), 0, None)])
  27         self.PartEqual(lin(tickdist=("1", "0.5", "0.25"), labeldist=()).defaultpart(0, 1, 1, 1),
  28                        [tick((0, 1), 0, None), tick((1, 4), 2, None), tick((1, 2), 1, None), tick((3, 4), 2, None), tick((1, 1), 0, None)])
  29         self.PartEqual(lin(tickdist=("1", "0.5"), labeldist=(), extendtick=None).defaultpart(0, 1.5, 1, 1),
  30                        [tick((0, 1), 0, None), tick((1, 2), 1, None), tick((1, 1), 0, None), tick((3, 2), 1, None)])
  31         self.PartEqual(lin(tickdist=("1", "0.5"), labeldist=(), extendtick=1).defaultpart(0, 1.2, 1, 1),
  32                        [tick((0, 1), 0, None), tick((1, 2), 1, None), tick((1, 1), 0, None), tick((3, 2), 1, None)])
  33         self.PartEqual(lin(tickdist=("1", "0.5"), labeldist=(), extendtick=0).defaultpart(0, 1.2, 1, 1),
  34                        [tick((0, 1), 0, None), tick((1, 2), 1, None), tick((1, 1), 0, None), tick((3, 2), 1, None), tick((2, 1), 0, None)])
  35         self.PartEqual(lin(labeldist="1", tickdist=()).defaultpart(0, 1, 1, 1),
  36                        [tick((0, 1), None, 0), tick((1, 1), None, 0)])
  37         self.PartEqual(lin(labeldist=("1", "0.5"), tickdist=()).defaultpart(0, 1, 1, 1),
  38                        [tick((0, 1), None, 0), tick((1, 2), None, 1), tick((1, 1), None, 0)])
  39         self.PartEqual(lin(labeldist=("1", "0.5", "0.25"), tickdist=()).defaultpart(0, 1, 1, 1),
  40                        [tick((0, 1), None, 0), tick((1, 4), None, 2), tick((1, 2), None, 1), tick((3, 4), None, 2), tick((1, 1), None, 0)])
  41         self.PartEqual(lin(labeldist=("1", "0.5"), tickdist=(), extendlabel=None).defaultpart(0, 1.5, 1, 1),
  42                        [tick((0, 1), None, 0), tick((1, 2), None, 1), tick((1, 1), None, 0), tick((3, 2), None, 1)])
  43         self.PartEqual(lin(labeldist=("1", "0.5"), tickdist=(), extendlabel=1).defaultpart(0, 1.2, 1, 1),
  44                        [tick((0, 1), None, 0), tick((1, 2), None, 1), tick((1, 1), None, 0), tick((3, 2), None, 1)])
  45         self.PartEqual(lin(labeldist=("1", "0.5"), tickdist=(), extendlabel=0).defaultpart(0, 1.2, 1, 1),
  46                        [tick((0, 1), None, 0), tick((1, 2), None, 1), tick((1, 1), None, 0), tick((3, 2), None, 1), tick((2, 1), None, 0)])
  47         self.PartEqual(lin(tickdist="1").defaultpart(0, 1, 1, 1),
  48                        [tick((0, 1), 0, 0), tick((1, 1), 0, 0)])
  49         self.PartEqual(lin(tickdist=("1", "0.5")).defaultpart(0, 1, 1, 1),
  50                        [tick((0, 1), 0, 0), tick((1, 2), 1, None), tick((1, 1), 0, 0)])
  51         self.PartEqual(lin(tickdist=("1", "0.5", "0.25")).defaultpart(0, 1, 1, 1),
  52                        [tick((0, 1), 0, 0), tick((1, 4), 2, None), tick((1, 2), 1, None), tick((3, 4), 2, None), tick((1, 1), 0, 0)])
  53         self.PartEqual(lin(tickdist=("1", "0.5"), extendtick=None).defaultpart(0, 1.5, 1, 1),
  54                        [tick((0, 1), 0, 0), tick((1, 2), 1, None), tick((1, 1), 0, 0), tick((3, 2), 1, None)])
  55         self.PartEqual(lin(tickdist=("1", "0.5"), extendtick=1).defaultpart(0, 1.2, 1, 1),
  56                        [tick((0, 1), 0, 0), tick((1, 2), 1, None), tick((1, 1), 0, 0), tick((3, 2), 1, None)])
  57         self.PartEqual(lin(tickdist=("1", "0.5"), extendtick=0).defaultpart(0, 1.2, 1, 1),
  58                        [tick((0, 1), 0, 0), tick((1, 2), 1, None), tick((1, 1), 0, 0), tick((3, 2), 1, None), tick((2, 1), 0, 0)])
  59         self.PartEqual(lin(labeldist="1").defaultpart(0, 1, 1, 1),
  60                        [tick((0, 1), 0, 0), tick((1, 1), 0, 0)])
  61         self.PartEqual(lin(labeldist=("1", "0.5")).defaultpart(0, 1, 1, 1),
  62                        [tick((0, 1), 0, 0), tick((1, 2), None, 1), tick((1, 1), 0, 0)])
  63         self.PartEqual(lin(labeldist=("1", "0.5", "0.25")).defaultpart(0, 1, 1, 1),
  64                        [tick((0, 1), 0, 0), tick((1, 4), None, 2), tick((1, 2), None, 1), tick((3, 4), None, 2), tick((1, 1), 0, 0)])
  65         self.PartEqual(lin(labeldist=("1", "0.5"), extendtick=None).defaultpart(0, 1.5, 1, 1),
  66                        [tick((0, 1), 0, 0), tick((1, 2), None, 1), tick((1, 1), 0, 0), tick((3, 2), None, 1)])
  67         self.PartEqual(lin(labeldist=("1", "0.5"), extendtick=None, extendlabel=1).defaultpart(0, 1.2, 1, 1),
  68                        [tick((0, 1), 0, 0), tick((1, 2), None, 1), tick((1, 1), 0, 0), tick((3, 2), None, 1)])
  69         self.PartEqual(lin(labeldist=("1", "0.5"), extendtick=None, extendlabel=0).defaultpart(0, 1.2, 1, 1),
  70                        [tick((0, 1), 0, 0), tick((1, 2), None, 1), tick((1, 1), 0, 0), tick((3, 2), None, 1), tick((2, 1), 0, 0)])
  71
  72     def testLogParter(self):
  73         self.PartEqual(log(tickpos=log.pre1exp, labelpos=()).defaultpart(1, 10, 1, 1),
  74                        [tick((1, 1), 0, None), tick((10, 1), 0, None)])
  75         self.PartEqual(log(tickpos=log.pre1exp, labelpos=()).defaultpart(1, 100, 1, 1),
  76                        [tick((1, 1), 0, None), tick((10, 1), 0, None), tick((100, 1), 0, None)])
  77         self.PartEqual(log(tickpos=log.pre1exp2, labelpos=()).defaultpart(1, 100, 1, 1),
  78                        [tick((1, 1), 0, None), tick((100, 1), 0, None)])
  79         self.PartEqual(log(tickpos=log.pre1to9exp, labelpos=()).defaultpart(1, 10, 1, 1),
  80                        [tick((1, 1), 0, None), tick((2, 1), 0, None), tick((3, 1), 0, None), tick((4, 1), 0, None), tick((5, 1), 0, None), tick((6, 1), 0, None), tick((7, 1), 0, None), tick((8, 1), 0, None), tick((9, 1), 0, None), tick((10, 1), 0, None)])
  81         self.PartEqual(log(tickpos=(log.pre1exp, log.pre1to9exp), labelpos=()).defaultpart(1, 10, 1, 1),
  82                        [tick((1, 1), 0, None), tick((2, 1), 1, None), tick((3, 1), 1, None), tick((4, 1), 1, None), tick((5, 1), 1, None), tick((6, 1), 1, None), tick((7, 1), 1, None), tick((8, 1), 1, None), tick((9, 1), 1, None), tick((10, 1), 0, None)])
  83         self.PartEqual(log(tickpos=(log.pre1exp, log.pre1to9exp)).defaultpart(1, 10, 1, 1),
  84                        [tick((1, 1), 0, 0), tick((2, 1), 1, None), tick((3, 1), 1, None), tick((4, 1), 1, None), tick((5, 1), 1, None), tick((6, 1), 1, None), tick((7, 1), 1, None), tick((8, 1), 1, None), tick((9, 1), 1, None), tick((10, 1), 0, 0)])
  85
  86
  87 if __name__ == "__main__":
  88     unittest.main()