Added tests for b_prefixes()

gmossessian · Aug 18, 2014 · 566bd81 · 566bd81
1 parent 4fe675b
commit 566bd81
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_dawg.py b/tests/test_dawg.py
@@ -182,6 +182,12 @@ def test_prefixes(self):
         assert d.prefixes("x") == []
         assert d.prefixes("bar") == ["bar"]
 
+    def test_b_prefixes(self):
+        d = self.dawg()
+        assert d.b_prefixes(b"foobarz") == [b"f", b"foo", b"foobar"]
+        assert d.b_prefixes(b"x") == []
+        assert d.b_prefixes(b"bar") == [b"bar"]
+
     def test_iterprefixes(self):
         d = self.dawg()
         assert list(d.iterprefixes("foobarz")) == d.prefixes("foobarz")