Add more tests

Airthings · Feb 5, 2024 · 24e5b44 · 24e5b44
1 parent ba81913
commit 24e5b44
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_device_type.py b/tests/test_device_type.py
@@ -16,6 +16,11 @@ def test_device_type():
     with pytest.raises(ValueError):
         AirthingsDeviceType("1234")
 
+    assert AirthingsDeviceType.from_raw_value("2900") == AirthingsDeviceType.WAVE_GEN_1
+    assert AirthingsDeviceType.from_raw_value("2920") == AirthingsDeviceType.WAVE_MINI
+    assert AirthingsDeviceType.from_raw_value("2930") == AirthingsDeviceType.WAVE_PLUS
+    assert AirthingsDeviceType.from_raw_value("2950") == AirthingsDeviceType.WAVE_RADON
+
     unknown_device = AirthingsDeviceType.from_raw_value("1234")
     assert unknown_device == AirthingsDeviceType.UNKNOWN
     assert unknown_device.product_name == "Unknown"