Files
Enzo/tests/OperatorTests.cpp
2025-06-29 20:49:30 +01:00

67 lines
2.0 KiB
C++

#include <catch2/catch_test_macros.hpp>
#include <memory>
#include "Engine/Operator/Attribute.h"
#include "Engine/Operator/AttributeHandle.h"
#include "Engine/Types.h"
#include "Engine/Operator/Geometry.h"
// #include "Engine/Operator/Primitive.h"
TEST_CASE("attrHandleInt")
{
using namespace enzo;
auto myAttrib = std::make_shared<ga::Attribute>("test", ga::AttrType::intT);
ga::AttributeHandleInt myHandle(myAttrib);
myHandle.addValue(5);
myHandle.addValue(6);
REQUIRE(myHandle.getValue(0)==5);
REQUIRE(myHandle.getValue(1)==6);
}
TEST_CASE("geometry")
{
using namespace enzo;
geo::Geometry geo;
// check add function
ga::AttributeHandle<int> myHandle = geo.addIntAttribute(ga::AttrOwner::POINT, "index");
myHandle.addValue(5);
myHandle.addValue(6);
REQUIRE(myHandle.getValue(0)==5);
REQUIRE(myHandle.getValue(1)==6);
// check getter
std::shared_ptr<ga::Attribute> myAttribute = geo.getAttribByName(ga::AttrOwner::POINT, "index");
ga::AttributeHandle<int> myHandle2(myAttribute);
REQUIRE(myHandle2.getValue(0)==5);
REQUIRE(myHandle2.getValue(1)==6);
// check failed lookup
std::shared_ptr<ga::Attribute> myAttribute2 = geo.getAttribByName(ga::AttrOwner::POINT, "nonExistant");
REQUIRE(myAttribute2==nullptr);
}
TEST_CASE("attrHandleFloat")
{
using namespace enzo;
std::shared_ptr<ga::Attribute> myAttrib = std::make_shared<ga::Attribute>("test", ga::AttrType::floatT);
ga::AttributeHandleFloat myHandle(myAttrib);
myHandle.addValue(5.3f);
myHandle.addValue(6.9f);
REQUIRE(myHandle.getValue(0)==5.3f);
REQUIRE(myHandle.getValue(1)==6.9f);
}
TEST_CASE("Attribute Type")
{
using namespace enzo;
REQUIRE(ga::AttributeType::intT == ga::AttributeType::intT);
REQUIRE(ga::AttributeType::intT != ga::AttributeType::floatT);
REQUIRE(ga::AttributeType::floatT == ga::AttributeType::floatT);
REQUIRE(ga::AttributeType::listT == ga::AttributeType::listT);
REQUIRE(ga::AttributeType::vectorT == ga::AttributeType::vectorT);
}