From 6d8c1eb7613f9f61ac15b3616b955cb1f39580eb Mon Sep 17 00:00:00 2001 From: Adam Byrne Date: Mon, 29 Jan 2024 14:53:31 +0000 Subject: [PATCH] fixed relative package issue --- agrinet/tests/testDataLoader.py | 12 +++++++++--- agrinet/tests/testLogManager.py | 8 +++++++- 2 files changed, 16 insertions(+), 4 deletions(-) diff --git a/agrinet/tests/testDataLoader.py b/agrinet/tests/testDataLoader.py index f595f32..7ee5c41 100644 --- a/agrinet/tests/testDataLoader.py +++ b/agrinet/tests/testDataLoader.py @@ -1,5 +1,11 @@ -import unittest +import sys +import os + +# fixes "ModuleNotFoundError: No module named 'utils'" +sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) +# flake8: noqa +import unittest import tensorflow as tf from utils.DataLoader import load_image_test, load_image_train, resize @@ -15,7 +21,7 @@ def tearDown(self): def test_load_image_train(self): # Condition: input_image.shape != real_image.shape from the train dataset - image_file = "../assets/green-field.jpg" + image_file = "assets/green-field.jpg" input_image, real_image = load_image_train(image_file) self.assertIsInstance(input_image, tf.Tensor) @@ -25,7 +31,7 @@ def test_load_image_train(self): def test_load_image_test(self): # Condition: input_image.shape == real_image.shape from the test dataset - image_file = "../assets/green-field.jpg" + image_file = "assets/green-field.jpg" input_image, real_image = load_image_test(image_file) diff --git a/agrinet/tests/testLogManager.py b/agrinet/tests/testLogManager.py index b8e2fa1..8031b68 100644 --- a/agrinet/tests/testLogManager.py +++ b/agrinet/tests/testLogManager.py @@ -1,7 +1,13 @@ +import sys +import os + +# fixes "ModuleNotFoundError: No module named 'utils'" +sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) + +# flake8: noqa import unittest from io import StringIO from unittest.mock import patch - from utils.LogManager import LogManager