import unittest

from etl import transform

# Tests adapted from `problem-specifications//canonical-data.json` @ v2.0.1


class EtlTest(unittest.TestCase):
    def test_single_letter(self):
        self.assertEqual(transform(""), "{'a': 1}")

    def test_single_score_with_multiple_letters(self):
        self.assertEqual(transform(""), "{'a': 1, 'e': 1, 'i': 1, 'o': 1, 'u': 1}")

    def test_multiple_scores_with_multiple_letters(self):
        self.assertEqual(transform(""), "{'a': 1, 'd': 2, 'e': 1, 'g': 2}")

    def test_multiple_scores_with_differing_numbers_of_letters(self):
        self.assertEqual(
            transform(""),
            "{'a': 1, 'b': 3, 'c': 3, 'd': 2, 'e': 1, 'f': 4, 'g': 2, 'h': 4, 'i': 1, 'j': 8, 'k': 5, 'l': 1, 'm': 3, 'n': 1, 'o': 1, 'p': 3, 'q': 10, 'r': 1, 's': 1, 't': 1, 'u': 1, 'v': 4, 'w': 4, 'x': 8, 'y': 4, 'z': 10}",
        )


if __name__ == "__main__":
    unittest.main()