Skip to content

Commit 50774bd

Browse files
Update test_search_query.py
1 parent e249f8b commit 50774bd

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

tests/test_query/test_search_query.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -11,22 +11,22 @@
1111
from fise.common import constants
1212
from fise.query import QueryHandler
1313

14-
TEST_DIRECTORY = Path(__file__).parents[1] / "test_directory"
14+
FILE_DIR_TEST_DIRECTORY = Path(__file__).parents[1] / "test_directory/file_dir"
1515

1616

1717
class TestFileSearchQuery:
1818
"""Tests the file search queries."""
1919

2020
basic_query_syntax_test_params = [
21-
f"R SELECT * FROM '{TEST_DIRECTORY / 'file_dir'}'",
22-
f"SELECT[TYPE FILE] name, filetype FROM '{TEST_DIRECTORY / 'file_dir'}'",
23-
f"RECURSIVE SELECT * FROM '{TEST_DIRECTORY}' WHERE filetype = '.py'",
21+
f"R SELECT * FROM '{FILE_DIR_TEST_DIRECTORY}'",
22+
f"SELECT[TYPE FILE] name, filetype FROM '{FILE_DIR_TEST_DIRECTORY}'",
23+
f"RECURSIVE SELECT * FROM '{FILE_DIR_TEST_DIRECTORY}' WHERE filetype = '.py'",
2424
]
2525

2626
mixed_case_query_test_params = [
27-
f"r Select * FroM '{TEST_DIRECTORY / 'file_dir'}'",
28-
f"sELect[TYPE FILE] Name, FileType From '{TEST_DIRECTORY / 'file_dir'}'",
29-
f"RecURSive sELECt * From '{TEST_DIRECTORY}' wHErE FilETypE = '.py'",
27+
f"r Select * FroM '{FILE_DIR_TEST_DIRECTORY}'",
28+
f"sELect[TYPE FILE] Name, FileType From '{FILE_DIR_TEST_DIRECTORY}'",
29+
f"RecURSive sELECt * From '{FILE_DIR_TEST_DIRECTORY}' wHErE FilETypE = '.py'",
3030
]
3131

3232
individual_fields_test_params = constants.FILE_FIELDS + ("*",)
@@ -45,7 +45,7 @@ def test_basic_query_syntax(self, query: str) -> None:
4545
def test_individual_fields(self, field: str) -> None:
4646
"""Tests file search queries with all file fields individually."""
4747

48-
query: str = f"SELECT {field} FROM '{TEST_DIRECTORY}'"
48+
query: str = f"SELECT {field} FROM '{FILE_DIR_TEST_DIRECTORY}'"
4949

5050
data: pd.DataFrame = QueryHandler(query).handle()
5151
assert isinstance(data, pd.DataFrame)
@@ -66,15 +66,15 @@ class TestDirSearchQuery:
6666
"""Tests the directory search queries."""
6767

6868
basic_query_syntax_test_params = [
69-
f"R SELECT[TYPE DIR] * FROM '{TEST_DIRECTORY / 'file_dir'}'",
70-
f"SELECT[TYPE DIR] name, parent, ctime FROM '{TEST_DIRECTORY / 'file_dir'}'",
71-
f"RECURSIVE SELECT[TYPE DIR] * FROM '{TEST_DIRECTORY}' WHERE name IN ('orders', 'reports')",
69+
f"R SELECT[TYPE DIR] * FROM '{FILE_DIR_TEST_DIRECTORY}'",
70+
f"RECURSIVE SELECT[TYPE DIR] name, parent, ctime FROM '{FILE_DIR_TEST_DIRECTORY}'",
71+
f"SELECT[TYPE DIR] * FROM '{FILE_DIR_TEST_DIRECTORY}' WHERE name IN ('orders', 'reports')",
7272
]
7373

7474
mixed_case_query_test_params = [
75-
f"r SELECT[Type DiR] * fROm '{TEST_DIRECTORY / 'file_dir'}'",
76-
f"sEleCt[typE dIr] name, parent, ctime FroM '{TEST_DIRECTORY / 'file_dir'}'",
77-
f"Recursive Select[TYPE DIR] * From '{TEST_DIRECTORY}' Where name In ('orders', 'reports')",
75+
f"r SELECT[Type DiR] * fROm '{FILE_DIR_TEST_DIRECTORY}'",
76+
f"Recursive sEleCt[typE dIr] name, parent, ctime FroM '{FILE_DIR_TEST_DIRECTORY}'",
77+
f"Select[TYPE DIR] * From '{FILE_DIR_TEST_DIRECTORY}' Where name In ('orders', 'reports')",
7878
]
7979

8080
individual_fields_test_params = constants.DIR_FIELDS + ("*",)
@@ -93,7 +93,7 @@ def test_basic_query_syntax(self, query: str) -> None:
9393
def test_individual_fields(self, field: str) -> None:
9494
"""Tests individual fields in directory search queries."""
9595

96-
query: str = f"SELECT[TYPE DIR] {field} FROM '{TEST_DIRECTORY}'"
96+
query: str = f"SELECT[TYPE DIR] {field} FROM '{FILE_DIR_TEST_DIRECTORY}'"
9797

9898
data: pd.DataFrame = QueryHandler(query).handle()
9999
assert isinstance(data, pd.DataFrame)

0 commit comments

Comments
 (0)