FilesExpand file tree

 master

/

test_prefilter.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

119 lines (97 loc) · 4.02 KB

 master

/

test_prefilter.py

Copy path

Top

File metadata and controls

• Code
• Blame

119 lines (97 loc) · 4.02 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

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

"""Tests for input manipulation machinery."""

#-----------------------------------------------------------------------------

# Imports

#-----------------------------------------------------------------------------

import nose.tools as nt

from IPython.core.prefilter import AutocallChecker

from IPython.testing.globalipapp import get_ipython

#-----------------------------------------------------------------------------

# Tests

#-----------------------------------------------------------------------------

ip = get_ipython()

def test_prefilter():

"""Test user input conversions"""

# pairs of (raw, expected correct) input

pairs = [ ('2+2','2+2'),

]

for raw, correct in pairs:

nt.assert_equal(ip.prefilter(raw), correct)

def test_prefilter_shadowed():

def dummy_magic(line): pass

prev_automagic_state = ip.automagic

ip.automagic = True

ip.autocall = 0

try:

# These should not be transformed - they are shadowed by other names

for name in ['if', 'zip', 'get_ipython']: # keyword, builtin, global

ip.register_magic_function(dummy_magic, magic_name=name)

res = ip.prefilter(name+' foo')

nt.assert_equal(res, name+' foo')

del ip.magics_manager.magics['line'][name]

# These should be transformed

for name in ['fi', 'piz', 'nohtypi_teg']:

ip.register_magic_function(dummy_magic, magic_name=name)

res = ip.prefilter(name+' foo')

nt.assert_not_equal(res, name+' foo')

del ip.magics_manager.magics['line'][name]

finally:

ip.automagic = prev_automagic_state

def test_autocall_binops():

"""See https://github.com/ipython/ipython/issues/81"""

ip.magic('autocall 2')

f = lambda x: x

ip.user_ns['f'] = f

try:

nt.assert_equal(ip.prefilter('f 1'),'f(1)')

for t in ['f +1', 'f -1']:

nt.assert_equal(ip.prefilter(t), t)

# Run tests again with a more permissive exclude_regexp, which will

# allow transformation of binary operations ('f -1' -> 'f(-1)').

pm = ip.prefilter_manager

ac = AutocallChecker(shell=pm.shell, prefilter_manager=pm,

config=pm.config)

try:

ac.priority = 1

ac.exclude_regexp = r'^[,&^\|\*/]|^is |^not |^in |^and |^or '

pm.sort_checkers()

nt.assert_equal(ip.prefilter('f -1'), 'f(-1)')

nt.assert_equal(ip.prefilter('f +1'), 'f(+1)')

finally:

pm.unregister_checker(ac)

finally:

ip.magic('autocall 0')

del ip.user_ns['f']

def test_issue_114():

"""Check that multiline string literals don't expand as magic

see http://github.com/ipython/ipython/issues/114"""

template = '"""\n%s\n"""'

# Store the current value of multi_line_specials and turn it off before

# running test, since it could be true (case in which the test doesn't make

# sense, as multiline string literals *will* expand as magic in that case).

msp = ip.prefilter_manager.multi_line_specials

ip.prefilter_manager.multi_line_specials = False

try:

for mgk in ip.magics_manager.lsmagic()['line']:

raw = template % mgk

nt.assert_equal(ip.prefilter(raw), raw)

finally:

ip.prefilter_manager.multi_line_specials = msp

def test_prefilter_attribute_errors():

"""Capture exceptions thrown by user objects on attribute access.

See http://github.com/ipython/ipython/issues/988."""

class X(object):

def __getattr__(self, k):

raise ValueError('broken object')

def __call__(self, x):

return x

# Create a callable broken object

ip.user_ns['x'] = X()

ip.magic('autocall 2')

try:

# Even if x throws an attribute error when looking at its rewrite

# attribute, we should not crash. So the test here is simply making

# the prefilter call and not having an exception.

ip.prefilter('x 1')

finally:

del ip.user_ns['x']

ip.magic('autocall 0')