test/unit/test_path.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.path import *
   9
  10 epsilon = 1e-5
  11 def isEqual(l1, l2):
  12     return abs(unit.topt(l1-l2))<epsilon
  13
  14
  15 class NormpathTestCase(unittest.TestCase):
  16     def testsplit(self):
  17         p = path(moveto(0,0), lineto(1,0), moveto(2,0), lineto(3,0))
  18         np = normpath(p)
  19
  20         # one split parameter
  21         sp = np.split([0])
  22         assert len(sp)==2 and sp[0] is None and isEqual(sp[1].arclen(), 2)
  23
  24         sp = np.split([0.5])
  25         assert len(sp)==2 and isEqual(sp[0].arclen(), 0.5) and isEqual(sp[1].arclen(), 1.5)
  26
  27         sp = np.split([1])
  28         assert len(sp)==2 and isEqual(sp[0].arclen(), 1) and isEqual(sp[1].arclen(), 1)
  29
  30         sp = np.split([1.5])
  31         assert len(sp)==2 and isEqual(sp[0].arclen(), 1.5) and isEqual(sp[1].arclen(), 0.5)
  32
  33         sp = np.split([2])
  34         assert len(sp)==2 and isEqual(sp[0].arclen(), 2) and sp[1] is None
  35
  36         # two split parameters
  37         sp = np.split([0, 0.5])
  38         assert len(sp)==3 and sp[0] is None and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 1.5)
  39
  40         sp = np.split([0, 1])
  41         assert len(sp)==3 and sp[0] is None and isEqual(sp[1].arclen(), 1) and isEqual(sp[2].arclen(), 1)
  42
  43         sp = np.split([0, 1.5])
  44         assert len(sp)==3 and sp[0] is None and isEqual(sp[1].arclen(), 1.5) and isEqual(sp[2].arclen(), 0.5)
  45
  46         sp = np.split([0, 2])
  47         assert len(sp)==3 and sp[0] is None and isEqual(sp[1].arclen(), 2) and sp[2] is None
  48
  49         sp = np.split([0.5, 1])
  50         assert len(sp)==3 and isEqual(sp[0].arclen(), 0.5) and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 1)
  51
  52         sp = np.split([0.5, 1.5])
  53         assert len(sp)==3 and isEqual(sp[0].arclen(), 0.5) and isEqual(sp[1].arclen(), 1) and isEqual(sp[2].arclen(), 0.5)
  54
  55         sp = np.split([0.5, 2])
  56         assert len(sp)==3 and isEqual(sp[0].arclen(), 0.5) and isEqual(sp[1].arclen(), 1.5) and sp[2] is None
  57
  58         sp = np.split([1, 1.5])
  59         assert len(sp)==3 and isEqual(sp[0].arclen(), 1) and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 0.5)
  60
  61         sp = np.split([1, 2])
  62         assert len(sp)==3 and isEqual(sp[0].arclen(), 1) and isEqual(sp[1].arclen(), 1) and sp[2] is None
  63
  64         sp = np.split([1.5, 2])
  65         assert len(sp)==3 and isEqual(sp[0].arclen(), 1.5) and isEqual(sp[1].arclen(), 0.5) and sp[2] is None
  66
  67         # three split parameters
  68         sp = np.split([0, 0.5, 1])
  69         assert len(sp)==4 and sp[0] is None and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 0.5) and isEqual(sp[3].arclen(), 1)
  70
  71         sp = np.split([0, 0.5, 1.5])
  72         assert len(sp)==4 and sp[0] is None and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 1) and isEqual(sp[3].arclen(), 0.5)
  73
  74         sp = np.split([0, 0.5, 2])
  75         assert len(sp)==4 and sp[0] is None and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 1.5) and sp[3] is None
  76
  77         sp = np.split([0, 1, 1.5])
  78         assert len(sp)==4 and sp[0] is None and isEqual(sp[1].arclen(), 1) and isEqual(sp[2].arclen(), 0.5) and isEqual(sp[3].arclen(), 0.5)
  79
  80         sp = np.split([0, 1, 2])
  81         assert len(sp)==4 and sp[0] is None and isEqual(sp[1].arclen(), 1) and isEqual(sp[2].arclen(), 1) and sp[3] is None
  82
  83
  84         sp = np.split([0, 1.5, 2])
  85         assert len(sp)==4 and sp[0] is None and isEqual(sp[1].arclen(), 1.5) and isEqual(sp[2].arclen(), 0.5) and sp[3] is None
  86
  87         sp = np.split([0.5, 1, 1.5])
  88         assert len(sp)==4 and isEqual(sp[0].arclen(), 0.5) and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 0.5) and isEqual(sp[3].arclen(), 0.5)
  89
  90         sp = np.split([0.5, 1.5, 2])
  91         assert len(sp)==4 and isEqual(sp[0].arclen(), 0.5) and isEqual(sp[1].arclen(), 1) and isEqual(sp[2].arclen(), 0.5) and sp[3] is None
  92
  93         sp = np.split([1, 1.5, 2])
  94         assert len(sp)==4 and isEqual(sp[0].arclen(), 1) and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 0.5) and sp[3] is None
  95
  96
  97         # four split parameters
  98         sp = np.split([0, 0.5, 1, 1.5])
  99         assert len(sp)==5 and sp[0] is None and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 0.5) and isEqual(sp[3].arclen(), 0.5) and isEqual(sp[4].arclen(), 0.5)
 100
 101         sp = np.split([0, 0.5, 1, 2])
 102         assert len(sp)==5 and sp[0] is None and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 0.5) and isEqual(sp[3].arclen(), 1) and sp[4] is None
 103
 104         sp = np.split([0, 0.5, 1.5, 2])
 105         assert len(sp)==5 and sp[0] is None and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 1) and isEqual(sp[3].arclen(), 0.5) and sp[4] is None
 106
 107         sp = np.split([0, 1, 1.5, 2])
 108         assert len(sp)==5 and sp[0] is None and isEqual(sp[1].arclen(), 1) and isEqual(sp[2].arclen(), 0.5) and isEqual(sp[3].arclen(), 0.5) and sp[4] is None
 109
 110         sp = np.split([0.5, 1, 1.5, 2])
 111         assert len(sp)==5 and isEqual(sp[0].arclen(), 0.5) and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 0.5) and isEqual(sp[3].arclen(), 0.5) and sp[4] is None
 112
 113
 114         # five split parameters
 115         sp = np.split([0, 0.5, 1, 1.5, 2])
 116         assert len(sp)==6 and sp[0] is None and isEqual(sp[1].arclen(), 0.5) and isEqual(sp[2].arclen(), 0.5) and isEqual(sp[3].arclen(), 0.5) and isEqual(sp[4].arclen(), 0.5) and sp[5] is None
 117
 118     def testshortnormsubpath(self):
 119         sp = normsubpath(epsilon=1)
 120         sp.append(normline(0,0, 0.5, 0))
 121         sp.append(normline(0.5,0, 1.5, 0))
 122
 123         sp.append(normline(1.5,0, 1.5, 0.3))
 124         sp.append(normline(1.5,0.3, 1.5, 0.6))
 125         sp.append(normline(1.5,0.6, 1.5, 0.9))
 126         sp.append(normline(1.5,0.9, 1.5, 1.2))
 127
 128         sp.append(normline(1.2, 1.5, 1.3, 1.6))
 129         sp.append(normcurve(1.3, 1.6, 1.4, 1.7, 1.3, 1.7, 1.3, 1.8))
 130         sp.append(normcurve(1.3, 1.8, 2.4, 2.7, 3.3, 3.7, 1.4, 1.8))
 131
 132         self.failUnlessEqual(str(sp), "subpath(open, [normline(0, 0, 1.5, 0), normline(1.5, 0, 1.5, 1.2), normcurve(1.5, 1.2, 2.4, 2.7, 3.3, 3.7, 1.4, 1.8)])")
 133
 134
 135 if __name__ == "__main__":
 136     unittest.main()