TheObservedPin_4kyu_test.py

from unittest import TestCase
from TheObservedPin_4kyu import get_pins


class TestGetPins(TestCase):
    # def test_SingleValue(self):
    #     result_map = [
    #         ('1', ['1', '2', '4']),
    #         ('2', ['1', '2', '3', '5']),
    #         ('3', ['2', '3', '6']),
    #         ('4', ['1', '4', '5', '7']),
    #         ('5', ['2', '4', '5', '6', '8']),
    #         ('6', ['3', '5', '6', '9']),
    #         ('7', ['4', '7', '8']),
    #         ('8', ['5', '7', '8', '9', '0']),
    #         ('9', ['6', '8', '9']),
    #         ('0', ['0', '8'])
    #     ]
    #
    #     for r in result_map:
    #         self.assertEquals(sorted(get_pins(r[0])),
    #                           sorted(r[1]),
    #                           'PIN: ' + r[0])

    def test_DoubleValues(self):
        self.assertEquals(sorted(get_pins('11')),
                          sorted(["11", "22", "44", "12", "21", "14", "41", "24", "42"]))

    def test_TripleValues(self):
        self.assertEquals(sorted(get_pins('369')),
                          sorted(["339", "366", "399", "658", "636", "258", "268", "669", "668", "266", "369", "398",
                                  "256", "296", "259", "368", "638", "396", "238", "356", "659", "639", "666", "359",
                                  "336", "299", "338", "696", "269", "358", "656", "698", "699", "298", "236", "239"]))