init

tests/wordplex_test.py

@@ -39,6 +39,9 @@ def test_pattern():
     wp.set_format("VC")
     assert wp.get_pattern() == [wp.vowels, wp.consonants]
 
+    wp.set_format("@#")
+    assert wp.get_pattern() == [wp.letters, wp.numbers]
+
 
 def test_set_format():
     wp.set_format("CVV")

wordplex/wordplex.py

@@ -31,6 +31,7 @@ def __init__(self):
         self._format = "VC"
         self._suffix = ""
         self._prefix = ""
+        self.alphabet_symbol = "@"
 
     def reset(self):
         self.set_suffix("")
@@ -57,8 +58,8 @@ def set_format_by_word(self, word: str):
         for a in word:
             if self.is_positive_integer(a):
                 new_format += "#"
-            if a.lower() == "@":
-                new_format += "@"
+            if a.lower() == self.alphabet_symbol:
+                new_format += self.alphabet_symbol
             if a.lower() in self.vowels:
                 new_format += "V"
             if a.lower() in self.consonants:
@@ -92,7 +93,7 @@ def get_pattern(self):
                 pattern.append(self.vowels)
             elif letter == "#":
                 pattern.append(self.numbers)
-            elif letter == "@":
+            elif letter == self.alphabet_symbol:
                 pattern.append(self.letters)
             else:
                 pattern.append([letter])