Coverage for app/tests/test_chart.py: 100%

1from app.chart import generate_altair_colors 

2 

3 

4def test_generate_altair_colors_first_three() -> None: 

5 result = generate_altair_colors(3) 

6 expected_result = ["#346f9e", "#ffde56", "#2e7d32"] 

7 assert result == expected_result 

8 

9 

10def test_generate_altair_colors_more_than_three() -> None: 

11 result = len(generate_altair_colors(4)) 

12 expected_result = 4 

13 assert result == expected_result