~ubuntu-branches/ubuntu/natty/python3.2/natty-updates

from test import support

import time

import unittest

import locale

import sysconfig

import warnings

class TimeTestCase(unittest.TestCase):

    def setUp(self):

        self.t = time.time()

    def test_data_attributes(self):

        time.altzone

        time.daylight

        time.timezone

        time.tzname

    def test_clock(self):

        time.clock()

    def test_conversions(self):

        self.assertEqual(time.ctime(self.t),

                         time.asctime(time.localtime(self.t)))

        self.assertEqual(int(time.mktime(time.localtime(self.t))),

                         int(self.t))

    def test_sleep(self):

        time.sleep(1.2)

    def test_strftime(self):

        tt = time.gmtime(self.t)

        for directive in ('a', 'A', 'b', 'B', 'c', 'd', 'H', 'I',

                          'j', 'm', 'M', 'p', 'S',

                          'U', 'w', 'W', 'x', 'X', 'y', 'Y', 'Z', '%'):

            format = ' %' + directive

            try:

                time.strftime(format, tt)

            except ValueError:

                self.fail('conversion specifier: %r failed.' % format)

    def _bounds_checking(self, func=time.strftime):

        # Make sure that strftime() checks the bounds of the various parts

        #of the time tuple (0 is valid for *all* values).

        # The year field is tested by other test cases above

        # Check month [1, 12] + zero support

        self.assertRaises(ValueError, func,

                            (1900, -1, 1, 0, 0, 0, 0, 1, -1))

        self.assertRaises(ValueError, func,

                            (1900, 13, 1, 0, 0, 0, 0, 1, -1))

        # Check day of month [1, 31] + zero support

        self.assertRaises(ValueError, func,

                            (1900, 1, -1, 0, 0, 0, 0, 1, -1))

        self.assertRaises(ValueError, func,

                            (1900, 1, 32, 0, 0, 0, 0, 1, -1))

        # Check hour [0, 23]

        self.assertRaises(ValueError, func,

                            (1900, 1, 1, -1, 0, 0, 0, 1, -1))

        self.assertRaises(ValueError, func,

                            (1900, 1, 1, 24, 0, 0, 0, 1, -1))

        # Check minute [0, 59]

        self.assertRaises(ValueError, func,

                            (1900, 1, 1, 0, -1, 0, 0, 1, -1))

        self.assertRaises(ValueError, func,

                            (1900, 1, 1, 0, 60, 0, 0, 1, -1))

        # Check second [0, 61]

        self.assertRaises(ValueError, func,

                            (1900, 1, 1, 0, 0, -1, 0, 1, -1))

        # C99 only requires allowing for one leap second, but Python's docs say

        # allow two leap seconds (0..61)

        self.assertRaises(ValueError, func,

                            (1900, 1, 1, 0, 0, 62, 0, 1, -1))

        # No check for upper-bound day of week;

        #  value forced into range by a ``% 7`` calculation.

        # Start check at -2 since gettmarg() increments value before taking

        #  modulo.

        self.assertRaises(ValueError, func,

                            (1900, 1, 1, 0, 0, 0, -2, 1, -1))

        # Check day of the year [1, 366] + zero support

        self.assertRaises(ValueError, func,

                            (1900, 1, 1, 0, 0, 0, 0, -1, -1))

        self.assertRaises(ValueError, func,

                            (1900, 1, 1, 0, 0, 0, 0, 367, -1))

    def test_strftime_bounding_check(self):

        self._bounds_checking(lambda tup: time.strftime('', tup))

    def test_default_values_for_zero(self):

        # Make sure that using all zeros uses the proper default values.

        # No test for daylight savings since strftime() does not change output

        # based on its value.

        expected = "2000 01 01 00 00 00 1 001"

        with support.check_warnings():

            result = time.strftime("%Y %m %d %H %M %S %w %j", (0,)*9)

        self.assertEqual(expected, result)

    def test_strptime(self):

        # Should be able to go round-trip from strftime to strptime without

        # throwing an exception.

        tt = time.gmtime(self.t)

        for directive in ('a', 'A', 'b', 'B', 'c', 'd', 'H', 'I',

                          'j', 'm', 'M', 'p', 'S',

                          'U', 'w', 'W', 'x', 'X', 'y', 'Y', 'Z', '%'):

            format = '%' + directive

            strf_output = time.strftime(format, tt)

            try:

                time.strptime(strf_output, format)

            except ValueError:

                self.fail("conversion specifier %r failed with '%s' input." %

                          (format, strf_output))

    def test_strptime_bytes(self):

        # Make sure only strings are accepted as arguments to strptime.

        self.assertRaises(TypeError, time.strptime, b'2009', "%Y")

        self.assertRaises(TypeError, time.strptime, '2009', b'%Y')

    def test_asctime(self):

        time.asctime(time.gmtime(self.t))

        # Max year is only limited by the size of C int.

        sizeof_int = sysconfig.get_config_var('SIZEOF_INT') or 4

        bigyear = (1 << 8 * sizeof_int - 1) - 1

        asc = time.asctime((bigyear, 6, 1) + (0,)*6)

        self.assertEqual(asc[-len(str(bigyear)):], str(bigyear))

        self.assertRaises(OverflowError, time.asctime, (bigyear + 1,) + (0,)*8)

        self.assertRaises(TypeError, time.asctime, 0)

        self.assertRaises(TypeError, time.asctime, ())

        self.assertRaises(TypeError, time.asctime, (0,) * 10)

    def test_asctime_bounding_check(self):

        self._bounds_checking(time.asctime)

    def test_ctime(self):

        t = time.mktime((1973, 9, 16, 1, 3, 52, 0, 0, -1))

        self.assertEqual(time.ctime(t), 'Sun Sep 16 01:03:52 1973')

        t = time.mktime((2000, 1, 1, 0, 0, 0, 0, 0, -1))

        self.assertEqual(time.ctime(t), 'Sat Jan  1 00:00:00 2000')

        for year in [-100, 100, 1000, 2000, 10000]:

            try:

                testval = time.mktime((year, 1, 10) + (0,)*6)

            except (ValueError, OverflowError):

                # If mktime fails, ctime will fail too.  This may happen

                # on some platforms.

                pass

            else:

                self.assertEqual(time.ctime(testval)[20:], str(year))

    @unittest.skipIf(not hasattr(time, "tzset"),

        "time module has no attribute tzset")

    def test_tzset(self):

        from os import environ

        # Epoch time of midnight Dec 25th 2002. Never DST in northern

        # hemisphere.

        xmas2002 = 1040774400.0

        # These formats are correct for 2002, and possibly future years

        # This format is the 'standard' as documented at:

        # http://www.opengroup.org/onlinepubs/007904975/basedefs/xbd_chap08.html

        # They are also documented in the tzset(3) man page on most Unix

        # systems.

        eastern = 'EST+05EDT,M4.1.0,M10.5.0'

        victoria = 'AEST-10AEDT-11,M10.5.0,M3.5.0'

        utc='UTC+0'

        org_TZ = environ.get('TZ',None)

        try:

            # Make sure we can switch to UTC time and results are correct

            # Note that unknown timezones default to UTC.

            # Note that altzone is undefined in UTC, as there is no DST

            environ['TZ'] = eastern

            time.tzset()

            environ['TZ'] = utc

            time.tzset()

            self.assertEqual(

                time.gmtime(xmas2002), time.localtime(xmas2002)

                )

            self.assertEqual(time.daylight, 0)

            self.assertEqual(time.timezone, 0)

            self.assertEqual(time.localtime(xmas2002).tm_isdst, 0)

            # Make sure we can switch to US/Eastern

            environ['TZ'] = eastern

            time.tzset()

            self.assertNotEqual(time.gmtime(xmas2002), time.localtime(xmas2002))

            self.assertEqual(time.tzname, ('EST', 'EDT'))

            self.assertEqual(len(time.tzname), 2)

            self.assertEqual(time.daylight, 1)

            self.assertEqual(time.timezone, 18000)

            self.assertEqual(time.altzone, 14400)

            self.assertEqual(time.localtime(xmas2002).tm_isdst, 0)

            self.assertEqual(len(time.tzname), 2)

            # Now go to the southern hemisphere.

            environ['TZ'] = victoria

            time.tzset()

            self.assertNotEqual(time.gmtime(xmas2002), time.localtime(xmas2002))

            self.assertTrue(time.tzname[0] == 'AEST', str(time.tzname[0]))

            self.assertTrue(time.tzname[1] == 'AEDT', str(time.tzname[1]))

            self.assertEqual(len(time.tzname), 2)

            self.assertEqual(time.daylight, 1)

            self.assertEqual(time.timezone, -36000)

            self.assertEqual(time.altzone, -39600)

            self.assertEqual(time.localtime(xmas2002).tm_isdst, 1)

        finally:

            # Repair TZ environment variable in case any other tests

            # rely on it.

            if org_TZ is not None:

                environ['TZ'] = org_TZ

            elif 'TZ' in environ:

                del environ['TZ']

            time.tzset()

    def test_insane_timestamps(self):

        # It's possible that some platform maps time_t to double,

        # and that this test will fail there.  This test should

        # exempt such platforms (provided they return reasonable

        # results!).

        for func in time.ctime, time.gmtime, time.localtime:

            for unreasonable in -1e200, 1e200:

                self.assertRaises(ValueError, func, unreasonable)

    def test_ctime_without_arg(self):

        # Not sure how to check the values, since the clock could tick

        # at any time.  Make sure these are at least accepted and

        # don't raise errors.

        time.ctime()

        time.ctime(None)

    def test_gmtime_without_arg(self):

        gt0 = time.gmtime()

        gt1 = time.gmtime(None)

        t0 = time.mktime(gt0)

        t1 = time.mktime(gt1)

        self.assertAlmostEqual(t1, t0, delta=0.2)

    def test_localtime_without_arg(self):

        lt0 = time.localtime()

        lt1 = time.localtime(None)

        t0 = time.mktime(lt0)

        t1 = time.mktime(lt1)

        self.assertAlmostEqual(t1, t0, delta=0.2)

class TestLocale(unittest.TestCase):

    def setUp(self):

        self.oldloc = locale.setlocale(locale.LC_ALL)

    def tearDown(self):

        locale.setlocale(locale.LC_ALL, self.oldloc)

    def test_bug_3061(self):

        try:

            tmp = locale.setlocale(locale.LC_ALL, "fr_FR")

        except locale.Error:

            # skip this test

            return

        # This should not cause an exception

        time.strftime("%B", (2009,2,1,0,0,0,0,0,0))

class _BaseYearTest(unittest.TestCase):

    accept2dyear = None

    def setUp(self):

        self.saved_accept2dyear = time.accept2dyear

        time.accept2dyear = self.accept2dyear

    def tearDown(self):

        time.accept2dyear = self.saved_accept2dyear

    def yearstr(self, y):

        raise NotImplementedError()

class _TestAsctimeYear:

    def yearstr(self, y):

        return time.asctime((y,) + (0,) * 8).split()[-1]

    def test_large_year(self):

        # Check that it doesn't crash for year > 9999

        self.assertEqual(self.yearstr(12345), '12345')

        self.assertEqual(self.yearstr(123456789), '123456789')

class _TestStrftimeYear:

    def yearstr(self, y):

        return time.strftime('%Y', (y,) + (0,) * 8).split()[-1]

    def test_large_year(self):

        # Check that it doesn't crash for year > 9999

        try:

            text = self.yearstr(12345)

        except ValueError:

            # strftime() is limited to [1; 9999] with Visual Studio

            return

        # Issue #10864: OpenIndiana is limited to 4 digits,

        # but Python doesn't raise a ValueError

        #self.assertEqual(text, '12345')

        #self.assertEqual(self.yearstr(123456789), '123456789')

        self.assertIn(text, ('2345', '12345'))

        self.assertIn(self.yearstr(123456789), ('123456789', '6789'))

class _Test2dYear(_BaseYearTest):

    accept2dyear = 1

    def test_year(self):

        with support.check_warnings():

            self.assertEqual(self.yearstr(0), '2000')

            self.assertEqual(self.yearstr(69), '1969')

            self.assertEqual(self.yearstr(68), '2068')

            self.assertEqual(self.yearstr(99), '1999')

    def test_invalid(self):

        self.assertRaises(ValueError, self.yearstr, -1)

        self.assertRaises(ValueError, self.yearstr, 100)

        self.assertRaises(ValueError, self.yearstr, 999)

class _Test4dYear(_BaseYearTest):

    accept2dyear = 0

    def test_year(self):

        self.assertIn(self.yearstr(1),     ('1', '0001'))

        self.assertIn(self.yearstr(68),   ('68', '0068'))

        self.assertIn(self.yearstr(69),   ('69', '0069'))

        self.assertIn(self.yearstr(99),   ('99', '0099'))

        self.assertIn(self.yearstr(999), ('999', '0999'))

        self.assertEqual(self.yearstr(9999), '9999')

    def test_negative(self):

        try:

            text = self.yearstr(-1)

        except ValueError:

            # strftime() is limited to [1; 9999] with Visual Studio

            return

        self.assertIn(text, ('-1', '-001'))

        self.assertEqual(self.yearstr(-1234), '-1234')

        self.assertEqual(self.yearstr(-123456), '-123456')

    def test_mktime(self):

        # Issue #1726687

        for t in (-2, -1, 0, 1):

            try:

                tt = time.localtime(t)

            except (OverflowError, ValueError):

                pass

            else:

                self.assertEqual(time.mktime(tt), t)

        # It may not be possible to reliably make mktime return error

        # on all platfom.  This will make sure that no other exception

        # than OverflowError is raised for an extreme value.

        try:

            time.mktime((-1, 1, 1, 0, 0, 0, -1, -1, -1))

        except OverflowError:

            pass

class TestAsctimeAccept2dYear(_TestAsctimeYear, _Test2dYear):

    pass

class TestStrftimeAccept2dYear(_TestStrftimeYear, _Test2dYear):

    pass

class TestAsctime4dyear(_TestAsctimeYear, _Test4dYear):

    pass

class TestStrftime4dyear(_TestStrftimeYear, _Test4dYear):

    pass

class Test2dyearBool(_TestAsctimeYear, _Test2dYear):

    accept2dyear = True

class Test4dyearBool(_TestAsctimeYear, _Test4dYear):

    accept2dyear = False

class TestAccept2YearBad(_TestAsctimeYear, _BaseYearTest):

    class X:

        def __bool__(self):

            raise RuntimeError('boo')

    accept2dyear = X()

    def test_2dyear(self):

        pass

    def test_invalid(self):

        self.assertRaises(RuntimeError, self.yearstr, 200)

def test_main():

    support.run_unittest(

        TimeTestCase,

        TestLocale,

        TestAsctimeAccept2dYear,

        TestStrftimeAccept2dYear,

        TestAsctime4dyear,

        TestStrftime4dyear,

        Test2dyearBool,

        Test4dyearBool,

        TestAccept2YearBad)

if __name__ == "__main__":

    test_main()