Task 2 Python LK

1 files changed, 48 insertions, 0 deletions

diff --git a/challenge-053/lubos-kolouch/python/ch-2.py b/challenge-053/lubos-kolouch/python/ch-2.py
new file mode 100644
index 0000000000..ebde8125fc
--- /dev/null
+++ b/challenge-053/lubos-kolouch/python/ch-2.py
@@ -0,0 +1,48 @@
+#!/usr/bin/env python
+
+from itertools import permutations
+
+
+class VowelString:
+
+    def check_valid(self, permutation):
+
+        ok_rules = ['ae', 'ai', 'ei', 'ia', 'ie',
+                    'io', 'iu', 'oa', 'ou', 'uo', 'ue']
+
+        for x, item in enumerate(permutation[:-1]):
+            if item + permutation[x+1] not in ok_rules:
+                return 0
+
+            return 1
+
+    def get_permutations(self,  count: int):
+
+        wovels = ['a', 'e', 'i', 'o', 'u']
+
+        result = []
+
+        for perm in permutations(wovels, count):
+            if self.check_valid(perm):
+                result.append(perm)
+
+        return(result)
+
+
+class TestVowelString:
+
+    def __init__(self):
+        self.vow = VowelString()
+
+    def do_tests(self):
+        assert len(self.vow.get_permutations(1)) == 0
+        assert len(self.vow.get_permutations(2)) == 11
+
+
+if __name__ == "__main__":
+    vowcomb = VowelString()
+
+    print(vowcomb.get_permutations(2))
+
+    vowtester = TestVowelString()
+    vowtester.do_tests()