mirror of https://github.com/exaloop/codon
108 lines
3.4 KiB
C++
108 lines
3.4 KiB
C++
#include "test.h"
|
|
|
|
#include <algorithm>
|
|
|
|
#include "codon/sir/util/matching.h"
|
|
|
|
using namespace codon::ir;
|
|
|
|
TEST_F(SIRCoreTest, FuncRealizationAndVarInsertionEraseAndIterators) {
|
|
auto *fn = module->Nr<BodiedFunc>();
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
|
|
auto *fnType = module->unsafeGetFuncType("**test_type**", module->getIntType(),
|
|
{module->getIntType()});
|
|
std::vector<std::string> names = {"foo"};
|
|
fn->realize(cast<types::FuncType>(fnType), names);
|
|
ASSERT_TRUE(fn->isGlobal());
|
|
|
|
ASSERT_EQ(1, std::distance(fn->arg_begin(), fn->arg_end()));
|
|
ASSERT_EQ(module->getIntType(), fn->arg_front()->getType());
|
|
|
|
auto *var = module->Nr<Var>(module->getIntType(), false, "hi");
|
|
fn->push_back(var);
|
|
ASSERT_EQ(1, std::distance(fn->begin(), fn->end()));
|
|
fn->erase(fn->begin());
|
|
ASSERT_EQ(0, std::distance(fn->begin(), fn->end()));
|
|
fn->insert(fn->begin(), var);
|
|
ASSERT_EQ(1, std::distance(fn->begin(), fn->end()));
|
|
ASSERT_EQ(module->getIntType(), fn->front()->getType());
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, BodiedFuncQueryAndReplace) {
|
|
auto *fn = module->Nr<BodiedFunc>();
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
fn->setBuiltin();
|
|
ASSERT_TRUE(fn->isBuiltin());
|
|
|
|
auto *body = fn->getBody();
|
|
ASSERT_FALSE(body);
|
|
ASSERT_EQ(0, fn->getUsedValues().size());
|
|
|
|
body = module->Nr<SeriesFlow>();
|
|
fn->setBody(body);
|
|
ASSERT_EQ(body, fn->getBody());
|
|
|
|
auto used = fn->getUsedValues();
|
|
ASSERT_EQ(1, used.size());
|
|
ASSERT_EQ(body, used[0]);
|
|
|
|
ASSERT_EQ(1, fn->replaceUsedValue(body, module->Nr<SeriesFlow>()));
|
|
ASSERT_DEATH(fn->replaceUsedValue(fn->getBody(), module->Nr<VarValue>(nullptr)), "");
|
|
ASSERT_NE(fn->getBody(), body);
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, BodiedFuncUnmangledName) {
|
|
auto *fn = module->Nr<BodiedFunc>("Int.foo");
|
|
fn->setUnmangledName("foo");
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
ASSERT_EQ("foo", fn->getUnmangledName());
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, BodiedFuncCloning) {
|
|
auto *fn = module->Nr<BodiedFunc>("fn");
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
|
|
fn->setBuiltin();
|
|
fn->setBody(module->Nr<SeriesFlow>());
|
|
ASSERT_TRUE(util::match(fn, cv->clone(fn)));
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, ExternalFuncUnmangledNameAndCloning) {
|
|
auto *fn = module->Nr<ExternalFunc>("fn");
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
|
|
fn->setUnmangledName("foo");
|
|
ASSERT_EQ("foo", fn->getUnmangledName());
|
|
ASSERT_TRUE(util::match(fn, cv->clone(fn)));
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, InternalFuncParentTypeUnmangledNameAndCloning) {
|
|
auto *fn = module->Nr<InternalFunc>("fn.1");
|
|
fn->setUnmangledName("fn");
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
|
|
fn->setParentType(module->getIntType());
|
|
ASSERT_EQ("fn", fn->getUnmangledName());
|
|
ASSERT_EQ(fn->getParentType(), module->getIntType());
|
|
ASSERT_TRUE(util::match(fn, cv->clone(fn)));
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, LLVMFuncUnmangledNameQueryAndReplace) {
|
|
auto *fn = module->Nr<LLVMFunc>("fn");
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
|
|
fn->setLLVMBody("body");
|
|
fn->setLLVMDeclarations("decl");
|
|
|
|
std::vector<types::Generic> literals = {types::Generic(1),
|
|
types::Generic(module->getIntType())};
|
|
fn->setLLVMLiterals(literals);
|
|
|
|
ASSERT_EQ("body", fn->getLLVMBody());
|
|
ASSERT_EQ("decl", fn->getLLVMDeclarations());
|
|
|
|
ASSERT_EQ(1, fn->replaceUsedType(module->getIntType(), module->getFloatType()));
|
|
ASSERT_EQ(module->getFloatType(), fn->literal_back().getTypeValue());
|
|
}
|