fix: Ensure importability of modules not included in __init__.py files (#5965)
* fix: ensure that modules not included in __init__.py files are importable * test: add test for module import in custom component
This commit is contained in:
parent
ea2685fb83
commit
0ef54c5ad9
2 changed files with 27 additions and 1 deletions
|
|
@ -127,6 +127,26 @@ class MyComponent(CustomComponent):
|
|||
assert result.value == "test"
|
||||
|
||||
|
||||
def test_create_class_module_import():
|
||||
code = """
|
||||
from langflow.custom import CustomComponent
|
||||
from PIL import ImageDraw
|
||||
|
||||
class ExternalClass:
|
||||
def __init__(self, value):
|
||||
self.value = value
|
||||
|
||||
class MyComponent(CustomComponent):
|
||||
def build(self):
|
||||
return ExternalClass("test")
|
||||
"""
|
||||
class_name = "MyComponent"
|
||||
created_class = create_class(code, class_name)
|
||||
instance = created_class()
|
||||
result = instance.build()
|
||||
assert result.value == "test"
|
||||
|
||||
|
||||
def test_create_class_with_multiple_external_classes():
|
||||
code = """
|
||||
from langflow.custom import CustomComponent
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue