From edbeab6c4e3a6950c96f82a352b2d9d6bfe74c19 Mon Sep 17 00:00:00 2001 From: Joost Lekkerkerker Date: Sat, 10 Aug 2024 12:53:43 +0200 Subject: [PATCH] Fix O-1PST not having model name (#173) --- src/airgradient/util.py | 2 +- tests/test_util.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/airgradient/util.py b/src/airgradient/util.py index aa8ae99..a226b82 100644 --- a/src/airgradient/util.py +++ b/src/airgradient/util.py @@ -7,7 +7,7 @@ def get_model_name(model_id: str) -> str | None: """Get model name from identifier.""" if model_id.startswith("I-9PSL"): return "AirGradient ONE" - if model_id.startswith("O-1PPT"): + if model_id.startswith("O-1"): return "AirGradient Open Air" if "DIY" in model_id: return "AirGradient DIY" diff --git a/tests/test_util.py b/tests/test_util.py index 14367a8..6cdd91b 100644 --- a/tests/test_util.py +++ b/tests/test_util.py @@ -13,6 +13,7 @@ ("I-9PSL", "AirGradient ONE"), ("I-9PSL-DE", "AirGradient ONE"), ("O-1PPT", "AirGradient Open Air"), + ("O-1PST", "AirGradient Open Air"), ("DIY-PRO-4.3", "AirGradient DIY"), ("DIY-PRO-3.7", "AirGradient DIY"), ("DIY-BASIC-4.0", "AirGradient DIY"),