FilesExpand file tree

markdown-readme

/

test_curtsies_coderunner.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

54 lines (43 loc) · 1.48 KB

markdown-readme

/

test_curtsies_coderunner.py

Copy path

Top

File metadata and controls

• Code
• Blame

54 lines (43 loc) · 1.48 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

import sys

import unittest

from unittest import mock

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, None)

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

sys.stdout = stdout

sys.stdout = stderr

c.load_code("1 + 1")

c.run_code()

c.run_code()

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(), None)

stderr = FakeOutput(c, lambda *args, **kwargs: None, 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, str)

def test_bytes(self):

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

out.write("native string type")