Add test

Author: armallen

python/tests/sbp/test_table.py

@@ -8,95 +8,105 @@
 # EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED
 # WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A PARTICULAR PURPOSE.
 
-from sbp.msg import SBP
-from sbp.table import _SBP_TABLE, dispatch
+import warnings
+
 from sbp import acquisition as acq
 from sbp import bootload as boot
+from sbp import ext_events as ext_events
 from sbp import file_io as file_io
 from sbp import flash as flash
+from sbp import imu as imu
+from sbp import integrity as integrity
 from sbp import linux as linux
 from sbp import logging as log
+from sbp import mag as mag
 from sbp import navigation as nav
+from sbp import ndb as ndb
 from sbp import observation as obs
+from sbp import orientation as orientation
 from sbp import piksi as piksi
+from sbp import profiling as profiling
+from sbp import sbas as sbas
 from sbp import settings as settings
+from sbp import signing as signing
+from sbp import solution_meta as solmeta
+from sbp import ssr as ssr
 from sbp import system as sys
+from sbp import telemetry as telemetry
 from sbp import tracking as trac
 from sbp import user as user
-from sbp import imu as imu
-from sbp import integrity as integrity
-from sbp import ext_events as ext_events
-from sbp import ndb as ndb
-from sbp import mag as mag
 from sbp import vehicle as vehicle
-from sbp import orientation as orientation
-from sbp import sbas as sbas
-from sbp import ssr as ssr
-from sbp import solution_meta as solmeta
-from sbp import signing as signing
-from sbp import telemetry as telemetry
-from sbp import profiling as profiling
+from sbp.msg import SBP
+from sbp.table import _SBP_TABLE, dispatch
 
-import warnings
 
 def test_table_count():
-  """
-  Test number of available messages to deserialize.
+    """
+    Test number of available messages to deserialize.
+
+    """
+    number_of_messages = 240
+    assert len(_SBP_TABLE) == number_of_messages
+
 
-  """
-  number_of_messages = 239
-  assert len(_SBP_TABLE) == number_of_messages
+def test_table_unique_count():
+    """
+    Test number of messages in packages equals total number of messages.
+    """
+    number_of_messages = (
+        len(acq.msg_classes)
+        + len(boot.msg_classes)
+        + len(file_io.msg_classes)
+        + len(flash.msg_classes)
+        + len(linux.msg_classes)
+        + len(log.msg_classes)
+        + len(nav.msg_classes)
+        + len(obs.msg_classes)
+        + len(piksi.msg_classes)
+        + len(settings.msg_classes)
+        + len(sys.msg_classes)
+        + len(trac.msg_classes)
+        + len(user.msg_classes)
+        + len(imu.msg_classes)
+        + len(integrity.msg_classes)
+        + len(ext_events.msg_classes)
+        + len(ndb.msg_classes)
+        + len(mag.msg_classes)
+        + len(vehicle.msg_classes)
+        + len(orientation.msg_classes)
+        + len(sbas.msg_classes)
+        + len(ssr.msg_classes)
+        + len(solmeta.msg_classes)
+        + len(signing.msg_classes)
+        + len(telemetry.msg_classes)
+        + len(profiling.msg_classes)
+    )
+    assert len(_SBP_TABLE) == number_of_messages
 
-def test_table_unqiue_count():
-  """
-  Test number of messages in packages equals total number of messages.
-  """
-  number_of_messages = (len(acq.msg_classes)
-                        + len(boot.msg_classes)
-                        + len(file_io.msg_classes)
-                        + len(flash.msg_classes)
-                        + len(linux.msg_classes)
-                        + len(log.msg_classes)
-                        + len(nav.msg_classes)
-                        + len(obs.msg_classes)
-                        + len(piksi.msg_classes)
-                        + len(settings.msg_classes)
-                        + len(sys.msg_classes)
-                        + len(trac.msg_classes)
-                        + len(user.msg_classes)
-                        + len(imu.msg_classes)
-                        + len(integrity.msg_classes)
-                        + len(ext_events.msg_classes)
-                        + len(ndb.msg_classes)
-                        + len(mag.msg_classes)
-                        + len(vehicle.msg_classes)
-                        + len(orientation.msg_classes)
-                        + len(sbas.msg_classes)
-                        + len(ssr.msg_classes)
-                        + len(solmeta.msg_classes)
-                        + len(signing.msg_classes)
-                        + len(telemetry.msg_classes)
-                        + len(profiling.msg_classes)
-                         )
-  assert len(_SBP_TABLE) == number_of_messages
 
 def test_available_messages():
-  """
-  Simple example with a limited dispatch table.
+    """
+    Simple example with a limited dispatch table.
 
-  """
-  table = {acq.SBP_MSG_ACQ_RESULT_DEP_A: acq.MsgAcqResultDepA,
-           log.SBP_MSG_PRINT_DEP: log.MsgPrintDep}
-  msg = SBP(msg_type=0x15, sender=1219, length=13,
-            payload=b'\x92$yA\x00\x00\xbcC\x81\xc1\xf9\xc5\x1d')
-  # TODO (Buro): Replace this message constructor once generated SBP
-  # can support kwargs for constructor, instead of requiring SBP
-  # object.
-  assert dispatch(msg, table) == acq.MsgAcqResultDepA(msg)
-  msg = SBP(msg_type=0xB0, sender=1219, length=4, payload='v1.2', crc=0xCE01)
-  with warnings.catch_warnings(record=True) as w:
-    dispatch(msg, table)
-    warnings.simplefilter("always")
-    assert len(w) == 1
-    assert issubclass(w[0].category, RuntimeWarning)
-    assert str(w[0].message).find("No message found for msg_type id 176 for msg*")
+    """
+    table = {
+        acq.SBP_MSG_ACQ_RESULT_DEP_A: acq.MsgAcqResultDepA,
+        log.SBP_MSG_PRINT_DEP: log.MsgPrintDep,
+    }
+    msg = SBP(
+        msg_type=0x15,
+        sender=1219,
+        length=13,
+        payload=b"\x92$yA\x00\x00\xbcC\x81\xc1\xf9\xc5\x1d",
+    )
+    # TODO (Buro): Replace this message constructor once generated SBP
+    # can support kwargs for constructor, instead of requiring SBP
+    # object.
+    assert dispatch(msg, table) == acq.MsgAcqResultDepA(msg)
+    msg = SBP(msg_type=0xB0, sender=1219, length=4, payload="v1.2", crc=0xCE01)
+    with warnings.catch_warnings(record=True) as w:
+        dispatch(msg, table)
+        warnings.simplefilter("always")
+        assert len(w) == 1
+        assert issubclass(w[0].category, RuntimeWarning)
+        assert str(w[0].message).find("No message found for msg_type id 176 for msg*")