test_time.py

import unittest
from aviation_weather import Time
from aviation_weather.exceptions import TimeDecodeError


class TestTime(unittest.TestCase):
    """Unit tests for aviation_weather.components.time.Time"""

    def test_valid(self):
        time_str = "161851Z"
        time_obj = Time(time_str)
        self.assertEqual(time_str, time_obj.raw)
        self.assertEqual(16, time_obj.day)
        self.assertEqual(18, time_obj.hour)
        self.assertEqual(51, time_obj.minute)
        self.assertEqual("Z", time_obj.timezone)

    def _test_invalid(self, raw):
        with self.assertRaises(TimeDecodeError):
            Time(raw)

    def test_invalid_empty(self):
        self._test_invalid("")

    def test_invalid_letters(self):
        self._test_invalid("ABCDEFZ")

    def test_invalid_short(self):
        self._test_invalid("1851Z")

    def test_invalid_long(self):
        self._test_invalid("02101851Z")

    def test_invalid_unit(self):
        self._test_invalid("101851A")

    def test_invalid_day(self):
        self._test_invalid("321456Z")

    def test_invalid_hour(self):
        self._test_invalid("092507Z")

    def test_invalid_minute(self):
        self._test_invalid("122384Z")