• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1#!/usr/bin/env python3
2
3# pylint: disable=unused-import,import-error
4
5import sys
6
7
8try:
9    from tempfile import TemporaryDirectory
10except ImportError:
11    import shutil
12    import tempfile
13
14
15    class TemporaryDirectory(object):
16        def __init__(self, suffix='', prefix='tmp', dir=None):
17            # pylint: disable=redefined-builtin
18            self.name = tempfile.mkdtemp(suffix, prefix, dir)
19
20
21        def __del__(self):
22            self.cleanup()
23
24
25        def __enter__(self):
26            return self.name
27
28
29        def __exit__(self, exc, value, tb):
30            self.cleanup()
31
32
33        def cleanup(self):
34            if self.name:
35                shutil.rmtree(self.name)
36                self.name = None
37
38
39if sys.version_info >= (3, 0):
40    from os import makedirs
41else:
42    import os
43
44
45    def makedirs(path, exist_ok):
46        if exist_ok and os.path.exists(path):
47            return
48        os.makedirs(path)
49
50
51if sys.version_info >= (3, 0):
52    from io import StringIO
53else:
54    from StringIO import StringIO
55
56
57try:
58    from unittest.mock import patch
59except ImportError:
60    import contextlib
61
62
63    @contextlib.contextmanager
64    def patch(target, mock):
65        obj, attr = target.rsplit('.')
66        obj = __import__(obj)
67        original_value = getattr(obj, attr)
68        setattr(obj, attr, mock)
69        try:
70            yield
71        finally:
72            setattr(obj, attr, original_value)
73
74
75if sys.version_info >= (3, 2):
76    from unittest import TestCase
77else:
78    import unittest
79
80
81    class TestCase(unittest.TestCase):
82        def assertRegex(self, text, expected_regex, msg=None):
83            # pylint: disable=deprecated-method
84            self.assertRegexpMatches(text, expected_regex, msg)
85
86
87        def assertNotRegex(self, text, unexpected_regex, msg=None):
88            # pylint: disable=deprecated-method
89            self.assertNotRegexpMatches(text, unexpected_regex, msg)
90