Add tests for PR #93

jsvine · Jul 15, 2019 · 8b8d8a8 · 8b8d8a8
1 parent a7b7b1a
commit 8b8d8a8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/test_basic.py b/test/test_basic.py
@@ -149,7 +149,12 @@ def test_bad_json(self):
 
     def test_custom_regex(self):
         with self.assertRaises(Exception) as context:
-            model = markovify.NewlineText('This sucks.', reject_reg=r'sucks')
+            model = markovify.NewlineText('This sentence contains a custom bad character: #.', reject_reg=r'#')
+
+        with self.assertRaises(Exception) as context:
+            model = markovify.NewlineText('This sentence (would normall fail')
+
+        model = markovify.NewlineText('This sentence (would normall fail', well_formed = False)
 
 if __name__ == '__main__':
     unittest.main()