/
test_module_paths.py
127 lines (95 loc) · 3.78 KB
/
test_module_paths.py
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
121
122
123
124
125
126
127
# encoding: utf-8
"""Tests for IPython.utils.module_paths.py"""
#-----------------------------------------------------------------------------
# Copyright (C) 2008-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 os
import shutil
import sys
import tempfile
from os.path import join, abspath, split
from IPython.testing.tools import make_tempfile
import IPython.utils.module_paths as mp
import nose.tools as nt
env = os.environ
TEST_FILE_PATH = split(abspath(__file__))[0]
TMP_TEST_DIR = tempfile.mkdtemp()
#
# Setup/teardown functions/decorators
#
old_syspath = sys.path
def make_empty_file(fname):
f = open(fname, 'w')
f.close()
def setup():
"""Setup testenvironment for the module:
"""
# Do not mask exceptions here. In particular, catching WindowsError is a
# problem because that exception is only defined on Windows...
os.makedirs(join(TMP_TEST_DIR, "xmod"))
os.makedirs(join(TMP_TEST_DIR, "nomod"))
make_empty_file(join(TMP_TEST_DIR, "xmod/__init__.py"))
make_empty_file(join(TMP_TEST_DIR, "xmod/sub.py"))
make_empty_file(join(TMP_TEST_DIR, "pack.py"))
make_empty_file(join(TMP_TEST_DIR, "packpyc.pyc"))
sys.path = [TMP_TEST_DIR]
def teardown():
"""Teardown testenvironment for the module:
- Remove tempdir
- restore sys.path
"""
# Note: we remove the parent test dir, which is the root of all test
# subdirs we may have created. Use shutil instead of os.removedirs, so
# that non-empty directories are all recursively removed.
shutil.rmtree(TMP_TEST_DIR)
sys.path = old_syspath
def test_get_init_1():
"""See if get_init can find __init__.py in this testdir"""
with make_tempfile(join(TMP_TEST_DIR, "__init__.py")):
assert mp.get_init(TMP_TEST_DIR)
def test_get_init_2():
"""See if get_init can find __init__.pyw in this testdir"""
with make_tempfile(join(TMP_TEST_DIR, "__init__.pyw")):
assert mp.get_init(TMP_TEST_DIR)
def test_get_init_3():
"""get_init can't find __init__.pyc in this testdir"""
with make_tempfile(join(TMP_TEST_DIR, "__init__.pyc")):
nt.assert_is_none(mp.get_init(TMP_TEST_DIR))
def test_get_init_4():
"""get_init can't find __init__ in empty testdir"""
nt.assert_is_none(mp.get_init(TMP_TEST_DIR))
def test_find_mod_1():
modpath = join(TMP_TEST_DIR, "xmod", "__init__.py")
nt.assert_equal(mp.find_mod("xmod"), modpath)
def test_find_mod_2():
modpath = join(TMP_TEST_DIR, "xmod", "__init__.py")
nt.assert_equal(mp.find_mod("xmod"), modpath)
def test_find_mod_3():
modpath = join(TMP_TEST_DIR, "xmod", "sub.py")
nt.assert_equal(mp.find_mod("xmod.sub"), modpath)
def test_find_mod_4():
modpath = join(TMP_TEST_DIR, "pack.py")
nt.assert_equal(mp.find_mod("pack"), modpath)
def test_find_mod_5():
modpath = join(TMP_TEST_DIR, "packpyc.pyc")
nt.assert_equal(mp.find_mod("packpyc"), modpath)
def test_find_module_1():
modpath = join(TMP_TEST_DIR, "xmod")
nt.assert_equal(mp.find_module("xmod"), modpath)
def test_find_module_2():
"""Testing sys.path that is empty"""
nt.assert_is_none(mp.find_module("xmod", []))
def test_find_module_3():
"""Testing sys.path that is empty"""
nt.assert_is_none(mp.find_module(None, None))
def test_find_module_4():
"""Testing sys.path that is empty"""
nt.assert_is_none(mp.find_module(None))
def test_find_module_5():
nt.assert_is_none(mp.find_module("xmod.nopack"))