blitztext 0.2.0__tar.gz → 0.2.1__tar.gz

{blitztext-0.2.0 → blitztext-0.2.1}/Cargo.lock

@@ -49,7 +49,7 @@ dependencies = [
 
 [[package]]
 name = "blitztext"
-version = "0.2.0"
+version = "0.2.1"
 dependencies = [
  "bincode",
  "criterion",

{blitztext-0.2.0 → blitztext-0.2.1}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "blitztext"
-version = "0.2.0"
+version = "0.2.1"
 edition = "2021"
 authors = ["Praise Oketola <oketola.praise@gmail.com>"]
 description = "A library for fast keyword extraction and replacement in strings."

{blitztext-0.2.0 → blitztext-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blitztext
-Version: 0.2.0
+Version: 0.2.1
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

{blitztext-0.2.0 → blitztext-0.2.1}/src/keyword_processor.rs

@@ -26,7 +26,7 @@ pub struct KeywordProcessor {
     /// Faster read-only copy of `root`, dropped when keywords change.
     #[serde(skip)]
     compiled: OnceLock<CompiledTrie>,
-    /// Number of nodes in `root`; counted on first use after deserializing.
+    /// Number of nodes in `root`, kept current by add and remove; counted once after deserializing.
     #[serde(skip)]
     node_count: OnceLock<usize>,
     /// Matching work done with `root` since keywords last changed, in text bytes.
@@ -84,7 +84,7 @@ impl KeywordProcessor {
             case_sensitive,
             allow_overlaps,
             compiled: OnceLock::new(),
-            node_count: OnceLock::new(),
+            node_count: OnceLock::from(1),
             work_since_change: AtomicUsize::new(0),
         }
     }
@@ -1594,6 +1594,17 @@ mod tests {
         assert_eq!(kp.replace_keywords("İstanbul istanbul", None), "İstanbul IST");
     }
 
+    #[test]
+    fn node_count_stays_exact_through_adds_and_removes() {
+        let mut kp = KeywordProcessor::new();
+        for keyword in ["rust", "rusty", "go", "café"] {
+            kp.add_keyword(keyword, None);
+        }
+        kp.remove_keyword("rusty");
+        kp.remove_keyword("go");
+        assert_eq!(kp.node_count.get(), Some(&kp.root.count_nodes()));
+    }
+
     #[test]
     fn non_ascii_keywords_can_be_listed_looked_up_and_removed() {
         let mut kp = KeywordProcessor::new();