ogn-python/tests/test_ognutils.py

30 wiersze
994 B
Python

import unittest
from ogn.ognutils import get_ddb, get_country_code
from ogn.model.address_origin import AddressOrigin
class TestStringMethods(unittest.TestCase):
def test_get_devices(self):
devices = get_ddb()
self.assertGreater(len(devices), 1000)
def test_get_ddb_from_file(self):
devices = get_ddb('tests/custom_ddb.txt')
self.assertEqual(len(devices), 3)
device = devices[0]
self.assertEqual(device.address, 'DD4711')
self.assertEqual(device.aircraft, 'HK36 TTC')
self.assertEqual(device.registration, 'D-EULE')
self.assertEqual(device.competition, '')
self.assertTrue(device.tracked)
self.assertTrue(device.identified)
self.assertEqual(device.address_origin, AddressOrigin.userdefined)
def test_get_country_code(self):
latitude = 48.0
longitude = 11.0
country_code = get_country_code(latitude, longitude)
self.assertEquals(country_code, 'de')