mirror of
https://github.com/run-llama/fs-explorer.git
synced 2026-07-01 21:45:00 -04:00
Clelia (Astra) Bertelli
75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
import pytest
|
|
import os
|
|
|
|
from fs_explorer.fs import (
|
|
describe_dir_content,
|
|
read_file,
|
|
grep_file_content,
|
|
glob_paths,
|
|
parse_file,
|
|
)
|
|
|
|
|
|
def test_describe_dir_content() -> None:
|
|
description = describe_dir_content("tests/testfiles")
|
|
assert (
|
|
description
|
|
== "Content of tests/testfiles\nFILES:\n- tests/testfiles/file1.txt\n- tests/testfiles/file2.md\nSUBFOLDERS:\n- tests/testfiles/last"
|
|
)
|
|
description = describe_dir_content("tests/testfile")
|
|
assert description == "No such directory: tests/testfile"
|
|
description = describe_dir_content("tests/testfiles/last")
|
|
assert (
|
|
description
|
|
== "Content of tests/testfiles/last\nFILES:\n- tests/testfiles/last/lastfile.txt\nThis folder does not have any sub-folders"
|
|
)
|
|
|
|
|
|
def test_read_file() -> None:
|
|
content = read_file("tests/testfiles/file1.txt")
|
|
assert content.strip() == "this is a test"
|
|
content = read_file("tests/testfiles/file2.txt")
|
|
assert content.strip() == "No such file: tests/testfiles/file2.txt"
|
|
|
|
|
|
def test_grep_file_content() -> None:
|
|
result = grep_file_content("tests/testfiles/file2.md", r"(are|is) a test")
|
|
assert result == "MATCHES for (are|is) a test in tests/testfiles/file2.md:\n\n- is"
|
|
result = grep_file_content("tests/testfiles/last/lastfile.txt", r"test")
|
|
assert result == "No matches found"
|
|
result = grep_file_content("tests/testfiles/file2.txt", r"test")
|
|
assert result == "No such file: tests/testfiles/file2.txt"
|
|
|
|
|
|
def test_glob_paths() -> None:
|
|
result = glob_paths("tests/testfiles", "file?.*")
|
|
assert (
|
|
result
|
|
== "MATCHES for file?.* in tests/testfiles:\n\n- ./tests/testfiles/file1.txt\n- ./tests/testfiles/file2.md"
|
|
)
|
|
result = glob_paths("tests/testfiles", "test*")
|
|
assert result == "No matches found"
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
condition=(os.getenv("LLAMA_CLOUD_API_KEY") is None),
|
|
reason="LLAMA_CLOUD_API_KEY is not available",
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_parse_file() -> None:
|
|
content = await parse_file("data/testfile.txt")
|
|
assert content.strip() == "This is a test."
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_parse_file_without_api_key() -> None:
|
|
# file does not exist
|
|
content = await parse_file("data/test.txt")
|
|
assert content == "No such file: data/test.txt"
|
|
# api key not set
|
|
content = await parse_file("data/testfile.txt")
|
|
assert (
|
|
content
|
|
== "Not possible to parse data/testfile.txt because it has not been cached and the necessary credentials (`LLAMA_CLOUD_API_KEY`) are not set in the environment"
|
|
)
|