Added tests for functions

1 files changed, 10 insertions, 0 deletions

diff --git a/src/sexp/mod.rs b/src/sexp/mod.rs
index 11c185f..4bb4fa2 100644
--- a/src/sexp/mod.rs
+++ b/src/sexp/mod.rs
@@ -35,6 +35,8 @@ pub enum SLeaf {
 
     Let,
 
+    Fun,
+
     Ty(Type),
     Arr,
 
@@ -75,6 +77,7 @@ impl SExp {
     pub fn is_value(&self) -> bool {
         match self {
             SCons(a, b) =>
+		scons(a.clone(), b.clone()).is_fun() ||
                 SCons(a.clone(), b.clone()).check_let().is_some() ||
                 (
                     (**a == Atom(Quote) || **a == Atom(Vector) || (**a).is_type_lit())
@@ -102,4 +105,11 @@ impl SExp {
 	    _ => false
 	}
     }
+
+    pub fn is_fun(&self) -> bool {
+	match self {
+	    SCons(a, _) if **a == Atom(Fun) => true,
+	    _ => false
+	}
+    }
 }