import sys import os import unittest # Add the project root directory to the Python path to resolve the ModuleNotFoundError. sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) from pytrace.core.node_base import TraceNode class TestNodeA(TraceNode): """A dummy node for testing purposes.""" CATEGORY = "Test" def process(self, **kwargs): # A simple process method return {'output': 1} class NodeLogicTests(unittest.TestCase): def test_node_creation(self): """Tests that a basic node can be instantiated.""" node = TestNodeA(uid="test_01", name="Test Node") self.assertIsInstance(node, TraceNode) self.assertEqual(node.uid, "test_01") def test_node_process(self): """Tests the dummy node's process method.""" node = TestNodeA(uid="test_02", name="Test Node 2") result = node.process() self.assertIn('output', result) self.assertEqual(result['output'], 1) if __name__ == '__main__': unittest.main()