1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
import unittest
from beziers.utils.curvefitter import CurveFit
from beziers.path import BezierPath
from beziers.point import Point
class CurveFitterMethods(unittest.TestCase):
def test_cf1(self):
nodes = [
Point(122,102), Point(35,200), Point(228,145), Point(190,46)
]
t = CurveFit._leftTangent(nodes)
self.assertAlmostEqual(t.x, -0.663890062102)
self.assertAlmostEqual(t.y,0.747830184896)
t = CurveFit._rightTangent(nodes)
self.assertAlmostEqual(t.x,-0.3583470773350791)
self.assertAlmostEqual(t.y, -0.9335884383203376)
def test_cf2(self):
nodes = [
Point(100,50),
Point(50,150),
Point(100,220),
Point(200,200),
Point(250,80),
Point(220,50)
]
path = BezierPath.fromPoints(nodes)
segs = path.asSegments()
self.assertEqual(len(segs),2)
self.assertEqual(segs[0].start, Point(100.0, 50.0))
self.assertAlmostEqual(segs[0][1].x, 83.333333333)
self.assertAlmostEqual(segs[0][1].y, 83.333333333)
self.assertEqual(segs[0].end, Point(50.0, 150.0))
self.assertAlmostEqual(segs[1][1].x, 50)
self.assertEqual(segs[1].end, Point(220.0, 50.0))
def not_a_test_cf3(self):
import matplotlib.pyplot as plt
import math
fig, ax = plt.subplots()
points = [
Point(100,50),
Point(50,150),
Point(150,250),
Point(200,220),
Point(250,80),
Point(220,50)
]
path = BezierPath.fromPoints(points)
centroid = path.bounds().centroid
path.rotate(centroid, math.pi/2)
path.balance()
path.plot(ax)
path.offset(Point(5,5)).plot(ax, color="red")
path.offset(Point(-5,-5)).plot(ax, color="green")
plt.show()
|
