1 files changed, 17 insertions, 4 deletions

diff --git a/compiler/src/ast/mod.rs b/compiler/src/ast/mod.rs
index 4587ea7..8e35baf 100644
--- a/compiler/src/ast/mod.rs
+++ b/compiler/src/ast/mod.rs
@@ -7,7 +7,10 @@ use crate::span::Spanned;
 pub use error::Error;
 pub use parser::Parser;
 
-#[derive(Clone, Debug)]
+#[cfg(test)]
+mod tests;
+
+#[derive(Clone, Debug, PartialEq)]
 pub enum Atom {
     Float(f64),
     Integer(i64),
@@ -17,11 +20,21 @@ pub enum Atom {
     Nil,
 }
 
-#[derive(Clone, Debug)]
+#[derive(Clone, Debug, PartialEq)]
 pub enum Expr {
     Atom(Atom),
     List(Vec<Spanned<Expr>>),
 }
 
-#[derive(Clone, Debug)]
-pub struct Program(pub Vec<Spanned<Expr>>);
+#[derive(Clone, Debug, PartialEq)]
+pub struct Program(Vec<Spanned<Expr>>);
+
+impl Program {
+    pub fn inner(&self) -> &[Spanned<Expr>] {
+        &self.0
+    }
+
+    pub fn into_inner(self) -> Vec<Spanned<Expr>> {
+        self.0
+    }
+}