Update test_entity.py

mattrayner · Sep 16, 2024 · 64fc91f · 64fc91f
1 parent f10af4b
commit 64fc91f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_entity.py b/tests/test_entity.py
@@ -313,10 +313,14 @@ async def test_pod_point_entity(hass, bypass_get_data):
     assert "/api/pod_point/static/uc-03.png" == entity.image
     assert True is entity.connected
 
-
+    entity.pod.model.name = "S7-UC-05-ACA"
     assert "S7-UC-05-ACA" == entity.model
     assert "/api/pod_point/static/uc-05.png" == entity.image
 
+    entity.pod.model.name = "S7-1C-05-ACA"
+    assert "S7-1C-05-ACA" == entity.model
+    assert "/api/pod_point/static/uc-05.png" == entity.image
+
     entity.pod.model.name = "XX-UP-XX-XX"
     assert "/api/pod_point/static/uc.png" == entity.image