fill3: Coding style

- Optimize splitting stdin by using a regex.
This commit is contained in:
Andrew Hamilton 2022-02-13 21:23:37 +10:00
parent ccdd657c10
commit f77d5743a1
2 changed files with 12 additions and 24 deletions

View file

@ -140,16 +140,16 @@ class WidgetTests(unittest.TestCase):
"B A")
class DigestTerminalInputTestCase(unittest.TestCase):
class TerminalInputTestCase(unittest.TestCase):
def test_digest_terminal_input(self):
self.assertRaises(UnboundLocalError, fill3.digest_terminal_input, "")
self.assertEqual(fill3.digest_terminal_input("a"), ["a"])
self.assertEqual(fill3.digest_terminal_input("ab"), ["ab"])
self.assertEqual(fill3.digest_terminal_input("a\nb"), ["a", "\n", "b"])
self.assertEqual(fill3.digest_terminal_input("a\tb"), ["a", "\t", "b"])
self.assertEqual(fill3.digest_terminal_input(terminal.UP * 2), [terminal.UP] * 2)
self.assertEqual(fill3.digest_terminal_input(terminal.CTRL_C * 2), [terminal.CTRL_C] * 2)
def test_input_regex(self):
self.assertEqual(fill3._INPUT_REGEX.findall(""), [])
self.assertEqual(fill3._INPUT_REGEX.findall("a"), ["a"])
self.assertEqual(fill3._INPUT_REGEX.findall("ab"), ["ab"])
self.assertEqual(fill3._INPUT_REGEX.findall("a\nb"), ["a", "\n", "b"])
self.assertEqual(fill3._INPUT_REGEX.findall("a\tb"), ["a", "\t", "b"])
self.assertEqual(fill3._INPUT_REGEX.findall(terminal.UP * 2), [terminal.UP] * 2)
self.assertEqual(fill3._INPUT_REGEX.findall(terminal.CTRL_C * 2), [terminal.CTRL_C] * 2)
if __name__ == "__main__":