65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
|
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()
|