From 8c08bb9e14daabe07af681ec5b8de81fd8be3a3f Mon Sep 17 00:00:00 2001
From: Dirkjan Ochtman <dirkjan@ochtman.nl>
Date: Thu, 4 Feb 2021 11:20:36 +0100
Subject: [PATCH] Add check_segments function

---
 src/test_cases.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/test_cases.rs b/src/test_cases.rs
index 2ed9076..0f53c99 100644
--- a/src/test_cases.rs
+++ b/src/test_cases.rs
@@ -15,6 +15,12 @@ pub fn assert_segments(segmenter: &Segmenter, s: &[&str]) {
     assert_eq!(cmp, s);
 }
 
+pub fn check_segments(segmenter: &Segmenter, s: &[&str]) -> bool {
+    let mut out = Vec::new();
+    segmenter.segment(&s.join(""), &mut out);
+    s == out.iter().map(|s| &*s).collect::<Vec<_>>()
+}
+
 /// Built-in test cases
 ///
 /// These are exposed so that you can test with different data sources.