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_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_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_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_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 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)
def setUp(self):
self.calculator = PomodoroCalculator(end='15:00')
