Made vigil into a package with a setup.py file.

2017-06-27 14:03:32 +01:00 · 2017-06-27 14:03:32 +01:00 · 5728e5cff3
commit 5728e5cff3
parent 49f8d87659
135 changed files with 76 additions and 50 deletions
--- a/termstr_test.py
+++ b/termstr_test.py
@ -1,127 +0,0 @@
-#!/usr/bin/env python3
-
-# Copyright (C) 2015-2017 Andrew Hamilton. All rights reserved.
-# Licensed under the Artistic License 2.0.
-
-import os
-import pickle
-import unittest
-
-os.environ["TERM"] = "xterm-256color"
-
-from termstr import TermStr, CharStyle
-import termstr
-
-
-class CacheFirstResultTestCase(unittest.TestCase):
-
-    def test_cache_first_result_decorator(self):
-        class A:
-            @termstr._cache_first_result
-            def a(self, foo):
-                return foo
-        a = A()
-        self.assertEqual(a.a(3), 3)
-        self.assertEqual(a.a(4), 3)
-
-        class B:
-            @termstr._cache_first_result
-            def b(self, foo):
-                return foo
-        b = B()
-        self.assertEqual(b.b(5), 5)
-
-
-class CharStyleTests(unittest.TestCase):
-
-    def setUp(self):
-        self.style = CharStyle()
-
-    def test_default_char_style(self):
-        self.assertEqual(self.style.fg_color, termstr.Color.white)
-        self.assertEqual(self.style.bg_color, termstr.Color.black)
-        self.assertEqual(self.style.is_bold, False)
-        self.assertEqual(self.style.is_underlined, False)
-
-    def test_pickle_char_style(self):
-        style = CharStyle()
-        loaded_style = pickle.loads(pickle.dumps(style))
-        self.assertEqual(style, loaded_style)
-        self.assertTrue(style is loaded_style)
-
-    def test_repr(self):
-        self.assertEqual(repr(self.style),
-                         "<CharStyle: fg:(255, 255, 255) bg:(0, 0, 0) attr:>")
-
-    def test_code_for_term(self):
-        self.assertEqual(self.style.code_for_term(),
-                         "\x1b(B\x1b[m\x1b[38;2;255;255;255m\x1b[48;2;0;0;0m")
-
-
-class TermStrTests(unittest.TestCase):
-
-    def test_termstr(self):
-        foo = TermStr("foo")
-        foobar = TermStr("foobar")
-        bold_style = CharStyle(3, 5, is_bold=True)
-        foo_bold = TermStr("foo", bold_style)
-        self.assertEqual(repr(foo_bold), "<TermStr: 'foo'>")
-        self.assertEqual(foo + "bar", TermStr("foobar"))
-        self.assertEqual(foo + TermStr("bar"),
-                         TermStr("foobar"))
-        self.assertEqual("bar" + foo, TermStr("barfoo"))
-        self.assertFalse(foo == foo_bold)
-        self.assertFalse(foo_bold == foo)
-        self.assertFalse("foo" == foo_bold)
-        self.assertTrue("food" != foo_bold)
-        self.assertFalse(foo != foo)
-        self.assertTrue(foo != foo_bold)
-        self.assertFalse(foo_bold == "foo")
-        self.assertTrue(foo_bold != "food")
-        self.assertEqual(foobar[:2], TermStr("fo"))
-        self.assertEqual(foobar[2:], TermStr("obar"))
-        self.assertEqual(foobar[::2], TermStr("foa"))
-        self.assertEqual(foobar[3], TermStr("b"))
-        self.assertEqual(foo_bold[1], TermStr("o", bold_style))
-        self.assertTrue(foo.startswith("fo"))
-        self.assertTrue(foo.endswith("oo"))
-        self.assertEqual(foo.index("o"), 1)
-        self.assertTrue("fo" in foo)
-        self.assertEqual(foo.find("oo"), 1)
-        self.assertEqual(TermStr("fo") * 2, TermStr("fofo"))
-        self.assertEqual(2 * TermStr("fo"), TermStr("fofo"))
-        self.assertEqual(foobar.split("b"), [TermStr("foo"),
-                                             TermStr("ar")])
-        self.assertEqual(foo.join(["C", "D"]), TermStr("CfooD"))
-        self.assertEqual(foo.join(["C", TermStr("D")]),
-                         TermStr("CfooD"))
-        self.assertEqual(foo.join([]), TermStr(""))
-        self.assertEqual(foo.join(["C"]), TermStr("C"))
-        bar = TermStr("bar", bold_style)
-        self.assertEqual((foo + "\n" + bar).splitlines(), [foo, bar])
-        self.assertEqual((foo + "\r\n" + bar).splitlines(), [foo, bar])
-        self.assertEqual((foo + "\n" + bar).splitlines(keepends=True),
-                         [TermStr("foo\n"), bar])
-        self.assertEqual((foo + "\r\n" + bar).splitlines(keepends=True),
-                         [TermStr("foo\r\n"), bar])
-        self.assertEqual(foo.ljust(5), foo + TermStr("  "))
-        self.assertEqual(foo.rjust(5), TermStr("  ") + foo)
-        self.assertEqual(TermStr("FOO").lower(), foo)
-        self.assertEqual(TermStr("FOO", bold_style).lower(), foo_bold)
-        self.assertEqual(TermStr("FOO").swapcase(), foo)
-        self.assertEqual(TermStr("FOO", bold_style).swapcase(), foo_bold)
-        phrase = TermStr("foo bar")
-        self.assertEqual(phrase.title(), TermStr("Foo Bar"))
-        self.assertEqual(phrase.capitalize(), TermStr("Foo bar"))
-        self.assertEqual(foo.upper(), TermStr("FOO"))
-        self.assertEqual(foo_bold.center(0), foo_bold)
-        self.assertEqual(foo_bold.center(7),
-                         TermStr("  ") + foo_bold + TermStr("  "))
-        self.assertEqual(foo_bold.ljust(0), foo_bold)
-        self.assertEqual(foo_bold.ljust(5), foo_bold + TermStr("  "))
-        self.assertEqual(foo_bold.rjust(0), foo_bold)
-        self.assertEqual(foo_bold.rjust(5), TermStr("  ") + foo_bold)
-
-
-if __name__ == "__main__":
-    unittest.main()