https://docs.python.org/3/library/unittest.html
非常简单:
class:
class VisionTool: def __init__(self): pass def get_current_player_cards(self, source_image_name):
pass
对应的测试文件
import unittest from vision_tool import VisionTool class TestVisionTool(unittest.TestCase): def setUp(self): self.vision_tool = VisionTool() def test_get_current_player_cards(self): # cards 1 current_player_cards = self.vision_tool.get_current_player_cards('test/test_current_player_cards_1.png') expected_cards = [ 'A_spade', 'A_heart', 'A_diamond', 'K_heart', 'Q_heart', 'Q_club', 'Q_diamond', 'J_diamond', 'T_spade', 'T_heart', 'T_diamond', '9_club', '8_spade', '7_spade', '7_heart', '7_club', '6_spade', '5_spade', '4_club', '3_spade' ] expected_cards.sort() self.assertEqual(expected_cards, current_player_cards) if __name__ == '__main__': unittest.main()
只运行某个单元测试;
python TestMyCase.py TestMyCase.testcase1