import unittest

from test import test_support



import time



class StructSeqTest(unittest.TestCase):



    def test_tuple(self):

        t = time.gmtime()

        astuple = tuple(t)

        self.assertEqual(len(t), len(astuple))

        self.assertEqual(t, astuple)



        # Check that slicing works the same way; at one point, slicing t[i:j] with

        # 0 < i < j could produce NULLs in the result.

        for i in xrange(-len(t), len(t)):

            self.assertEqual(t[i:], astuple[i:])

            for j in xrange(-len(t), len(t)):

                self.assertEqual(t[i:j], astuple[i:j])



        for j in xrange(-len(t), len(t)):

            self.assertEqual(t[:j], astuple[:j])



        self.assertRaises(IndexError, t.__getitem__, -len(t)-1)

        self.assertRaises(IndexError, t.__getitem__, len(t))

        for i in xrange(-len(t), len(t)-1):

            self.assertEqual(t[i], astuple[i])



    def test_repr(self):

        t = time.gmtime()

        self.assert_(repr(t))

        t = time.gmtime(0)

        self.assertEqual(repr(t),

            "time.struct_time(tm_year=1970, tm_mon=1, tm_mday=1, tm_hour=0, "

            "tm_min=0, tm_sec=0, tm_wday=3, tm_yday=1, tm_isdst=0)")



    def test_concat(self):

        t1 = time.gmtime()

        t2 = t1 + tuple(t1)

        for i in xrange(len(t1)):

            self.assertEqual(t2[i], t2[i+len(t1)])



    def test_repeat(self):

        t1 = time.gmtime()

        t2 = 3 * t1

        for i in xrange(len(t1)):

            self.assertEqual(t2[i], t2[i+len(t1)])

            self.assertEqual(t2[i], t2[i+2*len(t1)])



    def test_contains(self):

        t1 = time.gmtime()

        for item in t1:

            self.assert_(item in t1)

        self.assert_(-42 not in t1)



    def test_hash(self):

        t1 = time.gmtime()

        self.assertEqual(hash(t1), hash(tuple(t1)))



    def test_cmp(self):

        t1 = time.gmtime()

        t2 = type(t1)(t1)

        self.assertEqual(t1, t2)

        self.assert_(not (t1 < t2))

        self.assert_(t1 <= t2)

        self.assert_(not (t1 > t2))

        self.assert_(t1 >= t2)

        self.assert_(not (t1 != t2))



    def test_fields(self):

        t = time.gmtime()

        self.assertEqual(len(t), t.n_fields)

        self.assertEqual(t.n_fields, t.n_sequence_fields+t.n_unnamed_fields)



    def test_constructor(self):

        t = time.struct_time



        self.assertRaises(TypeError, t)

        self.assertRaises(TypeError, t, None)

        self.assertRaises(TypeError, t, "123")

        self.assertRaises(TypeError, t, "123", dict={})

        self.assertRaises(TypeError, t, "123456789", dict=None)



        s = "123456789"

        self.assertEqual("".join(t(s)), s)



    def test_eviltuple(self):

        class Exc(Exception):

            pass



        # Devious code could crash structseqs' contructors

        class C:

            def __getitem__(self, i):

                raise Exc

            def __len__(self):

                return 9



        self.assertRaises(Exc, time.struct_time, C())



    def test_reduce(self):

        t = time.gmtime()

        x = t.__reduce__()



    def test_extended_getslice(self):

        # Test extended slicing by comparing with list slicing.

        t = time.gmtime()

        L = list(t)

        indices = (0, None, 1, 3, 19, 300, -1, -2, -31, -300)

        for start in indices:

            for stop in indices:

                # Skip step 0 (invalid)

                for step in indices[1:]:

                    self.assertEqual(list(t[start:stop:step]),

                                     L[start:stop:step])



def test_main():

    test_support.run_unittest(StructSeqTest)



if __name__ == "__main__":

    test_main()

