nicestrings

nicestrings.py

#!/usr/bin/env python
import re

def is_nice(s):
    return (len(re.findall(r"[aeiou]", s)) >= 3 and
            bool(re.search(r"(\w)\1+", s)) and not
            bool(re.search(r"ab|cd|pq|xy", s)))

if __name__ == "__main__":
    with open('input1.txt', 'r') as f:
        print sum(is_nice(s) for s in f.readlines())

test_nicestrings.py

from nicestrings import is_nice

def test_is_nice_nice_string():
    # req 1: 3 vowels
    # req 2: 2 consecutive letters
    # req 3: no ab, cd, pq, xy
    assert(is_nice("aeibb") == True)

def test_is_nice_fails_req_1():
    # not enough vowels
    assert(is_nice("aaxx") == False)

def test_is_nice_fails_req_2():
    assert(is_nice("aei") == False)

def test_is_nice_fails_req_3():
    assert(is_nice("aabbccdd") == False)