← Back to branch summary

~awuerl/blitzortung-python/master

« back to all changes in this revision

Viewing changes to tests/test_util.py

  • Committer: Andreas Wuerl
  • Date: 2019-06-24 19:32:25 UTC
  • mto: This revision was merged to the branch mainline in revision 394.
  • Revision ID: git-v1:29d4be282c05035bdff8d4828573537ed6d8ceea
update tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/test_util.py
18
18
 
19
19
"""
20
20
 
21
 
import unittest
22
21
import datetime
23
22
 
24
 
from hamcrest import assert_that, contains
25
23
import pytz
 
24
from assertpy import assert_that
26
25
 
27
26
import blitzortung.util
28
27
 
29
28
 
30
 
class TimeIntervalsTest(unittest.TestCase):
 
29
class TimeIntervalsTest(object):
31
30
    def setUp(self):
32
31
        self.duration = datetime.timedelta(minutes=10)
33
32
 
40
39
 
41
40
        times = [time for time in blitzortung.util.time_intervals(self.start_time, self.duration, self.end_time)]
42
41
 
43
 
        assert_that(times, contains(
 
42
        assert_that(times).contains(
44
43
            datetime.datetime(2013, 8, 20, 11, 40, 0),
45
44
            datetime.datetime(2013, 8, 20, 11, 50, 0),
46
45
            datetime.datetime(2013, 8, 20, 12, 0, 0),
47
 
        ))
 
46
        )
48
47
 
49
48
    def test_time_intervals_generator_at_start_of_interval(self):
50
49
        self.initialize_times(datetime.datetime(2013, 8, 20, 12, 5, 0))
51
50
 
52
51
        times = [time for time in blitzortung.util.time_intervals(self.start_time, self.duration, self.end_time)]
53
52
 
54
 
        assert_that(times, contains(
 
53
        assert_that(times).contains(
55
54
            datetime.datetime(2013, 8, 20, 11, 40, 0),
56
55
            datetime.datetime(2013, 8, 20, 11, 50, 0),
57
56
            datetime.datetime(2013, 8, 20, 12, 0, 0),
58
 
        ))
 
57
        )
59
58
 
60
59
    def test_time_intervals_generator_before_start_of_interval(self):
61
60
        self.initialize_times(datetime.datetime(2013, 8, 20, 12, 4, 59))
62
61
 
63
62
        times = [time for time in blitzortung.util.time_intervals(self.start_time, self.duration, self.end_time)]
64
63
 
65
 
        assert_that(times, contains(
 
64
        assert_that(times).contains(
66
65
            datetime.datetime(2013, 8, 20, 11, 30, 0),
67
66
            datetime.datetime(2013, 8, 20, 11, 40, 0),
68
67
            datetime.datetime(2013, 8, 20, 11, 50, 0),
69
68
            datetime.datetime(2013, 8, 20, 12, 0, 0),
70
 
        ))
 
69
        )
71
70
 
72
71
    def test_time_intervals_generator_at_current_time(self):
73
72
        end_time = datetime.datetime.utcnow()
76
75
 
77
76
        times = [time for time in blitzortung.util.time_intervals(start_time, self.duration)]
78
77
 
79
 
        assert_that(times, contains(
 
78
        assert_that(times).contains(
80
79
            blitzortung.util.round_time(start_time, self.duration),
81
80
            blitzortung.util.round_time(end_time, self.duration)
82
 
        ))
 
81
        )