Coverage for scripts/tests/test_training.py: 100%

1from scripts.training import train_objects_model, train_seame_model 

2from unittest.mock import patch 

3 

4def test_train_objects_model(): 

5 with patch("scripts.training.YOLO") as MockYOLO: 

6 instance = MockYOLO.return_value 

7 instance.train.return_value = "train_success" 

8 result = train_objects_model() 

9 instance.train.assert_called_once() 

10 assert result == "train_success" 

11 

12def test_train_seame_model(): 

13 with patch("scripts.training.YOLO") as MockYOLO: 

14 instance = MockYOLO.return_value 

15 instance.train.return_value = "train_success" 

16 result = train_seame_model() 

17 instance.train.assert_called_once() 

18 assert result == "train_success"