57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
|
// Step 105 TDD Test: Optimization controls
|
||
|
|
//
|
||
|
|
// Tests:
|
||
|
|
// 1. StrategyAwareOptimizer blocks optimization when @Owner(Single) present
|
||
|
|
// 2. TransformEngine constant folding modifies nodes
|
||
|
|
|
||
|
|
#include <cassert>
|
||
|
|
#include <iostream>
|
||
|
|
#include "StrategyAwareOptimizer.h"
|
||
|
|
#include "TransformEngine.h"
|
||
|
|
#include "ast/Module.h"
|
||
|
|
#include "ast/Function.h"
|
||
|
|
#include "ast/Statement.h"
|
||
|
|
#include "ast/Expression.h"
|
||
|
|
#include "ast/Annotation.h"
|
||
|
|
|
||
|
|
int main() {
|
||
|
|
int passed = 0;
|
||
|
|
int failed = 0;
|
||
|
|
|
||
|
|
Module mod("m1", "mod", "python");
|
||
|
|
auto* fn = new Function("f1", "foo");
|
||
|
|
auto* owner = new OwnerAnnotation("a1", "Single");
|
||
|
|
fn->addChild("annotations", owner);
|
||
|
|
mod.addChild("functions", fn);
|
||
|
|
|
||
|
|
StrategyAwareOptimizer opt;
|
||
|
|
opt.setRoot(&mod);
|
||
|
|
auto res = opt.optimizeFunction("f1");
|
||
|
|
assert(!res.blocked.empty());
|
||
|
|
std::cout << "Test 1 PASS: optimization blocked by annotation" << std::endl;
|
||
|
|
++passed;
|
||
|
|
|
||
|
|
Module mod2("m2", "mod2", "python");
|
||
|
|
auto* fn2 = new Function("f2", "bar");
|
||
|
|
auto* ret = new Return();
|
||
|
|
ret->id = "r1";
|
||
|
|
auto* bin = new BinaryOperation("b1", "+");
|
||
|
|
auto* left = new IntegerLiteral("i1", 2);
|
||
|
|
auto* right = new IntegerLiteral("i2", 3);
|
||
|
|
bin->setChild("left", left);
|
||
|
|
bin->setChild("right", right);
|
||
|
|
ret->setChild("value", bin);
|
||
|
|
fn2->addChild("body", ret);
|
||
|
|
mod2.addChild("functions", fn2);
|
||
|
|
|
||
|
|
TransformEngine engine;
|
||
|
|
engine.setRoot(&mod2);
|
||
|
|
auto foldRes = engine.constantFolding();
|
||
|
|
assert(foldRes.applied && foldRes.nodesModified > 0);
|
||
|
|
std::cout << "Test 2 PASS: constant folding applied" << std::endl;
|
||
|
|
++passed;
|
||
|
|
|
||
|
|
std::cout << "\n=== Step 105 Results: " << passed << " passed, " << failed << " failed ===" << std::endl;
|
||
|
|
return failed > 0 ? 1 : 0;
|
||
|
|
}
|