52 lines
2.2 KiB
C++
52 lines
2.2 KiB
C++
|
// Step 871: Compute/runtime cost estimator from prior runs (10 tests)
|
||
|
|
#include "graduation/ComputeCostEstimator.h"
|
||
|
|
#include <iostream>
|
||
|
|
static int p=0,f=0;
|
||
|
|
#define T(n) { std::cout << " " << #n << "... "; }
|
||
|
|
#define P() { std::cout << "PASS\n"; ++p; }
|
||
|
|
#define F(m) { std::cout << "FAIL: " << m << "\n"; ++f; }
|
||
|
|
#define C(c,m) if(!(c)){F(m);return;}
|
||
|
|
|
||
|
|
void t1(){T(no_history_no_data);
|
||
|
|
auto e=ComputeCostEstimator::estimate("py->cpp",{});
|
||
|
|
C(!e.hasHistory,"no history");P();}
|
||
|
|
void t2(){T(with_history_has_data);
|
||
|
|
auto e=ComputeCostEstimator::estimate("py->cpp",{{"py->cpp",100,500}});
|
||
|
|
C(e.hasHistory,"history");P();}
|
||
|
|
void t3(){T(pair_id_set);
|
||
|
|
auto e=ComputeCostEstimator::estimate("rust->go",{});
|
||
|
|
C(e.pairId=="rust->go","pairId");P();}
|
||
|
|
void t4(){T(sample_count_correct);
|
||
|
|
std::vector<PriorRunRecord> h={{"py->cpp",100,500},{"py->cpp",200,600}};
|
||
|
|
auto e=ComputeCostEstimator::estimate("py->cpp",h);
|
||
|
|
C(e.sampleCount==2,"count");P();}
|
||
|
|
void t5(){T(average_duration);
|
||
|
|
std::vector<PriorRunRecord> h={{"py->cpp",100,500},{"py->cpp",200,600}};
|
||
|
|
auto e=ComputeCostEstimator::estimate("py->cpp",h);
|
||
|
|
C(e.estimatedDurationMs==150.0f,"avg");P();}
|
||
|
|
void t6(){T(average_tokens);
|
||
|
|
std::vector<PriorRunRecord> h={{"py->cpp",100,500},{"py->cpp",200,600}};
|
||
|
|
auto e=ComputeCostEstimator::estimate("py->cpp",h);
|
||
|
|
C(e.estimatedTokens==550.0f,"tokens");P();}
|
||
|
|
void t7(){T(single_run);
|
||
|
|
auto e=ComputeCostEstimator::estimate("py->cpp",{{"py->cpp",300,900}});
|
||
|
|
C(e.estimatedDurationMs==300.0f,"single");P();}
|
||
|
|
void t8(){T(zero_history_zero_estimate);
|
||
|
|
auto e=ComputeCostEstimator::estimate("py->cpp",{});
|
||
|
|
C(e.estimatedDurationMs==0.0f,"zero");P();}
|
||
|
|
void t9(){T(to_json_has_history);
|
||
|
|
auto e=ComputeCostEstimator::estimate("py->cpp",{{"py->cpp",100,500}});
|
||
|
|
auto j=ComputeCostEstimator::toJson(e);
|
||
|
|
C(j.contains("has_history"),"json");P();}
|
||
|
|
void t10(){T(to_json_has_sample_count);
|
||
|
|
auto e=ComputeCostEstimator::estimate("py->cpp",{{"py->cpp",100,500}});
|
||
|
|
auto j=ComputeCostEstimator::toJson(e);
|
||
|
|
C(j.contains("sample_count"),"count");P();}
|
||
|
|
|
||
|
|
int main(){
|
||
|
|
std::cout<<"Step 871: Compute cost estimator\n";
|
||
|
|
t1();t2();t3();t4();t5();t6();t7();t8();t9();t10();
|
||
|
|
std::cout<<"\nResults: "<<p<<"/"<<(p+f)<<" passed\n";
|
||
|
|
return f?1:0;
|
||
|
|
}
|