import unittest from src.graph_builder import build_graph class TestGraph(unittest.TestCase): def test_graph_runs(self): graph = build_graph() self.assertIsNotNone(graph) if __name__ == "__main__": unittest.main()