add test

Author: Menooker

tests/capi/test_c.cpp

@@ -80,13 +80,19 @@ static int testStream(const char *libpath) {
     KunStreamContextHandle ctx = kunCreateStream(exec, modu, num_stocks);
     CHECK(ctx);
 
+    // now dump the stream states
+    // use null buffer and zero size to get the real buffer size
     size_t buf_size = 0;
     auto status = kunStreamSerializeStates(ctx, KUN_INIT_MEMORY, nullptr, &buf_size);
     CHECK(status == KUN_INIT_ERROR);
+    // second try to allocate buffer and get the real data
+    size_t buf_size2 = buf_size;
     auto states_buffer = new char[buf_size];
     status = kunStreamSerializeStates(ctx, KUN_INIT_MEMORY, states_buffer, &buf_size);
     CHECK(status == KUN_SUCCESS);
+    CHECK(buf_size == buf_size2);
 
+    // run the stream
     size_t handleClose = kunQueryBufferHandle(ctx, "close");
     size_t handleOpen = kunQueryBufferHandle(ctx, "open");
     size_t handleHigh = kunQueryBufferHandle(ctx, "high");
@@ -96,27 +102,32 @@ static int testStream(const char *libpath) {
     size_t handleAlpha101 = kunQueryBufferHandle(ctx, "alpha101");
     // don't need to query the handles everytime when calling kunStreamPushData
 
-    kunStreamPushData(ctx, handleClose, dataclose);
-    kunStreamPushData(ctx, handleOpen, dataopen);
-    kunStreamPushData(ctx, handleHigh, datahigh);
-    kunStreamPushData(ctx, handleLow, datalow);
-    kunStreamPushData(ctx, handleVol, datavol);
-    kunStreamPushData(ctx, handleAmount, dataamount);
-
-    kunStreamRun(ctx);
-    memcpy(alpha101, kunStreamGetCurrentBuffer(ctx, handleAlpha101),
-           sizeof(float) * num_stocks);
-
-    for (size_t i = 0; i < num_stocks; i++) {
-        float expected =
-            (dataclose[i] - dataopen[i]) / (datahigh[i] - datalow[i] + 0.001);
-        if (std::abs(alpha101[i] - expected) > 1e-5) {
-            printf("Output error at %zu => %f, %f\n", i, alpha101[i], expected);
-            return 4;
+    auto run_and_check = [&]() {
+        kunStreamPushData(ctx, handleClose, dataclose);
+        kunStreamPushData(ctx, handleOpen, dataopen);
+        kunStreamPushData(ctx, handleHigh, datahigh);
+        kunStreamPushData(ctx, handleLow, datalow);
+        kunStreamPushData(ctx, handleVol, datavol);
+        kunStreamPushData(ctx, handleAmount, dataamount);
+
+        kunStreamRun(ctx);
+        memcpy(alpha101, kunStreamGetCurrentBuffer(ctx, handleAlpha101),
+            sizeof(float) * num_stocks);
+
+        for (size_t i = 0; i < num_stocks; i++) {
+            float expected =
+                (dataclose[i] - dataopen[i]) / (datahigh[i] - datalow[i] + 0.001);
+            if (std::abs(alpha101[i] - expected) > 1e-5) {
+                printf("Output error at %zu => %f, %f\n", i, alpha101[i], expected);
+                exit(4);
+            }
         }
-    }
-    
+    };
+    run_and_check();    
     kunDestoryStream(ctx);
+
+    // check restore stream from states
+    // create a new stream context from the states
     ctx = nullptr;
     KunStreamExtraArgs extra_args;
     extra_args.version = KUN_API_VERSION;
@@ -126,6 +137,8 @@ static int testStream(const char *libpath) {
     status = kunCreateStreamEx(exec, modu, num_stocks, &extra_args, &ctx);
     CHECK(status == KUN_SUCCESS);
     CHECK(ctx);
+    // run again to check if the states are restored correctly
+    run_and_check();
 
 
     delete[] dataclose;