From d6d1ec80ffcc0b13234b170a91b920371078a027 Mon Sep 17 00:00:00 2001
From: Joel Kronqvist <joel.kronqvist@iki.fi>
Date: Sun, 10 Aug 2025 19:16:51 +0300
Subject: Added tests for functions

---
 src/sexp/mod.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'src/sexp/mod.rs')

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
+	}
+    }
 }
-- 
cgit v1.2.3