FilesExpand file tree

 master

/

test_pretty.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

120 lines (94 loc) · 3.37 KB

 master

/

test_pretty.py

Copy path

Top

File metadata and controls

• Code
• Blame

120 lines (94 loc) · 3.37 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

120

"""Tests for IPython.lib.pretty.

"""

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

# Copyright (c) 2011, the IPython Development Team.

#

# Distributed under the terms of the Modified BSD License.

#

# The full license is in the file COPYING.txt, distributed with this software.

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

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

# Imports

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

from __future__ import print_function

# Third-party imports

import nose.tools as nt

# Our own imports

from IPython.lib import pretty

from IPython.testing.decorators import skip_without

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

# Classes and functions

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

class MyList(object):

def __init__(self, content):

self.content = content

def _repr_pretty_(self, p, cycle):

if cycle:

p.text("MyList(...)")

else:

with p.group(3, "MyList(", ")"):

for (i, child) in enumerate(self.content):

if i:

p.text(",")

p.breakable()

else:

p.breakable("")

p.pretty(child)

class MyDict(dict):

def _repr_pretty_(self, p, cycle):

p.text("MyDict(...)")

class Dummy1(object):

def _repr_pretty_(self, p, cycle):

p.text("Dummy1(...)")

class Dummy2(Dummy1):

_repr_pretty_ = None

class NoModule(object):

pass

NoModule.__module__ = None

def test_indentation():

"""Test correct indentation in groups"""

count = 40

gotoutput = pretty.pretty(MyList(range(count)))

expectedoutput = "MyList(\n" + ",\n".join(" %d" % i for i in range(count)) + ")"

nt.assert_equal(gotoutput, expectedoutput)

def test_dispatch():

"""

Test correct dispatching: The _repr_pretty_ method for MyDict

must be found before the registered printer for dict.

"""

gotoutput = pretty.pretty(MyDict())

expectedoutput = "MyDict(...)"

nt.assert_equal(gotoutput, expectedoutput)

def test_callability_checking():

"""

Test that the _repr_pretty_ method is tested for callability and skipped if

not.

"""

gotoutput = pretty.pretty(Dummy2())

expectedoutput = "Dummy1(...)"

nt.assert_equal(gotoutput, expectedoutput)

def test_sets():

"""

Test that set and frozenset use Python 3 formatting.

"""

objects = [set(), frozenset(), set([1]), frozenset([1]), set([1, 2]),

frozenset([1, 2]), set([-1, -2, -3])]

expected = ['set()', 'frozenset()', '{1}', 'frozenset({1})', '{1, 2}',

'frozenset({1, 2})', '{-3, -2, -1}']

for obj, expected_output in zip(objects, expected):

got_output = pretty.pretty(obj)

yield nt.assert_equal, got_output, expected_output

@skip_without('xxlimited')

def test_pprint_heap_allocated_type():

"""

Test that pprint works for heap allocated types.

"""

import xxlimited

output = pretty.pretty(xxlimited.Null)

nt.assert_equal(output, 'xxlimited.Null')

def test_pprint_nomod():

"""

Test that pprint works for classes with no __module__.

"""

output = pretty.pretty(NoModule)

nt.assert_equal(output, 'NoModule')