fix regex.group()

totalgood · Dec 1, 2019 · b180bfb · b180bfb
1 parent 0cd58b1
commit b180bfb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/nlpia/loaders.py b/src/nlpia/loaders.py
@@ -241,7 +241,7 @@ def load_anki_df(language='deu'):
     """ Load into a DataFrame statements in one language along with their translation into English
 
     >>> df = get_data('zsm')
-    >>> list(df.columns)
+    >>> list(list(df.columns)[:2])
     ['eng', 'zsm']
     >>> len(df) > 100
     True

diff --git a/src/nlpia/regexes.py b/src/nlpia/regexes.py
@@ -170,10 +170,10 @@ class REPattern:
     >>> pattern.match('Aaron Swartz')
     <re.Match object; span=(0, 12), match='Aaron Swartz'>
     >>> pattern.fullmatch('Aaron Swartz!!')
-    >>> pattern.fullmatch('Aaron Swartz')
-    <regex.Match object; span=(0, 12), match='Aaron Swartz'>
-    >>> pattern.match('Aaron Swartz!!')
-    <re.Match object; span=(0, 12), match='Aaron Swartz'>
+    >>> pattern.fullmatch('Aaron Swartz').group()
+    'Aaron Swartz'
+    >>> pattern.match('Aaron Swartz!!').group()
+    'Aaron Swartz'
     """
 
     def __init__(self, pattern):