def test_gss(self): """ A rect passed to gs and gss should create the same value on the pen """ dp1 = (DraftingPen().define(r=Rect(100, 100)).gss("$r")) dp2 = (DraftingPen().define(r=Rect(100, 100)).gs("$r")) self.assertEqual(dp1.value, dp2.value)
def test_collapse(self): dps = DraftingPens([ DraftingPens([DraftingPens([DraftingPen()])]), DraftingPens([DraftingPen()]), ]) dps.collapse() # should not mutate by default self.assertIsInstance(dps[0], DraftingPens) self.assertIsInstance(dps[0][0], DraftingPens) dps.collapse(onself=True) # now it should mutate self.assertEqual(len(dps), 2) self.assertNotIsInstance(dps[0], DraftingPens) self.assertNotIsInstance(dps[1], DraftingPens)
def test_pens_ambit(self): dps = ( DraftingPens([ DraftingPen(Rect(50, 50)), DraftingPen(Rect(100, 100, 100, 100)) ]) #.print(lambda x: x.tree()) ) ram = dps.ambit() self.assertEqual(ram, Rect(0, 0, 200, 200)) moves = [] dps.walk(lambda p, pos, _: moves.append([p, pos])) self.assertEqual(moves[0][0], dps) self.assertEqual(moves[0][1], -1) self.assertEqual(moves[1][1], 0)
def test_gs_relative_moves(self): r = Rect(100, 100) dp = (DraftingPen().define(r=r).gs("$r↖ ¬OX50OY-50 §OY-50")) self.assertEqual(len(dp.value), 4) self.assertEqual(dp.value[0][-1][0], Point(0, 100)) self.assertEqual(dp.value[1][-1][0], Point(50, 50)) self.assertEqual(dp.value[2][-1][0], Point(0, 50))
def test_gs_arrowcluster(self): r = Rect(100, 100) dp = (DraftingPen().define(r=r).gs("$r↖↗↘")) self.assertEqual(len(dp.value), 4) self.assertEqual(dp.value[0][-1][0], Point(0, 100)) self.assertEqual(dp.value[1][-1][0], Point(100, 100)) self.assertEqual(dp.value[2][-1][0], Point(100, 0))
def test_cairo_png(self): with test_image(self, "test_cairo.png") as (i, r): rr = Rect(0, 0, 100, 100) dp = (DraftingPen().define(r=rr, c=75).gs( "$r↗ ↘|$c|$r↓ ↙|$c|$r↖").align(r).scale(1.2).rotate(180).f( hsl(0.5, a=0.1)).s(hsl(0.9)).sw(5)) CairoPen.Composite(dp, r, i) self.assertEqual(len(dp.value), 4) self.assertEqual(type(dp), DraftingPen)
def remove_blanklines(self, blank=None): if not blank and self.blankfill: blank = self.blankfill for line in self._pens: txt = reduce(lambda acc, p: p.data.get("txt", "") + acc, line, "") if txt == blank: line._pens = [DraftingPen()] return self
def test_gs(self): r = Rect(0, 0, 100, 100) dps = DraftingPens() dp = (DraftingPen().define(r=r).gs("$r↖ $r↗ ↘|65|$r↙ ɜ")) self.assertEqual(len(dp.value), 4) self.assertEqual(dp.value[-2][-1][0], Point(100, 35)) self.assertEqual(dp.value[-1][0], "endPath") self.assertEqual(dp.unended(), False) dps.append(DraftingPens([dp])) self.assertEqual(len(dps.tree().splitlines()), 3) self.assertEqual(dps.tree().splitlines()[-1], " | | DraftingPen<4mvs:end/>")
def test_transforms(self): dp = (DraftingPen(Rect(100, 100)).frame(Rect(100, 100)).align(Rect(200, 200))) self.assertEqual(dp.frame().mxx, 150) self.assertEqual(dp.value[-2][-1][-1][0], 50) self.assertEqual(dp.copy().rotate(45).round().value, dp.copy().rotate(360 + 45).round().value) self.assertEqual(dp.copy().scale(2).ambit().w, 200)
def test_remove_blanks(self): dps = (DraftingPens([DraftingPen(Rect(50, 50)), DraftingPen()])) self.assertEqual(len(dps), 2) dps.remove_blanks() self.assertEqual(len(dps), 1)
def test_reverse(self): dp = (DraftingPen().define(r=Rect(100, 100)).gs("$r↖ $r↗ $r↘ ɜ")) p1 = dp.value[0][-1] p2 = dp.reverse().value[-2][-1] self.assertEqual(p1, p2)
def to_pen(self): from drafting.pens.draftingpen import DraftingPen return DraftingPen(self)