Improve error handling

1 files changed, 8 insertions, 2 deletions

diff --git a/benches/oneshot.rs b/benches/oneshot.rs
index 6ce81a63..9a57825d 100644
--- a/benches/oneshot.rs
+++ b/benches/oneshot.rs
@@ -44,7 +44,13 @@ fn bench_scan(iai: &mut Iai) {
 }
 
 fn bench_tokenize(iai: &mut Iai) {
-    iai.run(|| Tokens::new(black_box(SRC), black_box(TokenMode::Markup)).count());
+    iai.run(|| {
+        Tokens::new(
+            black_box(&SourceFile::detached(SRC)),
+            black_box(TokenMode::Markup),
+        )
+        .count()
+    });
 }
 
 fn bench_parse(iai: &mut Iai) {
@@ -53,7 +59,7 @@ fn bench_parse(iai: &mut Iai) {
 
 fn bench_eval(iai: &mut Iai) {
     let (mut ctx, id) = context();
-    let ast = ctx.parse(id).unwrap();
+    let ast = ctx.sources.get(id).ast().unwrap();
     iai.run(|| eval(&mut ctx, id, &ast).unwrap());
 }