24 lines
838 B
Python
24 lines
838 B
Python
import os
|
|
import tempfile
|
|
|
|
import folder_paths
|
|
|
|
|
|
def test_detection_folder_allows_onnx_extension():
|
|
_, extensions = folder_paths.folder_names_and_paths["detection"]
|
|
assert ".onnx" in extensions
|
|
|
|
|
|
def test_detection_folder_lists_onnx_files():
|
|
with tempfile.TemporaryDirectory() as detection_dir:
|
|
onnx_path = os.path.join(detection_dir, "vitpose-l-wholebody.onnx")
|
|
with open(onnx_path, "wb") as onnx_file:
|
|
onnx_file.write(b"onnx")
|
|
|
|
folder_paths.add_model_folder_path("detection", detection_dir, is_default=True)
|
|
folder_paths.filename_list_cache.pop("detection", None)
|
|
|
|
filenames = folder_paths.get_filename_list("detection")
|
|
assert "vitpose-l-wholebody.onnx" in filenames
|
|
assert folder_paths.get_full_path("detection", "vitpose-l-wholebody.onnx") == onnx_path
|