Add simple test for pattern API

This commit is contained in:
Manish Goregaokar 2017-12-17 14:44:03 -08:00
parent 75c07a37ff
commit efcc447ebf
2 changed files with 78 additions and 0 deletions

View file

@ -28,6 +28,7 @@
#![feature(iter_rfind)]
#![feature(iter_rfold)]
#![feature(nonzero)]
#![feature(pattern)]
#![feature(raw)]
#![feature(refcell_replace_swap)]
#![feature(sip_hash_13)]
@ -61,6 +62,7 @@ mod nonzero;
mod num;
mod ops;
mod option;
mod pattern;
mod ptr;
mod result;
mod slice;

View file

@ -0,0 +1,76 @@
use std::str::pattern::*;
// This macro makes it easier to write
// tests that do a series of iterations
macro_rules! search_asserts {
($haystack:expr, $needle:expr, $testname:expr, [$($func:ident),*], $result:expr) => {
let mut searcher = $needle.into_searcher($haystack);
let arr = [$( Step::from(searcher.$func()) ),+];
assert_eq!(&arr[..], &$result, $testname);
}
}
/// Combined enum for the results of next() and next_match()/next_reject()
#[derive(Debug, PartialEq, Eq)]
enum Step {
// variant names purposely chosen to
// be the same length for easy alignment
Matches(usize, usize),
Rejects(usize, usize),
InRange(usize, usize),
Done
}
use Step::*;
impl From<SearchStep> for Step {
fn from(x: SearchStep) -> Self {
match x {
SearchStep::Match(a, b) => Matches(a, b),
SearchStep::Reject(a, b) => Rejects(a, b),
SearchStep::Done => Done
}
}
}
impl From<Option<(usize, usize)>> for Step {
fn from(x: Option<(usize, usize)>) -> Self {
match x {
Some((a, b)) => InRange(a, b),
None => Done
}
}
}
#[test]
fn test_simple_iteration() {
search_asserts! ("abcdeabcd", 'a', "forward iteration for ASCII string",
// a b c d e a b c d EOF
[next, next, next, next, next, next, next, next, next, next],
[Matches(0, 1), Rejects(1, 2), Rejects(2, 3), Rejects(3, 4), Rejects(4, 5), Matches(5, 6), Rejects(6, 7), Rejects(7, 8), Rejects(8, 9), Done]
);
search_asserts! ("abcdeabcd", 'a', "reverse iteration for ASCII string",
// d c b a e d c b a EOF
[next_back, next_back, next_back, next_back, next_back, next_back, next_back, next_back, next_back, next_back],
[Rejects(8, 9), Rejects(7, 8), Rejects(6, 7), Matches(5, 6), Rejects(4, 5), Rejects(3, 4), Rejects(2, 3), Rejects(1, 2), Matches(0, 1), Done]
);
search_asserts! ("我爱我的猫", '我', "forward iteration for Chinese string",
// 我 愛 我 的 貓 EOF
[next, next, next, next, next, next],
[Matches(0, 3), Rejects(3, 6), Matches(6, 9), Rejects(9, 12), Rejects(12, 15), Done]
);
search_asserts! ("我的猫说meow", 'm', "forward iteration for mixed string",
// 我 的 猫 说 m e o w EOF
[next, next, next, next, next, next, next, next, next],
[Rejects(0, 3), Rejects(3, 6), Rejects(6, 9), Rejects(9, 12), Matches(12, 13), Rejects(13, 14), Rejects(14, 15), Rejects(15, 16), Done]
);
search_asserts! ("我的猫说meow", '猫', "reverse iteration for mixed string",
// w o e m 说 猫 的 我 EOF
[next_back, next_back, next_back, next_back, next_back, next_back, next_back, next_back, next_back],
[Rejects(15, 16), Rejects(14, 15), Rejects(13, 14), Rejects(12, 13), Rejects(9, 12), Matches(6, 9), Rejects(3, 6), Rejects(0, 3), Done]
);
}