datadog interview question

match_pattern.py

# Python 3.6

import re
import logging
import unittest
from itertools import zip_longest

logging.basicConfig(level=logging.DEBUG)


def match_pattern(pattern, test):
    new_pattern = ''.join(['_' * int(x) if x.isdigit() else x for x in pattern])
    logging.debug(new_pattern)

    for x, y in zip_longest(new_pattern, test):
        logging.debug(f'{x}, {y}')
        if x == y:
            continue
        elif x == '_':
            continue
        else:
            return False
    return True


def test_match_pattern():
    success = [('dat3g', 'datadog'), ]
    failure = [('dat2g', 'datadog'), ]

    for s in success:
        assert match_pattern(pattern=s[0], test=s[1]) is True

    for f in failure:
        assert match_pattern(pattern=f[0], test=f[1]) is False


if __name__ == '__main__':

    test_match_pattern()