IQ.Pilot Prebuilt Release @ ab07000

iqpilot/tests/test_auto_units.py

@@ -0,0 +1,91 @@
+from openpilot.common.params import Params
+from openpilot.iqpilot.common.auto_units import CONFIRMATIONS, AutoUnits
+
+SEATTLE = (47.6062, -122.3321)
+BERLIN = (52.5200, 13.4050)
+LONDON = (51.5074, -0.1278)
+
+
+class StubAutoUnits(AutoUnits):
+  def __init__(self, params):
+    super().__init__(params)
+    self.position = (0.0, 0.0, False)
+
+  def _position(self):
+    return self.position
+
+
+def settle(auto_units, position, count=CONFIRMATIONS, start=0.0):
+  auto_units.position = position
+  for i in range(count):
+    auto_units.update(now=start + i * 100.0)
+  return start + count * 100.0
+
+
+class TestAutoUnits:
+  def setup_method(self):
+    self.params = Params()
+    self.params.put_bool("IQAutoUnits", True)
+    self.params.remove("IQAutoUnitsRegion")
+    self.params.put_bool("IsMetric", False)
+    self.auto_units = StubAutoUnits(self.params)
+
+  def test_no_fix_does_nothing(self):
+    settle(self.auto_units, (0.0, 0.0, False))
+    assert self.params.get("IQAutoUnitsRegion") is None
+    assert not self.params.get_bool("IsMetric")
+
+  def test_disabled_does_nothing(self):
+    self.params.put_bool("IQAutoUnits", False)
+    settle(self.auto_units, (*BERLIN, True))
+    assert self.params.get("IQAutoUnitsRegion") is None
+    assert not self.params.get_bool("IsMetric")
+
+  def test_metric_region_switches_to_metric(self):
+    settle(self.auto_units, (*BERLIN, True))
+    assert self.params.get("IQAutoUnitsRegion") == "METRIC"
+    assert self.params.get_bool("IsMetric")
+
+  def test_mph_region_stays_imperial(self):
+    settle(self.auto_units, (*SEATTLE, True))
+    assert self.params.get("IQAutoUnitsRegion") == "US"
+    assert not self.params.get_bool("IsMetric")
+
+  def test_uk_stays_imperial(self):
+    self.params.put_bool("IsMetric", True)
+    settle(self.auto_units, (*LONDON, True))
+    assert self.params.get("IQAutoUnitsRegion") == "GB"
+    assert not self.params.get_bool("IsMetric")
+
+  def test_border_crossing_switches_units(self):
+    now = settle(self.auto_units, (*SEATTLE, True))
+    assert not self.params.get_bool("IsMetric")
+
+    settle(self.auto_units, (*BERLIN, True), start=now)
+    assert self.params.get("IQAutoUnitsRegion") == "METRIC"
+    assert self.params.get_bool("IsMetric")
+
+  def test_manual_override_is_kept_within_a_region(self):
+    now = settle(self.auto_units, (*SEATTLE, True))
+    self.params.put_bool("IsMetric", True)
+
+    settle(self.auto_units, (*SEATTLE, True), start=now)
+    assert self.params.get_bool("IsMetric")
+
+  def test_unconfirmed_region_is_not_applied(self):
+    settle(self.auto_units, (*BERLIN, True), count=CONFIRMATIONS - 1)
+    assert self.params.get("IQAutoUnitsRegion") is None
+    assert not self.params.get_bool("IsMetric")
+
+  def test_flapping_region_resets_confirmations(self):
+    now = 0.0
+    for position in (BERLIN, SEATTLE, BERLIN, SEATTLE):
+      now = settle(self.auto_units, (*position, True), count=1, start=now)
+    assert self.params.get("IQAutoUnitsRegion") is None
+    assert not self.params.get_bool("IsMetric")
+
+  def test_rate_limited(self):
+    self.auto_units.position = (*BERLIN, True)
+    for _ in range(CONFIRMATIONS * 4):
+      self.auto_units.update(now=1.0)
+    assert self.params.get("IQAutoUnitsRegion") is None