"""Tests for the pb_utils module.""" from auvsi_suas.models import pb_utils from auvsi_suas.proto import interop_api_pb2 from django.test import TestCase def TestFieldChociesFromEnum(TestCase): """Tests the FieldChoicesFromEnum utility.""" def test_generates_choices(self): """Tests generates choices for a sample enum.""" self.assertEqual([(1, 'N'), (2, 'NE'), (3, 'E'), (4, 'SE'), (5, 'S'), (6, 'SW'), (7, 'W'), (8, 'NW')], sorted( pb_utils.FieldChoicesFromEnum( interop_api_pb2.Odlc.Orientation)))