Add basic benchmarks

benches/benchmarks.rs

@@ -0,0 +1,45 @@
+#![feature(test)]
+
+extern crate test;
+
+#[cfg(test)]
+mod tests {
+    use starship::{modules, print};
+    use test::Bencher;
+    use clap::{App, Arg};
+
+    #[bench]
+    fn full_prompt_bench(b: &mut Bencher) {
+        b.iter(||{
+            let args = App::new("starship")
+                .arg(Arg::with_name("status_code"))
+                .get_matches_from(vec!["starship", "0"]);
+
+            starship::print::prompt(args)
+        });
+    }
+
+    #[bench]
+    fn char_section_bench(b: &mut Bencher) {
+        b.iter(|| {
+            let args = App::new("starship")
+                .arg(Arg::with_name("status_code"))
+                .get_matches_from(vec!["starship", "0"]);
+
+            let segment = modules::handle("char", &args);
+            print::print_segment(segment)
+        });
+    }
+
+    #[bench]
+    fn dir_section_bench(b: &mut Bencher) {
+        b.iter(|| {
+            let args = App::new("starship")
+                .arg(Arg::with_name("status_code"))
+                .get_matches_from(vec!["starship", "0"]);
+
+            let segment = modules::handle("dir", &args);
+            print::print_segment(segment)
+        });
+    }
+}

src/lib.rs

@@ -0,0 +1,2 @@
+pub mod modules;
+pub mod print;

src/main.rs

@@ -1,6 +1,5 @@
 #[macro_use]
 extern crate clap;
-extern crate ansi_term;
 
 mod modules;
 mod print;

src/print.rs

@@ -11,7 +11,7 @@ pub fn prompt(args: ArgMatches) {
     }
 }
 
-fn print_segment(segment: Segment) {
+pub fn print_segment(segment: Segment) {
     let Segment {
         prefix,
         value,