test_parse_time.py

import unittest

from datetime import timedelta

from data_diff.parse_time import parse_time_delta


class TestParseTime(unittest.TestCase):
    def setUp(self):
        pass

    def test_times(self):
        td = parse_time_delta("1w2d3h4min5s")
        assert td == timedelta(weeks=1, days=2, hours=3, minutes=4, seconds=5)

        assert parse_time_delta("1y") == timedelta(days=365)
        assert parse_time_delta("1mon") == timedelta(days=30)

        self.assertRaises(ValueError, parse_time_delta, "")
        self.assertRaises(ValueError, parse_time_delta, "1y1year")
        self.assertRaises(ValueError, parse_time_delta, "1x")