Implement +, * and - primitives

This commit is contained in:
2025-08-10 14:52:38 +01:00
parent 472a682757
commit b20a6749f7
7 changed files with 238 additions and 2 deletions

View File

@@ -14,6 +14,7 @@ add_test_suites(
eval_tests.c
expr_tests.c
parse_tests.c
prim_tests.c
store_tests.c
token_tests.c
)

147
tests/prim_tests.c Normal file
View File

@@ -0,0 +1,147 @@
#include "env.h"
#include "prim.h"
#include "unity.h"
static am_t am;
static store_t store;
void setUp(void)
{
am_init(&am);
store_init(&store);
env_init(&am, &store);
}
void tearDown(void)
{
}
static void test_add_empty_list_is_0(void)
{
am.expr = expr_str_symbol(&store, "+");
env_fetch(&am);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_PRIM_PROC, am.val->atom.type);
am.argl = expr_empty_list(&store);
am.val->atom.prim_proc(&am, &store);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_INTEGER, am.val->atom.type);
TEST_ASSERT_EQUAL(0, am.val->atom.integer);
}
static void test_add_1_2_3_is_6(void)
{
am.expr = expr_str_symbol(&store, "+");
env_fetch(&am);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_PRIM_PROC, am.val->atom.type);
am.argl = expr_pair(
&store, expr_integer(&store, 1),
expr_pair(
&store, expr_integer(&store, 2),
expr_pair(
&store, expr_integer(&store, 3), expr_empty_list(&store))));
am.val->atom.prim_proc(&am, &store);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_INTEGER, am.val->atom.type);
TEST_ASSERT_EQUAL(6, am.val->atom.integer);
}
static void test_mul_empty_list_is_1(void)
{
am.expr = expr_str_symbol(&store, "*");
env_fetch(&am);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_PRIM_PROC, am.val->atom.type);
am.argl = expr_empty_list(&store);
am.val->atom.prim_proc(&am, &store);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_INTEGER, am.val->atom.type);
TEST_ASSERT_EQUAL(1, am.val->atom.integer);
}
static void test_mul_2_3_4_is_24(void)
{
am.expr = expr_str_symbol(&store, "*");
env_fetch(&am);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_PRIM_PROC, am.val->atom.type);
am.argl = expr_pair(
&store, expr_integer(&store, 2),
expr_pair(
&store, expr_integer(&store, 3),
expr_pair(
&store, expr_integer(&store, 4), expr_empty_list(&store))));
am.val->atom.prim_proc(&am, &store);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_INTEGER, am.val->atom.type);
TEST_ASSERT_EQUAL(24, am.val->atom.integer);
}
static void test_sub_1_is_minus_1(void)
{
am.expr = expr_str_symbol(&store, "-");
env_fetch(&am);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_PRIM_PROC, am.val->atom.type);
am.argl = expr_pair(
&store, expr_integer(&store, 1), expr_empty_list(&store));
am.val->atom.prim_proc(&am, &store);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_INTEGER, am.val->atom.type);
TEST_ASSERT_EQUAL(-1, am.val->atom.integer);
}
static void test_sub_5_4_3_is_minus_2(void)
{
am.expr = expr_str_symbol(&store, "-");
env_fetch(&am);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_PRIM_PROC, am.val->atom.type);
am.argl = expr_pair(
&store, expr_integer(&store, 5),
expr_pair(
&store, expr_integer(&store, 4),
expr_pair(
&store, expr_integer(&store, 3), expr_empty_list(&store))));
am.val->atom.prim_proc(&am, &store);
TEST_ASSERT_NOT_NULL(am.val);
TEST_ASSERT_TRUE(am.val->is_atom);
TEST_ASSERT_EQUAL(ATOM_TYPE_INTEGER, am.val->atom.type);
TEST_ASSERT_EQUAL(-2, am.val->atom.integer);
}
int main(void)
{
UNITY_BEGIN();
RUN_TEST(test_add_empty_list_is_0);
RUN_TEST(test_add_1_2_3_is_6);
RUN_TEST(test_mul_empty_list_is_1);
RUN_TEST(test_mul_2_3_4_is_24);
RUN_TEST(test_sub_1_is_minus_1);
RUN_TEST(test_sub_5_4_3_is_minus_2);
return UNITY_END();
}