roc/crates/compiler/builtins/bitcode/tests/bitcode_test.rs

#[macro_use]
extern crate pretty_assertions;

#[cfg(test)]
mod bitcode {
    use roc_builtins_bitcode::{count_segments_, str_split_};

    #[test]
    fn count_segments() {
        assert_eq!(
            count_segments_((&"hello there").as_bytes(), (&"hello").as_bytes()),
            2
        );
        assert_eq!(
            count_segments_((&"a\nb\nc").as_bytes(), (&"\n").as_bytes()),
            3
        );
        assert_eq!(
            count_segments_((&"str").as_bytes(), (&"delimiter").as_bytes()),
            1
        );
    }

    #[test]
    fn str_split() {
        fn splits_to(string: &str, delimiter: &str, expectation: &[&[u8]]) {
            assert_eq!(
                str_split_(
                    &mut [(&"").as_bytes()].repeat(expectation.len()),
                    &string.as_bytes(),
                    &delimiter.as_bytes()
                ),
                expectation
            );
        }

        splits_to(
            "a!b!c",
            "!",
            &[(&"a").as_bytes(), (&"b").as_bytes(), (&"c").as_bytes()],
        );

        splits_to(
            "a!?b!?c!?",
            "!?",
            &[
                (&"a").as_bytes(),
                (&"b").as_bytes(),
                (&"c").as_bytes(),
                (&"").as_bytes(),
            ],
        );

        splits_to("abc", "!", &[(&"abc").as_bytes()]);

        splits_to(
            "tttttghittttt",
            "ttttt",
            &[(&"").as_bytes(), (&"ghi").as_bytes(), (&"").as_bytes()],
        );

        splits_to("def", "!!!!!!", &[(&"def").as_bytes()]);
    }
}