bpython/bpython/test/test_curtsies_coderunner.py at master · godhelpjun/bpython · GitHub

History

53 lines (39 loc) · 1.45 KB

Raw

# encoding: utf-8

import sys

from bpython.test import mock, unittest

from bpython.curtsiesfrontend.coderunner import CodeRunner, FakeOutput

class TestCodeRunner(unittest.TestCase):

def setUp(self):

self.orig_stdout = sys.stdout

self.orig_stderr = sys.stderr

def tearDown(self):

sys.stdout = self.orig_stdout

sys.stderr = self.orig_stderr

def test_simple(self):

c = CodeRunner(request_refresh=lambda: self.orig_stdout.flush() or

self.orig_stderr.flush())

stdout = FakeOutput(c, lambda *args, **kwargs: None)

stderr = FakeOutput(c, lambda *args, **kwargs: None)

sys.stdout = stdout

sys.stdout = stderr

c.load_code('1 + 1')

c.run_code()

def test_exception(self):

c = CodeRunner(request_refresh=lambda: self.orig_stdout.flush() or

self.orig_stderr.flush())

def ctrlc():

raise KeyboardInterrupt()

stdout = FakeOutput(c, lambda x: ctrlc())

stderr = FakeOutput(c, lambda *args, **kwargs: None)

sys.stdout = stdout

sys.stderr = stderr

c.load_code('1 + 1')

c.run_code()

class TestFakeOutput(unittest.TestCase):

def assert_unicode(self, s):

self.assertIsInstance(s, type(u''))

def test_bytes(self):

out = FakeOutput(mock.Mock(), self.assert_unicode)

out.write('native string type')

bpython/bpython/test/test_curtsies_coderunner.py at master · godhelpjun/bpython · GitHub

Expand file tree

Breadcrumbs

test_curtsies_coderunner.py

Latest commit

History

Breadcrumbs

test_curtsies_coderunner.py

File metadata and controls