FilesExpand file tree

main

/

test_interpreter.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

59 lines (43 loc) · 1.46 KB

main

/

test_interpreter.py

Copy path

Top

File metadata and controls

• Code
• Blame

59 lines (43 loc) · 1.46 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

55

56

57

58

59

import unittest

from curtsies.fmtfuncs import bold, green, magenta, cyan, red, plain

from bpython.curtsiesfrontend import interpreter

class Interpreter(interpreter.Interp):

def __init__(self):

super().__init__()

self.a = []

self.write = self.a.append

class TestInterpreter(unittest.TestCase):

def test_syntaxerror(self):

i = Interpreter()

i.runsource("1.1.1.1")

expected = (

" File "

+ green('"<input>"')

+ ", line "

+ bold(magenta("1"))

+ "\n 1.1.1.1\n ^^\n"

+ bold(red("SyntaxError"))

+ ": "

+ cyan("invalid syntax")

+ "\n"

)

a = i.a

self.assertMultiLineEqual(str(plain("").join(a)), str(expected))

self.assertEqual(plain("").join(a), expected)

def test_traceback(self):

i = Interpreter()

def f():

return 1 / 0

def gfunc():

return f()

i.runsource("gfunc()")

a = str(plain("").join(i.a))

self.assertIn("name 'gfunc' is not defined", a)

self.assertIn("NameErro", a)

def test_getsource_works_on_interactively_defined_functions(self):

source = "def foo(x):\n return x + 1\n"

i = interpreter.Interp()

i.runsource(source)

import inspect

inspected_source = inspect.getsource(i.locals["foo"])

self.assertEqual(inspected_source, source)