jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor directory_component.build() and add tests for loading projects and mdx files

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-08 17:06:53 -03:00
commit ecdc0be825
1 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/test_data_components.py
View file

@ -153,7 +153,17 @@ def test_directory_without_mocks():
# the directory component can be used to load the projects # the directory component can be used to load the projects
# and we can validate if the contents are the same as the projects variable # and we can validate if the contents are the same as the projects variable
setup_path = Path(setup.__file__).parent / "starter_projects" setup_path = Path(setup.__file__).parent / "starter_projects"
result = directory_component.build( results = directory_component.build(str(setup_path), use_multithreading=False)
str(setup_path), types=["json"], use_multithreading=False assert len(results) == len(projects)
) # each result is a Record that contains the content attribute
assert len(result) == len(projects) # each are dict that are exactly the same as one of the projects
for result in results:
assert result.text in projects
# in ../docs/docs/components there are many mdx files
# check if the directory component can load them
# just check if the number of results is the same as the number of files
docs_path = Path(__file__).parent.parent / "docs" / "docs" / "components"
results = directory_component.build(str(docs_path), use_multithreading=False)
docs_files = list(docs_path.glob("*.mdx"))
assert len(results) == len(docs_files)