COMP-880 Lab 3 Grading Tests

test_lab3_for_grading.py

"""Tests for lab module."""
import pytest
import schedule


def test_schedule_0():
    """Tests an argument of None, and ensures that no error occurs."""
    sch = schedule.Schedule(None, ['Math'])
    classes: list = sch.classes_for('Mon')
    for day in ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']:
        assert sch.classes_for(day) == ['Free day']


def test_schedule_1():
    """Tests an argument of None, and ensures that no error occurs."""
    with pytest.raises(TypeError):
        sch = schedule.Schedule(['Mon'], False)


def test_schedule_2():
    """Tests happy path."""
    sch = schedule.Schedule(['Tue', 'Thu'], [['Algebra', 'Chemistry'], ['French']])
    assert sch.classes_for('Tue') == ['Algebra', 'Chemistry']
    assert sch.classes_for('Thu') == ['French']
    assert sch.classes_for('Sun') == ['Free day']
    assert sch.classes_for('Holiday') == ['No such day']


def test_schedule_3():
    """Tests make_schedule does overrides old schedule."""
    sch = schedule.Schedule(['Tue', 'Thu'], [['Algebra', 'Chemistry'], ['French']])
    sch.make_schedule(['Mon'], [['Biology']])
    assert sch.classes_for('Mon') == ['Biology']
    assert sch.classes_for('Tue') == ['Free day']
    assert sch.classes_for('Thu') == ['Free day']
    assert sch.classes_for('Sun') == ['Free day']
    assert sch.classes_for('Holiday') == ['No such day']


def test_schedule_4():
    """Tests an argument of a list with positive and negative elements."""
    sch = schedule.Schedule(['Tue', 'Thu'], [['Algebra', 'Chemistry'], ['French']])
    print(str(sch))
    assert str(sch) == """Tue: Algebra, Chemistry
Thu: French
"""