57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
import math
|
|
from src.theme_nerv_fx import ScanlineOverlay, StatusFlicker
|
|
|
|
class TestThemeNervFx(unittest.TestCase):
|
|
@patch("src.theme_nerv_fx.imgui")
|
|
def test_scanline_overlay_render(self, mock_imgui):
|
|
# Setup
|
|
mock_draw_list = MagicMock()
|
|
mock_imgui.get_foreground_draw_list.return_value = mock_draw_list
|
|
mock_imgui.get_color_u32.return_value = 0x12345678
|
|
|
|
overlay = ScanlineOverlay()
|
|
width, height = 100.0, 10.0
|
|
|
|
# Act
|
|
overlay.render(width, height)
|
|
|
|
# Assert
|
|
mock_imgui.get_foreground_draw_list.assert_called_once()
|
|
# height is 10, range(0, 10, 2) is [0, 2, 4, 6, 8] -> 5 calls
|
|
self.assertEqual(mock_draw_list.add_line.call_count, 5)
|
|
|
|
# Verify some calls
|
|
mock_draw_list.add_line.assert_any_call((0.0, 0.0), (100.0, 0.0), 0x12345678, 1.0)
|
|
mock_draw_list.add_line.assert_any_call((0.0, 8.0), (100.0, 8.0), 0x12345678, 1.0)
|
|
|
|
@patch("src.theme_nerv_fx.time")
|
|
def test_status_flicker_get_alpha(self, mock_time):
|
|
flicker = StatusFlicker()
|
|
|
|
# Test at multiple points in time to ensure range [0.7, 1.0]
|
|
# sin(0) = 0 -> 0.85
|
|
mock_time.time.return_value = 0.0
|
|
self.assertAlmostEqual(flicker.get_alpha(), 0.85)
|
|
|
|
# sin(pi/2) = 1 -> 1.0
|
|
# time * 20.0 = pi/2 -> time = pi/40
|
|
mock_time.time.return_value = math.pi / 40.0
|
|
self.assertAlmostEqual(flicker.get_alpha(), 1.0)
|
|
|
|
# sin(3pi/2) = -1 -> 0.7
|
|
# time * 20.0 = 3pi/2 -> time = 3pi/40
|
|
mock_time.time.return_value = 3 * math.pi / 40.0
|
|
self.assertAlmostEqual(flicker.get_alpha(), 0.7)
|
|
|
|
# Verify range for many samples
|
|
for i in range(100):
|
|
mock_time.time.return_value = i * 0.1
|
|
alpha = flicker.get_alpha()
|
|
self.assertGreaterEqual(alpha, 0.7)
|
|
self.assertLessEqual(alpha, 1.0)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|