Coverage for tests / unit / ai / test_models.py: 100%

1"""Tests for AI data models.""" 

2 

3from __future__ import annotations 

4 

5from assertpy import assert_that 

6 

7from lintro.ai.models import AIFixSuggestion 

8 

9 

10def test_models_fix_suggestion_defaults(): 

11 """All AIFixSuggestion fields have expected defaults.""" 

12 fix = AIFixSuggestion() 

13 assert_that(fix.file).is_equal_to("") 

14 assert_that(fix.line).is_equal_to(0) 

15 assert_that(fix.code).is_equal_to("") 

16 assert_that(fix.tool_name).is_equal_to("") 

17 assert_that(fix.original_code).is_equal_to("") 

18 assert_that(fix.suggested_code).is_equal_to("") 

19 assert_that(fix.diff).is_equal_to("") 

20 assert_that(fix.explanation).is_equal_to("") 

21 assert_that(fix.confidence).is_equal_to("medium") 

22 assert_that(fix.risk_level).is_equal_to("") 

23 assert_that(fix.input_tokens).is_equal_to(0) 

24 assert_that(fix.output_tokens).is_equal_to(0) 

25 assert_that(fix.cost_estimate).is_equal_to(0.0)