test_richtext.py

import unittest
from coldtype import Rect, Font, Style
from coldtype.text.richtext import RichText

f1 = Font.ColdtypeObviously()
f2 = Font.MutatorSans()

class TestRichText(unittest.TestCase):
    def test_preserve_space(self):
        r = Rect(1200, 300)
        rt = RichText(r, "HELLO[i] COLDTYPE", dict(
            i=Style(f2, 200, wdth=0, wght=1),
            default=Style(f1, 200, wdth=0))).align(r)

        self.assertEqual(rt[0][0].data("txt"), "COLDTYPE")
        self.assertEqual(rt[0][1].data("txt"), "HELLO ")

        self.assertEqual(rt[0][1][0].glyphName, "space")
        self.assertEqual(rt[0][1][-1].glyphName, "H")

        self.assertEqual(rt[0][0][0].glyphName, "E")
        self.assertEqual(rt[0][0][-1].glyphName, "C")

        space_width = rt[0][1][0].ambit(th=0).w
        self.assertEqual(space_width, 50)

        self.assertGreater(
            rt[0][1].ambit(th=0).w - space_width,
            rt[0][1].ambit(th=1).w)
        
        rt.picklejar(r)
    
    def test_ligature(self):
        clarette = Font.Cacheable("~/Type/fonts/fonts/_wdths/ClaretteGX.ttf")

        txt = "fi¬joff≤asdf≥"
        r = Rect(1080, 300)
        
        gl = (RichText(r, txt,
            dict(
                default=Style(clarette, 200),
                asdf=Style(clarette, 200, wdth=1)),
            tag_delimiters=["≤", "≥"],
            visible_boundaries=["¶"],
            invisible_boundaries=["¬"])
            .align(r))
        
        self.assertEqual(gl[0][0].data("style_names")[0], "asdf")
        self.assertEqual(len(gl[0][1].data("style_names")), 0)
        self.assertEqual(gl[0][0][0].glyphName, "f_f")
        self.assertEqual(gl[0][1][0].glyphName, "f_i")
        
        gl.picklejar(r)

if __name__ == "__main__":
    unittest.main()