diff --git a/tests/test_date_filter.py b/tests/test_date_filter.py index 59ef697c..bc656701 100644 --- a/tests/test_date_filter.py +++ b/tests/test_date_filter.py @@ -3,9 +3,6 @@ import re from datetime import datetime from math import inf -# External Packages -import torch - # Application Packages from src.search_filter.date_filter import DateFilter from src.utils.rawconfig import Entry diff --git a/tests/test_file_filter.py b/tests/test_file_filter.py index e6c17299..28b0367f 100644 --- a/tests/test_file_filter.py +++ b/tests/test_file_filter.py @@ -1,6 +1,3 @@ -# External Packages -import torch - # Application Packages from src.search_filter.file_filter import FileFilter from src.utils.rawconfig import Entry @@ -9,7 +6,7 @@ from src.utils.rawconfig import Entry def test_no_file_filter(): # Arrange file_filter = FileFilter() - embeddings, entries = arrange_content() + entries = arrange_content() q_with_no_filter = 'head tail' # Act @@ -25,7 +22,7 @@ def test_no_file_filter(): def test_file_filter_with_non_existent_file(): # Arrange file_filter = FileFilter() - embeddings, entries = arrange_content() + entries = arrange_content() q_with_no_filter = 'head file:"nonexistent.org" tail' # Act @@ -41,7 +38,7 @@ def test_file_filter_with_non_existent_file(): def test_single_file_filter(): # Arrange file_filter = FileFilter() - embeddings, entries = arrange_content() + entries = arrange_content() q_with_no_filter = 'head file:"file 1.org" tail' # Act @@ -57,7 +54,7 @@ def test_single_file_filter(): def test_file_filter_with_partial_match(): # Arrange file_filter = FileFilter() - embeddings, entries = arrange_content() + entries = arrange_content() q_with_no_filter = 'head file:"1.org" tail' # Act @@ -73,7 +70,7 @@ def test_file_filter_with_partial_match(): def test_file_filter_with_regex_match(): # Arrange file_filter = FileFilter() - embeddings, entries = arrange_content() + entries = arrange_content() q_with_no_filter = 'head file:"*.org" tail' # Act @@ -89,7 +86,7 @@ def test_file_filter_with_regex_match(): def test_multiple_file_filter(): # Arrange file_filter = FileFilter() - embeddings, entries = arrange_content() + entries = arrange_content() q_with_no_filter = 'head tail file:"file 1.org" file:"file2.org"' # Act @@ -103,7 +100,6 @@ def test_multiple_file_filter(): def arrange_content(): - embeddings = torch.randn(4, 10) entries = [ Entry(compiled='', raw='First Entry', file= 'file 1.org'), Entry(compiled='', raw='Second Entry', file= 'file2.org'), diff --git a/tests/test_image_search.py b/tests/test_image_search.py index 97911164..374168f5 100644 --- a/tests/test_image_search.py +++ b/tests/test_image_search.py @@ -2,9 +2,6 @@ from pathlib import Path from PIL import Image -# External Packages -import pytest - # Internal Packages from src.utils.state import model from src.utils.constants import web_directory diff --git a/tests/test_word_filter.py b/tests/test_word_filter.py index 58069b24..2e662fd0 100644 --- a/tests/test_word_filter.py +++ b/tests/test_word_filter.py @@ -1,6 +1,5 @@ # Application Packages from src.search_filter.word_filter import WordFilter -from src.utils.config import SearchType from src.utils.rawconfig import Entry