FilesExpand file tree

 master

/

_paramtestpy2.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

97 lines (82 loc) · 3.15 KB

 master

/

_paramtestpy2.py

Copy path

Top

File metadata and controls

• Code
• Blame

97 lines (82 loc) · 3.15 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

"""Implementation of the parametric test support for Python 2.x

"""

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

# Copyright (C) 2009-2011 The IPython Development Team

#

# Distributed under the terms of the BSD License. The full license is in

# the file COPYING, distributed as part of this software.

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

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

# Imports

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

import sys

import unittest

from compiler.consts import CO_GENERATOR

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

# Classes and functions

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

def isgenerator(func):

try:

return func.func_code.co_flags & CO_GENERATOR != 0

except AttributeError:

return False

class ParametricTestCase(unittest.TestCase):

"""Write parametric tests in normal unittest testcase form.

Limitations: the last iteration misses printing out a newline when running

in verbose mode.

"""

def run_parametric(self, result, testMethod):

# But if we have a test generator, we iterate it ourselves

testgen = testMethod()

while True:

try:

# Initialize test

result.startTest(self)

# SetUp

try:

self.setUp()

except KeyboardInterrupt:

raise

except:

result.addError(self, sys.exc_info())

return

# Test execution

ok = False

try:

next(testgen)

ok = True

except StopIteration:

# We stop the loop

break

except self.failureException:

result.addFailure(self, sys.exc_info())

except KeyboardInterrupt:

raise

except:

result.addError(self, sys.exc_info())

# TearDown

try:

self.tearDown()

except KeyboardInterrupt:

raise

except:

result.addError(self, sys.exc_info())

ok = False

if ok: result.addSuccess(self)

finally:

result.stopTest(self)

def run(self, result=None):

if result is None:

result = self.defaultTestResult()

testMethod = getattr(self, self._testMethodName)

# For normal tests, we just call the base class and return that

if isgenerator(testMethod):

return self.run_parametric(result, testMethod)

else:

return super(ParametricTestCase, self).run(result)

def parametric(func):

"""Decorator to make a simple function into a normal test via unittest."""

class Tester(ParametricTestCase):

test = staticmethod(func)

Tester.__name__ = func.__name__

return Tester