test_tz.py 762 Bytes
Newer Older
1 2 3 4
# -*- coding: utf-8 -*-
import unittest

import pygeoip
William Tisäter committed
5
import pygeoip.timezone
6 7 8 9 10
from tests.config import CITY_DB_PATH


class TestGeoIPTimeZoneFunctions(unittest.TestCase):
    def setUp(self):
11
        self.gi = pygeoip.GeoIP(CITY_DB_PATH)
12

William Tisäter committed
13 14 15 16 17 18 19 20 21 22
    def testTimeZoneInternals(self):
        tz = pygeoip.timezone.time_zone_by_country_and_region('XX')
        self.assertEquals(tz, None)

        tz = pygeoip.timezone.time_zone_by_country_and_region('US', 'NY')
        self.assertEquals(tz, 'America/New_York')

        tz = pygeoip.timezone.time_zone_by_country_and_region('US', 'XX')
        self.assertEquals(tz, None)

23
    def testTimeZoneByAddr(self):
24 25
        tz = self.gi.time_zone_by_addr('64.17.254.216')
        self.assertEquals(tz, 'America/Los_Angeles')