# -*- coding: utf-8 -*-

from StringIO import StringIO
from obspy import UTCDateTime
from obspy.xseed.fields import Float, VariableString
import unittest


class FieldsTestCase(unittest.TestCase):
    """
    Fields test suite.
    """
    def setUp(self):
        pass

    def tearDown(self):
        pass

    def test_formatExponential(self):
        field = Float(1, "test", 12, mask='%+1.5e', strict=True)
        self.assertEquals(field.write('2.5'), '+2.50000E+00')

    def test_readDateTime(self):
        field = VariableString(1, "test", 1, 22, 'T', strict=True)
        #1
        orig = '1992,002,00:00:00.0000~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(1992, 1, 2))
        self.assertEquals(field.write(dt), '1992,002~')
        #1
        orig = '1992,002~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(1992, 1, 2))
        self.assertEquals(field.write(dt), '1992,002~')
        #2
        orig = '1992,005,01:02:03.4567~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(1992, 1, 5, 1, 2, 3, 456700))
        self.assertEquals(field.write(dt), orig)
        #3
        orig = '1992,005,01:02:03.0001~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(1992, 1, 5, 1, 2, 3, 100))
        self.assertEquals(field.write(dt), orig)
        #4
        orig = '1992,005,01:02:03.1000~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(1992, 1, 5, 1, 2, 3, 100000))
        self.assertEquals(field.write(dt), orig)
        #5
        orig = '1987,023,04:23:05.1~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(1987, 1, 23, 4, 23, 5, 100000))
        self.assertEquals(field.write(dt), '1987,023,04:23:05.1000~')
        #6
        orig = '1987,023,04:23:05.123~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(1987, 1, 23, 4, 23, 5, 123000))
        self.assertEquals(field.write(dt), '1987,023,04:23:05.1230~')
        #
        orig = '2008,358,01:30:22.0987~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2008, 12, 23, 01, 30, 22, 98700))
        self.assertEquals(field.write(dt), orig)
        #
        orig = '2008,358,01:30:22.9876~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2008, 12, 23, 01, 30, 22, 987600))
        self.assertEquals(field.write(dt), orig)
        #
        orig = '2008,358,01:30:22.0005~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2008, 12, 23, 01, 30, 22, 500))
        self.assertEquals(field.write(dt), orig)
        #
        orig = '2008,358,01:30:22.0000~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2008, 12, 23, 01, 30, 22, 0))
        self.assertEquals(field.write(dt), orig)

    def test_readCompactDateTime(self):
        field = VariableString(1, "test", 0, 22, 'T', strict=True,
                               compact=True)
        #1
        orig = '1992,002~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(1992, 1, 2))
        self.assertEquals(field.write(dt), orig)
        #2
        orig = '2007,199~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2007, 7, 18))
        self.assertEquals(field.write(dt), orig)
        #3 - wrong syntax
        orig = '1992'
        self.assertRaises(Exception, field.read, StringIO(orig))
        orig = '1992,'
        self.assertRaises(Exception, field.read, StringIO(orig))
        orig = '1992~'
        self.assertRaises(Exception, field.read, StringIO(orig))
        orig = '1992,~'
        self.assertRaises(Exception, field.read, StringIO(orig))
        #5 - empty datetime
        orig = '~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, '')
        self.assertEquals(field.write(dt), '~')
        #6 - bad syntax
        orig = ''
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, '')
        self.assertEquals(field.write(dt), '~')
        #7
        orig = '2007,199'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2007, 7, 18))
        self.assertEquals(field.write(dt), '2007,199~')
        #8
        orig = '2009,074,12~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2009, 3, 15, 12))
        self.assertEquals(field.write(dt), orig)
        #9
        orig = '2008,358,01:30:22.0012~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2008, 12, 23, 01, 30, 22, 1200))
        self.assertEquals(field.write(dt), orig)
        #
        orig = '2008,358,00:00:22~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2008, 12, 23, 00, 00, 22, 0))
        self.assertEquals(field.write(dt), orig)
        #
        orig = '2008,358,00:30~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2008, 12, 23, 00, 30, 0, 0))
        self.assertEquals(field.write(dt), orig)
        #
        orig = '2008,358,01~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2008, 12, 23, 01, 0, 0, 0))
        self.assertEquals(field.write(dt), orig)
        #
        orig = '2008,358~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2008, 12, 23, 0, 0, 0, 0))
        self.assertEquals(field.write(dt), orig)
        #
        orig = '2008,358,01:30:22.5~'
        dt = field.read(StringIO(orig))
        self.assertEquals(dt, UTCDateTime(2008, 12, 23, 01, 30, 22, 500000))
        self.assertEquals(field.write(dt), '2008,358,01:30:22.5000~')


def suite():
    return unittest.makeSuite(FieldsTestCase, 'test')


if __name__ == '__main__':
    unittest.main(defaultTest='suite')
