Skip to content

Commit 1ce19f5

Browse files
committed
TrackImportStack: Fix the unit test to be generic
https://github.com/IBM/import-tracker#27
1 parent 240b3f5 commit 1ce19f5

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

test/test_main.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from contextlib import contextmanager
77
import json
88
import logging
9+
import os
910
import sys
1011

1112
# Third Party
@@ -225,15 +226,22 @@ def test_import_stack_tracking(capsys):
225226
}
226227

227228
# Check one of the stacks to make sure it's correct
229+
test_lib_dir = os.path.realpath(
230+
os.path.join(
231+
os.path.dirname(__file__),
232+
"sample_libs",
233+
"inter_mod_deps",
234+
)
235+
)
228236
assert parsed_out["inter_mod_deps.submod2"] == {
229237
"alog": [
230238
{
231-
"filename": "/Users/ghart/Projects/github/IBM/import-tracker/test/sample_libs/inter_mod_deps/submod1/__init__.py",
239+
"filename": f"{test_lib_dir}/submod1/__init__.py",
232240
"lineno": 6,
233241
"code_context": ["import alog"],
234242
},
235243
{
236-
"filename": "/Users/ghart/Projects/github/IBM/import-tracker/test/sample_libs/inter_mod_deps/__init__.py",
244+
"filename": f"{test_lib_dir}/__init__.py",
237245
"lineno": 17,
238246
"code_context": [
239247
"from . import submod1, submod2, submod3, submod4, submod5"
@@ -242,12 +250,12 @@ def test_import_stack_tracking(capsys):
242250
],
243251
"yaml": [
244252
{
245-
"filename": "/Users/ghart/Projects/github/IBM/import-tracker/test/sample_libs/inter_mod_deps/submod2/__init__.py",
253+
"filename": f"{test_lib_dir}/submod2/__init__.py",
246254
"lineno": 6,
247255
"code_context": ["import yaml"],
248256
},
249257
{
250-
"filename": "/Users/ghart/Projects/github/IBM/import-tracker/test/sample_libs/inter_mod_deps/__init__.py",
258+
"filename": f"{test_lib_dir}/__init__.py",
251259
"lineno": 17,
252260
"code_context": [
253261
"from . import submod1, submod2, submod3, submod4, submod5"

0 commit comments

Comments
 (0)