import os import sys # 添加项目根目录到Python路径 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..')) import unittest from unittest.mock import MagicMock from vision.camera import CameraController # from core.vision import Vision class TestVision(unittest.TestCase): def setUp(self): self.jj=2 # self.testclass2 = MagicMock() # self.testclass = TestClass(self.testclass2) def test_capture_frame(self): # 测试capture_frame方法 camera=CameraController() result = camera.capture_frame() self.assertIsNone(result, msg="capture_frame方法测试失败") camera.capture_frame_exec.assert_called_once() # def test_first(self): # 测试TestClass的add方法 # mock_testclass2 = MagicMock() # # mock_testclass2.i = 1 # # mock_testclass2.j = 2 # test_class = TestClass(mock_testclass2) # result = test_class.add() # # 验证结果 # self.assertEqual(result, 3, msg="add方法测试失败") # def test_second(self): # 测试TestClass2的mock行为 # mock_testclass2 = MagicMock(spec=TestClass2) # mock_testclass2.sub.return_value = 1 # result = mock_testclass2.sub() # # 测试返回值 # self.assertEqual(result, 1, msg="sub方法测试失败") # mock_testclass2.sub.assert_called_once() class TestClass: def __init__(self,testclass2): self.testclass2 = testclass2 pass def add(self): return self.testclass2.i + self.testclass2.j class TestClass2: def __init__(self): self.i = 1 self.j = 2 pass def sub(self): return self.j - self.i if __name__ == '__main__': unittest.main()