mirror of
https://github.com/OpenSPG/openspg.git
synced 2025-08-02 22:12:38 +00:00
35 lines
641 B
Python
35 lines
641 B
Python
|
|
import unittest
|
|
|
|
from knext.component.builder import UserDefinedExtractor
|
|
|
|
|
|
class TestUserDefinedExtractor(unittest.TestCase):
|
|
"""UserDefinedExtractor unit test stubs"""
|
|
|
|
def setUp(self):
|
|
self.component = UserDefinedExtractor()
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def testExecute(self):
|
|
"""Test execute"""
|
|
|
|
|
|
class TestLLMBasedExtractor(unittest.TestCase):
|
|
"""LLMBasedExtractor unit test stubs"""
|
|
|
|
def setUp(self):
|
|
self.component = LLMBasedExtractor()
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def testExecute(self):
|
|
"""Test execute"""
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|