verify: move out some code to utils

1 files changed, 20 insertions, 1 deletions

diff --git a/tests/test_utils.py b/tests/test_utils.py
index a2033ac..38d50a7 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -1,7 +1,7 @@
 import pytest
 
 from fuzzycat.utils import (author_similarity_score, cut, jaccard, nwise, slugify_string,
-                            token_n_grams, tokenize_string)
+                            token_n_grams, tokenize_string, parse_page_string, dict_key_exists)
 
 
 def test_slugify_string():
@@ -63,3 +63,22 @@ def test_nwise():
     assert list(nwise("1234")) == [("1", "2"), ("3", "4")]
     assert list(nwise("1234", n=1)) == [("1", ), ("2", ), ("3", ), ("4", )]
     assert list(nwise([1, 2, 3, 4, 5], n=3)) == [(1, 2, 3), (4, 5)]
+
+def test_dict_key_exists():
+    assert dict_key_exists({}, "") is False
+    assert dict_key_exists({"a": "a"}, "a") == True
+    assert dict_key_exists({"a": "a"}, "b") == False
+    assert dict_key_exists({"a": {"b": "c"}}, "a.b") == True
+    assert dict_key_exists({"a": {"b": None}}, "a.b") == True
+    assert dict_key_exists({"a": {"b": "c"}}, "a.b.c") == False
+
+def test_page_page_string():
+    reject = ("", "123-2", "123-120", "123a-124", "-2-1")
+    for s in reject:
+        with pytest.raises(ValueError):
+            assert parse_page_string(s)
+    assert parse_page_string("123") == (123, 123, 1)
+    assert parse_page_string("123-5") == (123, 125, 3)
+    assert parse_page_string("123-125") == (123, 125, 3)
+    assert parse_page_string("123-124a") == (123, 124, 2)
+    assert parse_page_string("1-1000") == (1, 1000, 1000)