From 3f93be488638fdf6668e0e03e2b1634bb969ca80 Mon Sep 17 00:00:00 2001 From: Matthias Baumgartner Date: Mon, 16 Jan 2023 15:39:16 +0100 Subject: random fixes --- test/reader/image/test_image.py | 5 ++++- test/reader/image/test_raw_image.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) (limited to 'test') diff --git a/test/reader/image/test_image.py b/test/reader/image/test_image.py index c60ca75..26f6a93 100644 --- a/test/reader/image/test_image.py +++ b/test/reader/image/test_image.py @@ -20,7 +20,10 @@ from bsie.reader.image import Image class TestImage(unittest.TestCase): def setUp(self): - importlib.import_module(__package__ + '.load_nef').get() + if __package__ is None or __package__ == '': # direct call or local discovery + importlib.import_module('load_nef', __package__).get() + else: # parent discovery + importlib.import_module('.load_nef', __package__).get() def test_construct(self): image = Image({}) diff --git a/test/reader/image/test_raw_image.py b/test/reader/image/test_raw_image.py index 3d5f887..ba21b5a 100644 --- a/test/reader/image/test_raw_image.py +++ b/test/reader/image/test_raw_image.py @@ -23,7 +23,10 @@ from bsie.reader.image._raw import RawImage class TestRawImage(unittest.TestCase): def setUp(self): - importlib.import_module(__package__ + '.load_nef').get() + if __package__ is None or __package__ == '': # direct call or local discovery + importlib.import_module('load_nef', __package__).get() + else: # parent discovery + importlib.import_module('.load_nef', __package__).get() def test_call(self): rdr = RawImage() -- cgit v1.2.3