def test_amount_functionality(self):
"""
Does specifying desired amount of pomodori work?
"""
pomodori = PomodoroCalculator(end='12', amount=True).pomodori_schedule()
del pomodori['segments']
# Use known functionality to confirm correctness
pomodori_reference = PomodoroCalculator(end='06:39:00').pomodori_schedule()
del pomodori_reference['segments']
self.assertDictEqual(
pomodori,
{
'end': datetime(2018, 9, 1, 6, 15),
'start': datetime(2018, 9, 1, 0, 0),
'seconds-per-pomodoro': 1500,
'total-breaks': 11,
'total-pomodori': 12,
'total-rest-time': 4500,
'total-work-time': 18000,
}
)
self.assertDictEqual(pomodori_reference, pomodori)
# Try shorter item
pomodori_short = PomodoroCalculator(end='1', amount=True).pomodori_schedule()
del pomodori_short['segments']
self.assertDictEqual(
pomodori_short,
{
'end': datetime(2018, 9, 1, 0, 25),
'start': datetime(2018, 9, 1, 0, 0),
'seconds-per-pomodoro': 1500,
'total-breaks': 0,
'total-pomodori': 1,
'total-rest-time': 0,
'total-work-time': 1500,
}
)
pomodori_final = PomodoroCalculator(end='3', amount=True).pomodori_schedule()
del pomodori_final['segments']
self.assertDictEqual(
pomodori_final,
{
'end': datetime(2018, 9, 1, 1, 25),
'start': datetime(2018, 9, 1, 0, 0),
'seconds-per-pomodoro': 1500,
'total-breaks': 2,
'total-pomodori': 3,
'total-rest-time': 600,
'total-work-time': 4500,
}
)
def main():
arguments = docopt(__doc__, version=__version__)
calc = PomodoroCalculator(
end=arguments['<end-time>'],
start=arguments['--from'],
pomodoro_length=int(arguments['--pomodoro']),
group_length=int(arguments['--group']),
short_break=int(arguments['--break']),
long_break=int(arguments['--long-break']),
time_interval=(arguments['-i']),
)
colours = {
'pomodoro': Style.BRIGHT + Fore.RED,
'short-break': Fore.BLUE,
'long-break': Fore.CYAN,
}
init(autoreset=True)
pomodori_count = 0
for segment in calc.pomodori_schedule():
line_dict = {}
if segment['type'] == 'pomodoro':
pomodori_count += 1
line_dict['id'] = pomodori_count
line_dict['name'] = 'Pomodoro'
elif segment['type'] == 'short-break':
line_dict['id'] = ''
line_dict['name'] = 'short break'
else:
line_dict['id'] = ''
line_dict['name'] = 'long break'
line_dict['start'] = segment['start'].strftime('%H:%M')
line_dict['end'] = segment['end'].strftime('%H:%M')
line = "{id:>2} {name:<12} {start} ⇾ {end}".format(**line_dict)
print(colours[segment['type']] + line)
total = '{:>26} {:>2}'.format('Total Pomodori:', pomodori_count)
print(Style.BRIGHT + Fore.WHITE + total)
hours = round(pomodori_count * int(arguments['--pomodoro']) / 60, 1)
total = '{:>26} {:>2}h'.format('Total Work:', hours)
print(Style.BRIGHT + Fore.WHITE + total)
def test_start_time_initialisation(self):
"""
Does the `start` attribute initialise properly?
"""
self.assertEqual(
PomodoroCalculator(end='18:30').start,
datetime(2014, 1, 1, 15, 30),
)
self.assertEqual(
PomodoroCalculator(end='18:30', start='16:30').start,
datetime(2014, 1, 1, 16, 30),
)
def test_pomodori_never_ends_with_break(self):
"""
The list that the `pomodori_schedule` method returns can never end in a
short or long break.
"""
times = [
'{:02d}:{:02d}'.format(h, m)
for m in range(0, 60)
for h in range(0, 24)
]
for time in times:
pomodori = PomodoroCalculator(end=time).pomodori_schedule()
if pomodori:
self.assertEqual(
pomodori['segments'][0].get('type'),
'pomodoro',
)
self.assertNotEqual(
pomodori['segments'][0].get('type'),
'short-break',
)
self.assertNotEqual(
pomodori['segments'][0].get('type'),
'long-break',
)
def test_pomodori_none(self):
"""
Does the `pomodori_schedule` method return `None` if there's
no time?
"""
self.assertIsNone(
PomodoroCalculator(end='00:05').pomodori_schedule()
)
def test_long_break_seconds(self):
"""
Does the `long_break_seconds` property work correctly?
"""
calculator = PomodoroCalculator(end='18:00', long_break=10)
self.assertEqual(self.calculator.long_break_seconds, 15 * 60)
self.assertEqual(calculator.long_break_seconds, 10 * 60)
def test_pomodori_does_not_overflow(self):
"""
The `pomodori_schedule` method should not return any entities that go
past the time limit.
"""
pomodori = PomodoroCalculator(end='15:00').pomodori_schedule()
self.assertLess(
pomodori['segments'][-1]['end'], datetime(2014, 1, 1, 15)
)
def test_pomodori_single_pomodoro(self):
"""
Does the `pomodori_schedule` method return a single Pomodoro if there's
just enough time?
"""
pomodori = PomodoroCalculator(end='00:25').pomodori_schedule()
self.assertEqual(len(pomodori['segments']), 1)
self.assertEqual(pomodori['segments'][0].get('type'), 'pomodoro')
def test_end_time_initialisation(self):
"""
Does the `end` attribute initialise properly?
"""
self.assertEqual(
PomodoroCalculator(end='18:30').end,
datetime(2014, 1, 1, 18, 30),
)
self.assertEqual(
PomodoroCalculator(end='18:30', start='19:30').end,
datetime(2014, 1, 2, 18, 30),
)
self.assertEqual(
PomodoroCalculator(end='10:00', interval=True).end,
datetime(2014, 1, 2, 1, 30),
)
def test_get_item(self):
"""
Does the `_get_item` method work correctly?
"""
pomodoro = PomodoroCalculator(end='03:00')
short_break = PomodoroCalculator(end='03:00')
long_break = PomodoroCalculator(end='03:00')
self.assertDictEqual(
pomodoro._get_item(3600, 'pomodoro', 1),
{
'index': 1,
'pomodori-index': 1,
'type': 'pomodoro',
'start': datetime(2014, 1, 1, 2, 0),
'end': datetime(2014, 1, 1, 2, 25),
'length': 1500,
},
)
self.assertDictEqual(
short_break._get_item(3600, 'short-break', 2),
{
'index': 2,
'pomodori-index': 2,
'type': 'short-break',
'start': datetime(2014, 1, 1, 2, 0),
'end': datetime(2014, 1, 1, 2, 5),
'length': 300,
},
)
self.assertDictEqual(
long_break._get_item(3600, 'long-break', 3),
{
'index': 3,
'pomodori-index': 2,
'type': 'long-break',
'start': datetime(2014, 1, 1, 2, 0),
'end': datetime(2014, 1, 1, 2, 15),
'length': 900,
},
)
def test_pomodori(self):
"""
Does the `pomodori_schedule` method return the correct Pomodori
entities?
"""
pomodori = PomodoroCalculator(end='14:50').pomodori_schedule()
expected_segments = [
(pomodori['segments'][-1]['start'], datetime(2014, 1, 1, 14, 10)),
(pomodori['segments'][-1]['end'], datetime(2014, 1, 1, 14, 35)),
(len([e for e in pomodori['segments'] if e['type'] == 'pomodoro']), 5),
(len([e for e in pomodori['segments'] if e['type'] == 'short-break']), 3),
(len([e for e in pomodori['segments'] if e['type'] == 'long-break']), 1),
]
for expected_segment in expected_segments:
self.assertEqual(expected_segment[0], expected_segment[1])
def test_pomodori_meta_data(self):
"""
Does the `pomodori_schedule` method return the correct meta
data about the Pomodori entities?
"""
pomodori = PomodoroCalculator(end='14:30').pomodori_schedule()
del pomodori['segments']
self.assertDictEqual(
pomodori,
{
'end': datetime(2014, 1, 1, 13, 55),
'start': datetime(2014, 1, 1, 12, 0),
'seconds-per-pomodoro': 1500,
'total-breaks': 3,
'total-pomodori': 4,
'total-rest-time': 900,
'total-work-time': 6000,
},
)
def main():
arguments = docopt(__doc__, version=__version__)
schedule = PomodoroCalculator(
end=arguments['<end-time>'],
start=arguments['--from'],
pomodoro_length=int(arguments['--pomodoro']),
group_length=int(arguments['--group']),
short_break=int(arguments['--break']),
long_break=int(arguments['--long-break']),
interval=arguments['--interval'],
amount=arguments['--amount']
).pomodori_schedule()
if schedule is None:
print('Oops, something went wrong! We need more time for our Pomodoros.')
return
if arguments['--json']:
print(json.dumps(schedule, cls=DateTimeEncoder))
else:
print(report_output(schedule, arguments['--nocolour'], arguments['--extensive-report']))
def setUp(self):
self.calculator = PomodoroCalculator(end='15:00')
class PomodoroTest(unittest.TestCase):
def setUp(self):
self.calculator = PomodoroCalculator(end='15:00')
def test_retriever_class_exists(self):
"""
Does the `PomodoroCalculator` class exist?
"""
self.assertTrue('PomodoroCalculator' in globals())
@freeze_time('2014-01-01 00:00:00')
def test_create_datetime_with_hours(self):
"""
Does `_create_datetime` work correctly with only hours provided?
"""
self.assertEqual(
self.calculator._create_datetime('15'),
datetime(2014, 1, 1, 15),
)
@freeze_time('2014-01-01 00:00:00')
def test_create_datetime_with_hours_and_minutes(self):
"""
Does `_create_datetime` work correctly with hours and minutes?
"""
self.assertEqual(
self.calculator._create_datetime('15:30'),
datetime(2014, 1, 1, 15, 30),
)
@freeze_time('2014-01-01 00:00:00')
def test_create_datetime_with_hours_minutes_and_seconds(self):
"""
Does `_create_datetime` work correctly with hours, minutes and seconds?
"""
self.assertEqual(
self.calculator._create_datetime('15:30:25'),
datetime(2014, 1, 1, 15, 30, 25),
)
@freeze_time('2014-01-02 00:00:00')
def test_create_datetime_with_different_day(self):
"""
Does `_create_datetime` add another day is `tomorrow` is passed?
"""
self.assertEqual(
self.calculator._create_datetime('15:30:25', tomorrow=True),
datetime(2014, 1, 3, 15, 30, 25),
)
def test_compare_times(self):
"""
Does `_compare_times` work correctly?
"""
times = [
('3', '5'),
('03', '05'),
('5:30', '6:30'),
('05:0', '06:0'),
('05:01', '06:00'),
('05:01:05', '06:00:20'),
('05:01:5', '06:00:2'),
]
for t in times:
self.assertFalse(self.calculator._compare_times(t[0], t[1]))
self.assertTrue(self.calculator._compare_times(t[1], t[0]))
@freeze_time('2014-01-01 15:30:00')
def test_start_time_initialisation(self):
"""
Does the `start` attribute initialise properly?
"""
self.assertEqual(
PomodoroCalculator(end='18:30').start,
datetime(2014, 1, 1, 15, 30),
)
self.assertEqual(
PomodoroCalculator(end='18:30', start='16:30').start,
datetime(2014, 1, 1, 16, 30),
)
@freeze_time('2014-01-01 15:30:00')
def test_end_time_initialisation(self):
"""
Does the `end` attribute initialise properly?
"""
self.assertEqual(
PomodoroCalculator(end='18:30').end,
datetime(2014, 1, 1, 18, 30),
)
self.assertEqual(
PomodoroCalculator(end='18:30', start='19:30').end,
datetime(2014, 1, 2, 18, 30),
)
def test_short_break_seconds(self):
"""
Does the `short_break_seconds` property work correctly?
"""
calculator = PomodoroCalculator(end='18:00', short_break=10)
self.assertEqual(self.calculator.short_break_seconds, 5 * 60)
self.assertEqual(calculator.short_break_seconds, 10 * 60)
def test_long_break_seconds(self):
"""
Does the `long_break_seconds` property work correctly?
"""
calculator = PomodoroCalculator(end='18:00', long_break=10)
self.assertEqual(self.calculator.long_break_seconds, 15 * 60)
self.assertEqual(calculator.long_break_seconds, 10 * 60)
def test_total_seconds(self):
"""
Does the `total_seconds` property work correctly?
"""
calculator = PomodoroCalculator(end='18:00', start='18:15')
self.assertEqual(calculator.long_break_seconds, 15 * 60)
@freeze_time('2014-01-01 00:00:00')
def test_get_item(self):
"""
Does the `_get_item` method work correctly?
"""
pomodoro = PomodoroCalculator(end='03:00')
short_break = PomodoroCalculator(end='03:00')
long_break = PomodoroCalculator(end='03:00')
self.assertDictEqual(
pomodoro._get_item(3600, 'pomodoro'),
{
'type': 'pomodoro',
'start': datetime(2014, 1, 1, 2),
'end': datetime(2014, 1, 1, 2, 25),
'time': 1500,
},
)
self.assertDictEqual(
short_break._get_item(3600, 'short-break'),
{
'type': 'short-break',
'start': datetime(2014, 1, 1, 2),
'end': datetime(2014, 1, 1, 2, 5),
'time': 300,
},
)
self.assertDictEqual(
long_break._get_item(3600, 'long-break'),
{
'type': 'long-break',
'start': datetime(2014, 1, 1, 2),
'end': datetime(2014, 1, 1, 2, 15),
'time': 900,
},
)
@freeze_time('2014-01-01 00:00:00')
def test_pomodori_empty(self):
"""
Does the `pomodori_schedule` method return an empty list if there's no
time?
"""
self.assertListEqual(
PomodoroCalculator(end='00:05').pomodori_schedule(),
[],
)
@freeze_time('2014-01-01 00:00:00')
def test_pomodori_single_pomodoro(self):
"""
Does the `pomodori_schedule` method return a single Pomodoro if there's
just enough time?
"""
pomodori = PomodoroCalculator(end='00:25').pomodori_schedule()
self.assertEqual(len(pomodori), 1)
self.assertEqual(pomodori[0].get('type'), 'pomodoro')
@freeze_time('2014-01-01 00:00:00')
def test_pomodori_never_ends_with_break(self):
"""
The list that the `pomodori_schedule` method returns can never end in a
short or long break.
"""
times = [
'{:02d}:{:02d}'.format(h, m)
for m in range(0, 60)
for h in range(0, 24)
]
for time in times:
pomodori = PomodoroCalculator(end=time).pomodori_schedule()
if pomodori:
self.assertEqual(pomodori[0].get('type'), 'pomodoro')
self.assertNotEqual(pomodori[0].get('type'), 'short-break')
self.assertNotEqual(pomodori[0].get('type'), 'long-break')
@freeze_time('2014-01-01 12:00:00')
def test_pomodori(self):
"""
Does the `pomodori_schedule` method return the correct Pomodori
entities?
"""
pomodori = PomodoroCalculator(end='14:35').pomodori_schedule()
expected = [
(pomodori[-1]['start'], datetime(2014, 1, 1, 14, 10)),
(pomodori[-1]['end'], datetime(2014, 1, 1, 14, 35)),
(len([e for e in pomodori if e['type'] == 'pomodoro']), 5),
(len([e for e in pomodori if e['type'] == 'short-break']), 3),
(len([e for e in pomodori if e['type'] == 'long-break']), 1),
]
for expectation in expected:
self.assertEqual(expectation[0], expectation[1])
@freeze_time('2014-01-01 12:00:00')
def test_pomodori_does_not_overflow(self):
"""
The `pomodori_schedule` method should not return any entities that go
past the time limit.
"""
pomodori = PomodoroCalculator(end='15:00').pomodori_schedule()
self.assertLess(pomodori[-1]['end'], datetime(2014, 1, 1, 15))
def test_pomodori_segment_generator(self):
"""
Are 'segment' (Pomodori, short breaks and long breaks) strings
generated correctly?
"""
segments = []
i = 0
for segment in self.calculator.pomodori_segments():
segments.append(segment)
i += 1
if i == 16:
break
self.assertListEqual(
segments,
[
'pomodoro',
'short-break',
'pomodoro',
'short-break',
'pomodoro',
'short-break',
'pomodoro',
'long-break',
'pomodoro',
'short-break',
'pomodoro',
'short-break',
'pomodoro',
'short-break',
'pomodoro',
'long-break',
]
)
@freeze_time('2014-01-01 12:00:00')
def test_create_datetime_with_interval(self):
self.assertEqual(
self.calculator._create_datetime_with_interval(datetime.now(), "02:00"),
datetime(2014, 1, 1, 14),
)
def test_create_time_dict(self):
self.assertDictEqual(
self.calculator._create_time_dict("02:00:01"),
{
'hour':2,
'minute':0,
'second':1,
}
)
def test_interval_option(self):
pomodori = PomodoroCalculator(end='00:55',time_interval=True)
segments = pomodori.pomodori_schedule()
self.assertEqual(len(segments), 3)
def test_interval_option(self):
pomodori = PomodoroCalculator(end='00:55',time_interval=True)
segments = pomodori.pomodori_schedule()
self.assertEqual(len(segments), 3)
def test_total_seconds(self):
"""
Does the `total_seconds` property work correctly?
"""
calculator = PomodoroCalculator(end='18:00', start='18:15')
self.assertEqual(calculator.long_break_seconds, 15 * 60)
class PomodoroTest(unittest.TestCase):
def setUp(self):
self.calculator = PomodoroCalculator(end='15:00')
def test_retriever_class_exists(self):
"""
Does the `PomodoroCalculator` class exist?
"""
self.assertTrue('PomodoroCalculator' in globals())
@freeze_time('2014-01-01 00:00:00')
def test_create_datetime_with_hours(self):
"""
Does `_create_datetime` work correctly with only hours provided?
"""
self.assertEqual(
self.calculator._create_datetime('15'),
datetime(2014, 1, 1, 15),
)
@freeze_time('2014-01-01 00:00:00')
def test_create_datetime_with_hours_and_minutes(self):
"""
Does `_create_datetime` work correctly with hours and minutes?
"""
self.assertEqual(
self.calculator._create_datetime('15:30'),
datetime(2014, 1, 1, 15, 30),
)
@freeze_time('2014-01-01 00:00:00')
def test_create_datetime_with_hours_minutes_and_seconds(self):
"""
Does `_create_datetime` work correctly with hours, minutes and seconds?
"""
self.assertEqual(
self.calculator._create_datetime('15:30:25'),
datetime(2014, 1, 1, 15, 30, 25),
)
def test_create_timedelta_with_hours(self):
"""
Does `_create_timedelta` work correctly with only hours provided?
"""
self.assertEqual(
self.calculator._create_timedelta('15'),
timedelta(hours=15),
)
def test_create_timedelta_with_hours_and_minutes(self):
"""
Does `_create_timedelta` work correctly with hours and minutes?
"""
self.assertEqual(
self.calculator._create_timedelta('15:30'),
timedelta(hours=15, minutes=30),
)
def test_create_timedelta_with_hours_minutes_and_seconds(self):
"""
Does `_create_timedelta` work correctly with hours, minutes and seconds?
"""
self.assertEqual(
self.calculator._create_timedelta('15:30:25'),
timedelta(hours=15, minutes=30, seconds=25),
)
@freeze_time('2014-01-01 15:30:00')
def test_start_time_initialisation(self):
"""
Does the `start` attribute initialise properly?
"""
self.assertEqual(
PomodoroCalculator(end='18:30').start,
datetime(2014, 1, 1, 15, 30),
)
self.assertEqual(
PomodoroCalculator(end='18:30', start='16:30').start,
datetime(2014, 1, 1, 16, 30),
)
@freeze_time('2014-01-01 15:30:00')
def test_end_time_initialisation(self):
"""
Does the `end` attribute initialise properly?
"""
self.assertEqual(
PomodoroCalculator(end='18:30').end,
datetime(2014, 1, 1, 18, 30),
)
self.assertEqual(
PomodoroCalculator(end='18:30', start='19:30').end,
datetime(2014, 1, 2, 18, 30),
)
self.assertEqual(
PomodoroCalculator(end='10:00', interval=True).end,
datetime(2014, 1, 2, 1, 30),
)
def test_short_break_seconds(self):
"""
Does the `short_break_seconds` property work correctly?
"""
calculator = PomodoroCalculator(end='18:00', short_break=10)
self.assertEqual(self.calculator.short_break_seconds, 5 * 60)
self.assertEqual(calculator.short_break_seconds, 10 * 60)
def test_long_break_seconds(self):
"""
Does the `long_break_seconds` property work correctly?
"""
calculator = PomodoroCalculator(end='18:00', long_break=10)
self.assertEqual(self.calculator.long_break_seconds, 15 * 60)
self.assertEqual(calculator.long_break_seconds, 10 * 60)
def test_total_seconds(self):
"""
Does the `total_seconds` property work correctly?
"""
calculator = PomodoroCalculator(end='18:00', start='18:15')
self.assertEqual(calculator.long_break_seconds, 15 * 60)
@freeze_time('2014-01-01 00:00:00')
def test_get_item(self):
"""
Does the `_get_item` method work correctly?
"""
pomodoro = PomodoroCalculator(end='03:00')
short_break = PomodoroCalculator(end='03:00')
long_break = PomodoroCalculator(end='03:00')
self.assertDictEqual(
pomodoro._get_item(3600, 'pomodoro', 1),
{
'index': 1,
'pomodori-index': 1,
'type': 'pomodoro',
'start': datetime(2014, 1, 1, 2, 0),
'end': datetime(2014, 1, 1, 2, 25),
'length': 1500,
},
)
self.assertDictEqual(
short_break._get_item(3600, 'short-break', 2),
{
'index': 2,
'pomodori-index': 2,
'type': 'short-break',
'start': datetime(2014, 1, 1, 2, 0),
'end': datetime(2014, 1, 1, 2, 5),
'length': 300,
},
)
self.assertDictEqual(
long_break._get_item(3600, 'long-break', 3),
{
'index': 3,
'pomodori-index': 2,
'type': 'long-break',
'start': datetime(2014, 1, 1, 2, 0),
'end': datetime(2014, 1, 1, 2, 15),
'length': 900,
},
)
@freeze_time('2014-01-01 00:00:00')
def test_pomodori_none(self):
"""
Does the `pomodori_schedule` method return `None` if there's
no time?
"""
self.assertIsNone(
PomodoroCalculator(end='00:05').pomodori_schedule()
)
@freeze_time('2014-01-01 00:00:00')
def test_pomodori_single_pomodoro(self):
"""
Does the `pomodori_schedule` method return a single Pomodoro if there's
just enough time?
"""
pomodori = PomodoroCalculator(end='00:25').pomodori_schedule()
self.assertEqual(len(pomodori['segments']), 1)
self.assertEqual(pomodori['segments'][0].get('type'), 'pomodoro')
@freeze_time('2014-01-01 00:00:00')
def test_pomodori_never_ends_with_break(self):
"""
The list that the `pomodori_schedule` method returns can never end in a
short or long break.
"""
times = [
'{:02d}:{:02d}'.format(h, m)
for m in range(0, 60)
for h in range(0, 24)
]
for time in times:
pomodori = PomodoroCalculator(end=time).pomodori_schedule()
if pomodori:
self.assertEqual(
pomodori['segments'][0].get('type'),
'pomodoro',
)
self.assertNotEqual(
pomodori['segments'][0].get('type'),
'short-break',
)
self.assertNotEqual(
pomodori['segments'][0].get('type'),
'long-break',
)
@freeze_time('2014-01-01 12:00:00')
def test_pomodori(self):
"""
Does the `pomodori_schedule` method return the correct Pomodori
entities?
"""
pomodori = PomodoroCalculator(end='14:50').pomodori_schedule()
expected_segments = [
(pomodori['segments'][-1]['start'], datetime(2014, 1, 1, 14, 10)),
(pomodori['segments'][-1]['end'], datetime(2014, 1, 1, 14, 35)),
(len([e for e in pomodori['segments'] if e['type'] == 'pomodoro']), 5),
(len([e for e in pomodori['segments'] if e['type'] == 'short-break']), 3),
(len([e for e in pomodori['segments'] if e['type'] == 'long-break']), 1),
]
for expected_segment in expected_segments:
self.assertEqual(expected_segment[0], expected_segment[1])
@freeze_time('2014-01-01 12:00:00')
def test_pomodori_meta_data(self):
"""
Does the `pomodori_schedule` method return the correct meta
data about the Pomodori entities?
"""
pomodori = PomodoroCalculator(end='14:30').pomodori_schedule()
del pomodori['segments']
self.assertDictEqual(
pomodori,
{
'end': datetime(2014, 1, 1, 13, 55),
'start': datetime(2014, 1, 1, 12, 0),
'seconds-per-pomodoro': 1500,
'total-breaks': 3,
'total-pomodori': 4,
'total-rest-time': 900,
'total-work-time': 6000,
},
)
@freeze_time('2014-01-01 12:00:00')
def test_pomodori_does_not_overflow(self):
"""
The `pomodori_schedule` method should not return any entities that go
past the time limit.
"""
pomodori = PomodoroCalculator(end='15:00').pomodori_schedule()
self.assertLess(
pomodori['segments'][-1]['end'], datetime(2014, 1, 1, 15)
)
def test_pomodori_segment_generator(self):
"""
Are 'segment' (Pomodori, short breaks and long breaks) strings
generated correctly?
"""
self.assertListEqual(
list(islice(self.calculator.pomodori_segments(), 16)),
[
'pomodoro',
'short-break',
'pomodoro',
'short-break',
'pomodoro',
'short-break',
'pomodoro',
'long-break',
'pomodoro',
'short-break',
'pomodoro',
'short-break',
'pomodoro',
'short-break',
'pomodoro',
'long-break',
]
)
@freeze_time('2018-09-01 0:00:00')
def test_amount_functionality(self):
"""
Does specifying desired amount of pomodori work?
"""
pomodori = PomodoroCalculator(end='12', amount=True).pomodori_schedule()
del pomodori['segments']
# Use known functionality to confirm correctness
pomodori_reference = PomodoroCalculator(end='06:39:00').pomodori_schedule()
del pomodori_reference['segments']
self.assertDictEqual(
pomodori,
{
'end': datetime(2018, 9, 1, 6, 15),
'start': datetime(2018, 9, 1, 0, 0),
'seconds-per-pomodoro': 1500,
'total-breaks': 11,
'total-pomodori': 12,
'total-rest-time': 4500,
'total-work-time': 18000,
}
)
self.assertDictEqual(pomodori_reference, pomodori)
# Try shorter item
pomodori_short = PomodoroCalculator(end='1', amount=True).pomodori_schedule()
del pomodori_short['segments']
self.assertDictEqual(
pomodori_short,
{
'end': datetime(2018, 9, 1, 0, 25),
'start': datetime(2018, 9, 1, 0, 0),
'seconds-per-pomodoro': 1500,
'total-breaks': 0,
'total-pomodori': 1,
'total-rest-time': 0,
'total-work-time': 1500,
}
)
pomodori_final = PomodoroCalculator(end='3', amount=True).pomodori_schedule()
del pomodori_final['segments']
self.assertDictEqual(
pomodori_final,
{
'end': datetime(2018, 9, 1, 1, 25),
'start': datetime(2018, 9, 1, 0, 0),
'seconds-per-pomodoro': 1500,
'total-breaks': 2,
'total-pomodori': 3,
'total-rest-time': 600,
'total-work-time': 4500,
}
)
def test_humanise_seconds(self):
"""
Does the `humanise_seconds` utility function work correctly?
"""
self.assertEqual(humanise_seconds(60), '1 minute')
self.assertEqual(humanise_seconds(120), '2 minutes')
self.assertEqual(humanise_seconds(65), '1 minute')
self.assertEqual(humanise_seconds(3600), '1 hour')
self.assertEqual(humanise_seconds(7200), '2 hours')
self.assertEqual(humanise_seconds(7265), '2 hours, 1 minute')
def setUp(self):
self.calculator = PomodoroCalculator(end='15:00')
