creating tests

test_driven_function.py

import unittest

class TestParseGrades(unittest.TestCase):
    def test_join_string_grades_success(self):
        actual = parse_grades('2, 4, 6, 8, 10, 12')
        expected = ['2','4','6','8','10','12']
        self.assertEqual(actual, expected)

    def test_parse_grades_success(self):
        actual = parse_grades('K-4')
        expected = ['K', '1', '2', '3', '4']
        self.assertEqual(actual, expected)

    def test_multiple_separators(self):
        actual = parse_grades('2-4, 6, 9-12')
        expected = ['2', '3', '4', '6', '9', '10', '11', '12']
        self.assertEqual(actual, expected)

    def test_grades_list_with_ampersand(self):
        actual = parse_grades('6-12 & Ungraded')
        expected = ['6', '7', '8', '9', '10', '11', '12', 'Ungraded']
        self.assertEqual(actual, expected)

    def test_small_lists(self):
        actual = parse_grades('PK-K')
        expected = ['PK', 'K']
        self.assertEqual(actual, expected)

    def test_multiple_separators2(self):
        actual = parse_grades('PK-K, 2-4, 6 & Ungraded')
        expected = ['PK', 'K', '2', '3', '4', '6', 'Ungraded']
        self.assertEqual(actual, expected)

    def test_multiple_separators3(self):
        actual = parse_grades('5-10')
        expected = ['5', '6', '7', '8', '9', '10']
        self.assertEqual(actual, expected)

    def test_multiple_separators4(self):
        actual = parse_grades('K, 5-12')
        expected = ['K', '5', '6', '7', '8', '9', '10', '11', '12']
        self.assertEqual(actual, expected)

    def test_multiple_separators5(self):
        actual = parse_grades( 'PK-3, 5-7')
        expected = ['PK', 'K', '1', '2', '3', '5', '6', '7']
        self.assertEqual(actual, expected)

    def test_complete_dataset(self):
        # create a loop that goes thru dataset and invoke parse_grades with each element
        # used this to create dictionary of grade combo and parsed_grades
        separated_grades_list = []
        for i in unique_grades_combination:
            separated_grades_list.append(parse_grades(i))

        return separated_grades_list