|
|
# -*- coding: utf-8 -*-
# Copyright (c) 2021, Brandon Nielsen# All rights reserved.## This software may be modified and distributed under the terms# of the BSD license. See the LICENSE file for details.
import unittest
import aniso8601from aniso8601.builders import DatetimeTuple, DateTuple, TimeTuple, TimezoneTuplefrom aniso8601.exceptions import ISOFormatErrorfrom aniso8601.resolution import TimeResolutionfrom aniso8601.tests.compat import mockfrom aniso8601.time import ( _get_time_resolution, get_datetime_resolution, get_time_resolution, parse_datetime, parse_time,)
class TestTimeResolutionFunctions(unittest.TestCase): def test_get_time_resolution(self): self.assertEqual(get_time_resolution("01:23:45"), TimeResolution.Seconds) self.assertEqual(get_time_resolution("24:00:00"), TimeResolution.Seconds) self.assertEqual(get_time_resolution("23:21:28,512400"), TimeResolution.Seconds) self.assertEqual(get_time_resolution("23:21:28.512400"), TimeResolution.Seconds) self.assertEqual(get_time_resolution("01:23"), TimeResolution.Minutes) self.assertEqual(get_time_resolution("24:00"), TimeResolution.Minutes) self.assertEqual(get_time_resolution("01:23,4567"), TimeResolution.Minutes) self.assertEqual(get_time_resolution("01:23.4567"), TimeResolution.Minutes) self.assertEqual(get_time_resolution("012345"), TimeResolution.Seconds) self.assertEqual(get_time_resolution("240000"), TimeResolution.Seconds) self.assertEqual(get_time_resolution("0123"), TimeResolution.Minutes) self.assertEqual(get_time_resolution("2400"), TimeResolution.Minutes) self.assertEqual(get_time_resolution("01"), TimeResolution.Hours) self.assertEqual(get_time_resolution("24"), TimeResolution.Hours) self.assertEqual(get_time_resolution("12,5"), TimeResolution.Hours) self.assertEqual(get_time_resolution("12.5"), TimeResolution.Hours) self.assertEqual( get_time_resolution("232128.512400+00:00"), TimeResolution.Seconds ) self.assertEqual(get_time_resolution("0123.4567+00:00"), TimeResolution.Minutes) self.assertEqual(get_time_resolution("01.4567+00:00"), TimeResolution.Hours) self.assertEqual(get_time_resolution("01:23:45+00:00"), TimeResolution.Seconds) self.assertEqual(get_time_resolution("24:00:00+00:00"), TimeResolution.Seconds) self.assertEqual( get_time_resolution("23:21:28.512400+00:00"), TimeResolution.Seconds ) self.assertEqual(get_time_resolution("01:23+00:00"), TimeResolution.Minutes) self.assertEqual(get_time_resolution("24:00+00:00"), TimeResolution.Minutes) self.assertEqual( get_time_resolution("01:23.4567+00:00"), TimeResolution.Minutes ) self.assertEqual( get_time_resolution("23:21:28.512400+11:15"), TimeResolution.Seconds ) self.assertEqual( get_time_resolution("23:21:28.512400-12:34"), TimeResolution.Seconds ) self.assertEqual( get_time_resolution("23:21:28.512400Z"), TimeResolution.Seconds ) self.assertEqual( get_time_resolution("06:14:00.000123Z"), TimeResolution.Seconds )
def test_get_datetime_resolution(self): self.assertEqual( get_datetime_resolution("2019-06-05T01:03:11.858714"), TimeResolution.Seconds, ) self.assertEqual( get_datetime_resolution("2019-06-05T01:03:11"), TimeResolution.Seconds ) self.assertEqual( get_datetime_resolution("2019-06-05T01:03"), TimeResolution.Minutes ) self.assertEqual(get_datetime_resolution("2019-06-05T01"), TimeResolution.Hours)
def test_get_time_resolution_badtype(self): testtuples = (None, 1, False, 1.234)
for testtuple in testtuples: with self.assertRaises(ValueError): get_time_resolution(testtuple)
def test_get_time_resolution_badstr(self): testtuples = ("A6:14:00.000123Z", "06:14:0B", "bad", "")
for testtuple in testtuples: with self.assertRaises(ISOFormatError): get_time_resolution(testtuple)
def test_get_time_resolution_internal(self): self.assertEqual( _get_time_resolution(TimeTuple(hh="01", mm="02", ss="03", tz=None)), TimeResolution.Seconds, ) self.assertEqual( _get_time_resolution(TimeTuple(hh="01", mm="02", ss=None, tz=None)), TimeResolution.Minutes, ) self.assertEqual( _get_time_resolution(TimeTuple(hh="01", mm=None, ss=None, tz=None)), TimeResolution.Hours, )
class TestTimeParserFunctions(unittest.TestCase): def test_parse_time(self): testtuples = ( ("01:23:45", {"hh": "01", "mm": "23", "ss": "45", "tz": None}), ("24:00:00", {"hh": "24", "mm": "00", "ss": "00", "tz": None}), ( "23:21:28,512400", {"hh": "23", "mm": "21", "ss": "28.512400", "tz": None}, ), ( "23:21:28.512400", {"hh": "23", "mm": "21", "ss": "28.512400", "tz": None}, ), ( "01:03:11.858714", {"hh": "01", "mm": "03", "ss": "11.858714", "tz": None}, ), ( "14:43:59.9999997", {"hh": "14", "mm": "43", "ss": "59.9999997", "tz": None}, ), ("01:23", {"hh": "01", "mm": "23", "ss": None, "tz": None}), ("24:00", {"hh": "24", "mm": "00", "ss": None, "tz": None}), ("01:23,4567", {"hh": "01", "mm": "23.4567", "ss": None, "tz": None}), ("01:23.4567", {"hh": "01", "mm": "23.4567", "ss": None, "tz": None}), ("012345", {"hh": "01", "mm": "23", "ss": "45", "tz": None}), ("240000", {"hh": "24", "mm": "00", "ss": "00", "tz": None}), ("232128,512400", {"hh": "23", "mm": "21", "ss": "28.512400", "tz": None}), ("232128.512400", {"hh": "23", "mm": "21", "ss": "28.512400", "tz": None}), ("010311.858714", {"hh": "01", "mm": "03", "ss": "11.858714", "tz": None}), ( "144359.9999997", {"hh": "14", "mm": "43", "ss": "59.9999997", "tz": None}, ), ("0123", {"hh": "01", "mm": "23", "ss": None, "tz": None}), ("2400", {"hh": "24", "mm": "00", "ss": None, "tz": None}), ("01", {"hh": "01", "mm": None, "ss": None, "tz": None}), ("24", {"hh": "24", "mm": None, "ss": None, "tz": None}), ("12,5", {"hh": "12.5", "mm": None, "ss": None, "tz": None}), ("12.5", {"hh": "12.5", "mm": None, "ss": None, "tz": None}), ( "232128,512400+00:00", { "hh": "23", "mm": "21", "ss": "28.512400", "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "232128.512400+00:00", { "hh": "23", "mm": "21", "ss": "28.512400", "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "0123,4567+00:00", { "hh": "01", "mm": "23.4567", "ss": None, "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "0123.4567+00:00", { "hh": "01", "mm": "23.4567", "ss": None, "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "01,4567+00:00", { "hh": "01.4567", "mm": None, "ss": None, "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "01.4567+00:00", { "hh": "01.4567", "mm": None, "ss": None, "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "01:23:45+00:00", { "hh": "01", "mm": "23", "ss": "45", "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "24:00:00+00:00", { "hh": "24", "mm": "00", "ss": "00", "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "23:21:28.512400+00:00", { "hh": "23", "mm": "21", "ss": "28.512400", "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "01:23+00:00", { "hh": "01", "mm": "23", "ss": None, "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "24:00+00:00", { "hh": "24", "mm": "00", "ss": None, "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "01:23.4567+00:00", { "hh": "01", "mm": "23.4567", "ss": None, "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }, ), ( "23:21:28.512400+11:15", { "hh": "23", "mm": "21", "ss": "28.512400", "tz": TimezoneTuple(False, None, "11", "15", "+11:15"), }, ), ( "23:21:28.512400-12:34", { "hh": "23", "mm": "21", "ss": "28.512400", "tz": TimezoneTuple(True, None, "12", "34", "-12:34"), }, ), ( "23:21:28.512400Z", { "hh": "23", "mm": "21", "ss": "28.512400", "tz": TimezoneTuple(False, True, None, None, "Z"), }, ), ( "06:14:00.000123Z", { "hh": "06", "mm": "14", "ss": "00.000123", "tz": TimezoneTuple(False, True, None, None, "Z"), }, ), )
for testtuple in testtuples: with mock.patch.object( aniso8601.time.PythonTimeBuilder, "build_time" ) as mockBuildTime:
mockBuildTime.return_value = testtuple[1]
result = parse_time(testtuple[0])
self.assertEqual(result, testtuple[1]) mockBuildTime.assert_called_once_with(**testtuple[1])
def test_parse_time_badtype(self): testtuples = (None, 1, False, 1.234)
for testtuple in testtuples: with self.assertRaises(ValueError): parse_time(testtuple, builder=None)
def test_parse_time_badstr(self): testtuples = ( "A6:14:00.000123Z", "06:14:0B", "06:1 :02", "0000,70:24,9", "00.27:5332", "bad", "", )
for testtuple in testtuples: with self.assertRaises(ISOFormatError): parse_time(testtuple, builder=None)
def test_parse_time_mockbuilder(self): mockBuilder = mock.Mock()
expectedargs = {"hh": "01", "mm": "23", "ss": "45", "tz": None}
mockBuilder.build_time.return_value = expectedargs
result = parse_time("01:23:45", builder=mockBuilder)
self.assertEqual(result, expectedargs) mockBuilder.build_time.assert_called_once_with(**expectedargs)
mockBuilder = mock.Mock()
expectedargs = { "hh": "23", "mm": "21", "ss": "28.512400", "tz": TimezoneTuple(False, None, "00", "00", "+00:00"), }
mockBuilder.build_time.return_value = expectedargs
result = parse_time("232128.512400+00:00", builder=mockBuilder)
self.assertEqual(result, expectedargs) mockBuilder.build_time.assert_called_once_with(**expectedargs)
mockBuilder = mock.Mock()
expectedargs = { "hh": "23", "mm": "21", "ss": "28.512400", "tz": TimezoneTuple(False, None, "11", "15", "+11:15"), }
mockBuilder.build_time.return_value = expectedargs
result = parse_time("23:21:28.512400+11:15", builder=mockBuilder)
self.assertEqual(result, expectedargs) mockBuilder.build_time.assert_called_once_with(**expectedargs)
def test_parse_datetime(self): testtuples = ( ( "2019-06-05T01:03:11,858714", ( DateTuple("2019", "06", "05", None, None, None), TimeTuple("01", "03", "11.858714", None), ), ), ( "2019-06-05T01:03:11.858714", ( DateTuple("2019", "06", "05", None, None, None), TimeTuple("01", "03", "11.858714", None), ), ), ( "1981-04-05T23:21:28.512400Z", ( DateTuple("1981", "04", "05", None, None, None), TimeTuple( "23", "21", "28.512400", TimezoneTuple(False, True, None, None, "Z"), ), ), ), ( "1981095T23:21:28.512400-12:34", ( DateTuple("1981", None, None, None, None, "095"), TimeTuple( "23", "21", "28.512400", TimezoneTuple(True, None, "12", "34", "-12:34"), ), ), ), ( "19810405T23:21:28+00", ( DateTuple("1981", "04", "05", None, None, None), TimeTuple( "23", "21", "28", TimezoneTuple(False, None, "00", None, "+00") ), ), ), ( "19810405T23:21:28+00:00", ( DateTuple("1981", "04", "05", None, None, None), TimeTuple( "23", "21", "28", TimezoneTuple(False, None, "00", "00", "+00:00"), ), ), ), )
for testtuple in testtuples: with mock.patch.object( aniso8601.time.PythonTimeBuilder, "build_datetime" ) as mockBuildDateTime:
mockBuildDateTime.return_value = testtuple[1]
result = parse_datetime(testtuple[0])
self.assertEqual(result, testtuple[1]) mockBuildDateTime.assert_called_once_with(*testtuple[1])
def test_parse_datetime_spacedelimited(self): expectedargs = ( DateTuple("2004", None, None, "53", "6", None), TimeTuple( "23", "21", "28.512400", TimezoneTuple(True, None, "12", "34", "-12:34") ), )
with mock.patch.object( aniso8601.time.PythonTimeBuilder, "build_datetime" ) as mockBuildDateTime:
mockBuildDateTime.return_value = expectedargs
result = parse_datetime("2004-W53-6 23:21:28.512400-12:34", delimiter=" ")
self.assertEqual(result, expectedargs) mockBuildDateTime.assert_called_once_with(*expectedargs)
def test_parse_datetime_commadelimited(self): expectedargs = ( DateTuple("1981", "04", "05", None, None, None), TimeTuple( "23", "21", "28.512400", TimezoneTuple(False, True, None, None, "Z") ), )
with mock.patch.object( aniso8601.time.PythonTimeBuilder, "build_datetime" ) as mockBuildDateTime:
mockBuildDateTime.return_value = expectedargs
result = parse_datetime("1981-04-05,23:21:28,512400Z", delimiter=",")
self.assertEqual(result, expectedargs) mockBuildDateTime.assert_called_once_with(*expectedargs)
def test_parse_datetime_baddelimiter(self): testtuples = ( "1981-04-05,23:21:28,512400Z", "2004-W53-6 23:21:28.512400-12:3", "1981040523:21:28", )
for testtuple in testtuples: with self.assertRaises(ISOFormatError): parse_datetime(testtuple, builder=None)
def test_parse_datetime_badtype(self): testtuples = (None, 1, False, 1.234)
for testtuple in testtuples: with self.assertRaises(ValueError): parse_datetime(testtuple, builder=None)
def test_parse_datetime_badstr(self): testtuples = ( "1981-04-05TA6:14:00.000123Z", "2004-W53-6T06:14:0B", "2014-01-230T23:21:28+00", "201401230T01:03:11.858714", "9999 W53T49", "9T0000,70:24,9", "bad", "", )
for testtuple in testtuples: with self.assertRaises(ISOFormatError): parse_datetime(testtuple, builder=None)
def test_parse_datetime_mockbuilder(self): mockBuilder = mock.Mock()
expectedargs = ( DateTuple("1981", None, None, None, None, "095"), TimeTuple( "23", "21", "28.512400", TimezoneTuple(True, None, "12", "34", "-12:34") ), )
mockBuilder.build_datetime.return_value = expectedargs
result = parse_datetime("1981095T23:21:28.512400-12:34", builder=mockBuilder)
self.assertEqual(result, expectedargs) mockBuilder.build_datetime.assert_called_once_with(*expectedargs)
|
