VM-test.cpp 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732
  1. #include "test_common.h"
  2. extern "C" {
  3. extern char log_buff[LOG_BUFF_MAX][LOG_SIZE];
  4. /* test head */
  5. VMParameters* pikaVM_runAsmWithPars(PikaObj* self,
  6. VMParameters* locals,
  7. VMParameters* globals,
  8. char* pikaAsm);
  9. char* Parser_LineToAsm(Args* buffs, char* line, Stack* blockStack);
  10. char* Parser_linesToAsm(Args* outBuffs, char* multiLine);
  11. int32_t __clearInvokeStackes(VMParameters* locals);
  12. int32_t pikaVM_runAsmLine(PikaObj* self,
  13. VMParameters* locals,
  14. VMParameters* globals,
  15. char* pikaAsm,
  16. int32_t lineAddr);
  17. }
  18. TEST(VM, num1) {
  19. char* line = "1";
  20. Args* buffs = New_strBuff();
  21. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  22. printf("%s", pikaAsm);
  23. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  24. __platform_printf("BEGIN\r\n");
  25. pikaVM_runAsm(self, pikaAsm);
  26. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  27. EXPECT_STREQ(log_buff[0], "1\r\n");
  28. obj_deinit(self);
  29. args_deinit(buffs);
  30. EXPECT_EQ(pikaMemNow(), 0);
  31. }
  32. TEST(VM, a_1) {
  33. char* line = "a = 1";
  34. Args* buffs = New_strBuff();
  35. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  36. printf("%s", pikaAsm);
  37. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  38. pikaVM_runAsm(self, pikaAsm);
  39. ASSERT_EQ(args_getInt(self->list, "a"), 1);
  40. obj_deinit(self);
  41. args_deinit(buffs);
  42. EXPECT_EQ(pikaMemNow(), 0);
  43. }
  44. TEST(VM, a_1d1) {
  45. char* line = "a = 1.1";
  46. Args* buffs = New_strBuff();
  47. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  48. printf("%s", pikaAsm);
  49. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  50. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  51. ASSERT_FLOAT_EQ(args_getFloat(globals->list, "a"), 1.1);
  52. obj_deinit(self);
  53. // obj_deinit(globals);
  54. args_deinit(buffs);
  55. EXPECT_EQ(pikaMemNow(), 0);
  56. }
  57. TEST(VM, str_xy) {
  58. char* line = "a = 'xy'";
  59. Args* buffs = New_strBuff();
  60. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  61. printf("%s", pikaAsm);
  62. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  63. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  64. ASSERT_STREQ(args_getStr(globals->list, "a"), "xy");
  65. obj_deinit(self);
  66. // obj_deinit(globals);
  67. args_deinit(buffs);
  68. EXPECT_EQ(pikaMemNow(), 0);
  69. }
  70. TEST(VM, str_xy_space) {
  71. char* line = "a = 'xy '";
  72. Args* buffs = New_strBuff();
  73. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  74. printf("%s", pikaAsm);
  75. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  76. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  77. ASSERT_STREQ(args_getStr(globals->list, "a"), "xy ");
  78. obj_deinit(self);
  79. args_deinit(buffs);
  80. // obj_deinit(globals);
  81. EXPECT_EQ(pikaMemNow(), 0);
  82. }
  83. TEST(VM, ref_a_b) {
  84. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  85. Args* buffs = New_strBuff();
  86. pikaVM_runAsm(self, Parser_LineToAsm(buffs, "a = 'xy '", NULL));
  87. args_deinit(buffs);
  88. obj_deinit(self);
  89. // obj_deinit(globals);
  90. EXPECT_EQ(pikaMemNow(), 0);
  91. }
  92. TEST(VM, Run_add) {
  93. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  94. Args* buffs = New_strBuff();
  95. char* pikaAsm = Parser_LineToAsm(buffs, "a = plusInt(1,2)", NULL);
  96. __platform_printf("%s", pikaAsm);
  97. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  98. args_deinit(buffs);
  99. int a = args_getInt(globals->list, "a");
  100. ASSERT_EQ(a, 3);
  101. obj_deinit(self);
  102. // obj_deinit(globals);
  103. EXPECT_EQ(pikaMemNow(), 0);
  104. }
  105. TEST(VM, Run_add_multy) {
  106. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  107. Args* buffs = New_strBuff();
  108. pikaVM_runAsm(self, Parser_LineToAsm(buffs, "b = 2", NULL));
  109. args_deinit(buffs);
  110. obj_deinit(self);
  111. // obj_deinit(globals);
  112. EXPECT_EQ(pikaMemNow(), 0);
  113. }
  114. TEST(VM, Run_add_1_2_3) {
  115. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  116. Args* buffs = New_strBuff();
  117. char* pikaAsm =
  118. Parser_LineToAsm(buffs, "a = plusInt(1, plusInt(2,3) )", NULL);
  119. __platform_printf("%s", pikaAsm);
  120. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  121. args_deinit(buffs);
  122. int a = args_getInt(globals->list, "a");
  123. ASSERT_EQ(a, 6);
  124. obj_deinit(self);
  125. // obj_deinit(globals);
  126. EXPECT_EQ(pikaMemNow(), 0);
  127. }
  128. extern PikaMemInfo pikaMemInfo;
  129. TEST(VM, WHILE) {
  130. pikaMemInfo.heapUsedMax = 0;
  131. Args* buffs = New_strBuff();
  132. char* lines =(char *)
  133. "a = 1\n"
  134. "b = 0\n"
  135. "while a:\n"
  136. " b = 1\n"
  137. " a = 0\n"
  138. "\n";
  139. printf("%s", lines);
  140. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  141. printf("%s", pikaAsm);
  142. pikaMemInfo.heapUsedMax = 0;
  143. PikaObj* self = New_TinyObj(NULL);
  144. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  145. EXPECT_EQ(args_getInt(globals->list, "a"), 0);
  146. EXPECT_EQ(args_getInt(globals->list, "b"), 1);
  147. // obj_deinit(globals);
  148. args_deinit(buffs);
  149. obj_deinit(self);
  150. EXPECT_EQ(pikaMemNow(), 0);
  151. }
  152. TEST(VM, a_1_1) {
  153. char* line = "a = 1 + 1";
  154. Args* buffs = New_strBuff();
  155. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  156. printf("%s", pikaAsm);
  157. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  158. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  159. int res = args_getInt(globals->list, "a");
  160. obj_deinit(self);
  161. args_deinit(buffs);
  162. ASSERT_EQ(res, 2);
  163. EXPECT_EQ(pikaMemNow(), 0);
  164. }
  165. TEST(VM, a_1_1d1) {
  166. char* line = "a = 1 + 1.1";
  167. Args* buffs = New_strBuff();
  168. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  169. printf("%s", pikaAsm);
  170. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  171. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  172. float res = args_getFloat(globals->list, "a");
  173. obj_deinit(self);
  174. // obj_deinit(globals);
  175. args_deinit(buffs);
  176. ASSERT_FLOAT_EQ(res, 2.1);
  177. EXPECT_EQ(pikaMemNow(), 0);
  178. }
  179. TEST(VM, a_jjcc) {
  180. char* line = "a = (1 + 1.1) * 3 - 2 /4.0";
  181. printf("%s\r\n", line);
  182. Args* buffs = New_strBuff();
  183. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  184. printf("%s", pikaAsm);
  185. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  186. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  187. float res = args_getFloat(globals->list, "a");
  188. obj_deinit(self);
  189. args_deinit(buffs);
  190. // obj_deinit(globals);
  191. ASSERT_FLOAT_EQ(res, 5.8);
  192. EXPECT_EQ(pikaMemNow(), 0);
  193. }
  194. TEST(VM, while_a_1to10) {
  195. char* line =
  196. "a = 0\n"
  197. "while a < 10:\n"
  198. " a += 1\n"
  199. " print(a)\n"
  200. "\n";
  201. Args* buffs = New_strBuff();
  202. char* pikaAsm = Parser_linesToAsm(buffs, line);
  203. printf("%s", pikaAsm);
  204. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  205. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  206. int res = args_getInt(globals->list, "a");
  207. obj_deinit(self);
  208. args_deinit(buffs);
  209. // obj_deinit(globals);
  210. ASSERT_FLOAT_EQ(res, 10);
  211. EXPECT_EQ(pikaMemNow(), 0);
  212. }
  213. TEST(VM, mem_x) {
  214. char* line =
  215. "mem = PikaStdLib.MemChecker()\n"
  216. "mem.x = 1\n"
  217. "print(mem.x)\n"
  218. "mem.x = 2\n"
  219. "print(mem.x)\n"
  220. "\n";
  221. Args* buffs = New_strBuff();
  222. char* pikaAsm = Parser_linesToAsm(buffs, line);
  223. printf("%s", pikaAsm);
  224. PikaObj* self = newRootObj("", New_PikaMain);
  225. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  226. int res = obj_getInt(globals, "mem.x");
  227. EXPECT_EQ(2, res);
  228. obj_deinit(self);
  229. // obj_deinit(globals);
  230. args_deinit(buffs);
  231. EXPECT_EQ(pikaMemNow(), 0);
  232. }
  233. TEST(VM, RET_instruct) {
  234. char* pikaAsm =
  235. "B1\n"
  236. "0 NUM 13\n"
  237. "0 RET \n"
  238. "0 NUM 2\n"
  239. "0 RET \n"
  240. "B0\n";
  241. PikaObj* self = New_TinyObj(NULL);
  242. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  243. Arg* return_arg = args_getArg(globals->list, "return");
  244. int num = arg_getInt(return_arg);
  245. EXPECT_EQ(num, 13);
  246. obj_deinit(self);
  247. // obj_deinit(globals);
  248. EXPECT_EQ(pikaMemNow(), 0);
  249. }
  250. TEST(VM, RUN_DEF) {
  251. char* pikaAsm =
  252. "B0\n"
  253. "0 DEF test()\n"
  254. "0 JMP 1\n"
  255. "B1\n"
  256. "0 NUM 1\n"
  257. "0 RET \n"
  258. "B0\n"
  259. "0 RUN test\n"
  260. "0 OUT a\n";
  261. PikaObj* self = New_TinyObj(NULL);
  262. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  263. int num = obj_getInt(globals, "a");
  264. EXPECT_EQ(num, 1);
  265. obj_deinit(self);
  266. // obj_deinit(globals);
  267. EXPECT_EQ(pikaMemNow(), 0);
  268. }
  269. TEST(VM, RUN_global) {
  270. char* pikaAsm =
  271. "B0\n"
  272. "0 NUM 1\n"
  273. "0 OUT a\n"
  274. "B0\n"
  275. "0 DEF test()\n"
  276. "0 JMP 1\n"
  277. "B1\n"
  278. "0 REF a\n"
  279. "0 RET \n"
  280. "B0\n"
  281. "0 RUN test\n"
  282. "0 OUT b\n";
  283. PikaObj* self = New_TinyObj(NULL);
  284. pikaVM_runAsm(self, pikaAsm);
  285. int a = obj_getInt(self, "a");
  286. int b = obj_getInt(self, "b");
  287. EXPECT_EQ(a, 1);
  288. EXPECT_EQ(b, 1);
  289. obj_deinit(self);
  290. // obj_deinit(globals);
  291. EXPECT_EQ(pikaMemNow(), 0);
  292. }
  293. TEST(VM, RUN_local_b) {
  294. char* pikaAsm =
  295. "B0\n"
  296. "0 NUM 1\n"
  297. "0 OUT a\n"
  298. "B0\n"
  299. "0 DEF test()\n"
  300. "0 JMP 1\n"
  301. "B1\n"
  302. "0 REF a\n"
  303. "0 OUT b\n"
  304. "1 REF b\n"
  305. "1 REF a\n"
  306. "0 OPT +\n"
  307. "0 RET \n"
  308. "B0\n"
  309. "0 RUN test\n"
  310. "0 OUT c\n";
  311. PikaObj* self = newRootObj("", New_BaseObj);
  312. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  313. int a = obj_getInt(globals, "a");
  314. int b = obj_getInt(globals, "b");
  315. int c = obj_getInt(globals, "c");
  316. EXPECT_EQ(a, 1);
  317. /* b is local, should not be exist in globals */
  318. EXPECT_EQ(b, -999999999);
  319. EXPECT_EQ(c, 2);
  320. obj_deinit(self);
  321. // obj_deinit(globals);
  322. EXPECT_EQ(pikaMemNow(), 0);
  323. }
  324. TEST(VM, RUN_DEF_add) {
  325. char* pikaAsm =
  326. "B0\n"
  327. "0 DEF add(a,b)\n"
  328. "0 JMP 1\n"
  329. "B1\n"
  330. "1 REF b\n"
  331. "1 REF a\n"
  332. "0 OPT +\n"
  333. "0 RET \n"
  334. "B0\n"
  335. "1 NUM 1\n"
  336. "1 NUM 2\n"
  337. "0 RUN add\n"
  338. "0 OUT c\n";
  339. PikaObj* self = newRootObj("", New_BaseObj);
  340. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  341. int a = obj_getInt(globals, "a");
  342. int b = obj_getInt(globals, "b");
  343. int c = obj_getInt(globals, "c");
  344. /* a is local, should not be exist in globals */
  345. EXPECT_EQ(a, -999999999);
  346. /* b is local, should not be exist in globals */
  347. EXPECT_EQ(b, -999999999);
  348. EXPECT_EQ(c, 3);
  349. obj_deinit(self);
  350. // obj_deinit(globals);
  351. EXPECT_EQ(pikaMemNow(), 0);
  352. }
  353. TEST(VM, run_def_add) {
  354. char* line =
  355. "def add(a, b):\n"
  356. " return a + b\n"
  357. "c = add(1,3)\n"
  358. "\n";
  359. Args* buffs = New_strBuff();
  360. char* pikaAsm = Parser_linesToAsm(buffs, line);
  361. printf("%s", pikaAsm);
  362. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  363. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  364. int c = args_getInt(globals->list, "c");
  365. obj_deinit(self);
  366. args_deinit(buffs);
  367. // obj_deinit(globals);
  368. ASSERT_FLOAT_EQ(c, 4);
  369. EXPECT_EQ(pikaMemNow(), 0);
  370. }
  371. TEST(VM, equ) {
  372. char* line =
  373. "if -1 == -1:\n"
  374. " a = 1\n"
  375. "\n";
  376. Args* buffs = New_strBuff();
  377. char* pikaAsm = Parser_linesToAsm(buffs, line);
  378. printf("%s", pikaAsm);
  379. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  380. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  381. int c = args_getInt(globals->list, "a");
  382. obj_deinit(self);
  383. args_deinit(buffs);
  384. // obj_deinit(globals);
  385. ASSERT_FLOAT_EQ(c, 1);
  386. EXPECT_EQ(pikaMemNow(), 0);
  387. }
  388. TEST(VM, if_elif) {
  389. char* line =
  390. "a = 2\n"
  391. "b = 0\n"
  392. "if a > 1:\n"
  393. " b = 1\n"
  394. "elif a > 0:\n"
  395. " b = 2\n"
  396. "\n";
  397. Args* buffs = New_strBuff();
  398. char* pikaAsm = Parser_linesToAsm(buffs, line);
  399. printf("%s", pikaAsm);
  400. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  401. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  402. int b = args_getInt(globals->list, "b");
  403. obj_deinit(self);
  404. args_deinit(buffs);
  405. ASSERT_FLOAT_EQ(b, 1);
  406. EXPECT_EQ(pikaMemNow(), 0);
  407. }
  408. TEST(VM, if_else) {
  409. char* line =
  410. "a = 0\n"
  411. "b = 0\n"
  412. "if a > 1:\n"
  413. " b = 1\n"
  414. "else:\n"
  415. " b = 2\n"
  416. "\n";
  417. Args* buffs = New_strBuff();
  418. char* pikaAsm = Parser_linesToAsm(buffs, line);
  419. printf("%s", pikaAsm);
  420. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  421. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  422. int b = args_getInt(globals->list, "b");
  423. obj_deinit(self);
  424. args_deinit(buffs);
  425. // obj_deinit(globals);
  426. ASSERT_FLOAT_EQ(b, 2);
  427. EXPECT_EQ(pikaMemNow(), 0);
  428. }
  429. TEST(VM, if_else_) {
  430. char* line =
  431. "a = 2\n"
  432. "b = 0\n"
  433. "if a > 1:\n"
  434. " b = 1\n"
  435. "else:\n"
  436. " b = 2\n"
  437. "\n";
  438. Args* buffs = New_strBuff();
  439. char* pikaAsm = Parser_linesToAsm(buffs, line);
  440. printf("%s", pikaAsm);
  441. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  442. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  443. int b = args_getInt(globals->list, "b");
  444. obj_deinit(self);
  445. args_deinit(buffs);
  446. // obj_deinit(globals);
  447. ASSERT_FLOAT_EQ(b, 1);
  448. EXPECT_EQ(pikaMemNow(), 0);
  449. }
  450. TEST(VM, EST) {
  451. char* pikaAsm =
  452. "B0\n"
  453. "0 EST a\n"
  454. "0 OUT b\n";
  455. PikaObj* self = newRootObj("", New_BaseObj);
  456. pikaVM_runAsm(self, pikaAsm);
  457. int a = obj_getInt(self, "a");
  458. int b = obj_getInt(self, "b");
  459. /* a is local, should not be exist in globals */
  460. EXPECT_EQ(a, -999999999);
  461. /* b is local, should not be exist in globals */
  462. EXPECT_EQ(b, 0);
  463. obj_deinit(self);
  464. // obj_deinit(globals);
  465. EXPECT_EQ(pikaMemNow(), 0);
  466. }
  467. TEST(VM, RUN_AS) {
  468. char* pikaAsm =
  469. "B0\n"
  470. "0 RUN TinyObj\n"
  471. "0 OUT as\n"
  472. "B0\n"
  473. "0 RAS as\n"
  474. "B0\n"
  475. "0 NUM 1\n"
  476. "0 OUT x\n"
  477. "B0\n"
  478. "0 RAS $origin\n"
  479. "B0\n"
  480. "0 NUM 2\n"
  481. "0 OUT x\n";
  482. PikaObj* self = newRootObj("", New_PikaMain);
  483. pikaVM_runAsm(self, pikaAsm);
  484. PikaObj* as = obj_getObj(self, "as");
  485. int x_as_ = obj_getInt(as, "x");
  486. int x_as = obj_getInt(self, "as.x");
  487. int x_origin = obj_getInt(self, "x");
  488. EXPECT_EQ(x_as_, 1);
  489. EXPECT_EQ(x_as, 1);
  490. EXPECT_EQ(x_origin, 2);
  491. obj_deinit(self);
  492. // obj_deinit(globals);
  493. EXPECT_EQ(pikaMemNow(), 0);
  494. }
  495. TEST(VM, RUN_NEW) {
  496. char* pikaAsm =
  497. "B0\n"
  498. "0 RUN TinyObj\n"
  499. "0 OUT newObj\n"
  500. "0 NUM 1\n"
  501. "0 OUT newObj.x\n"
  502. "B0\n"
  503. "0 NEW newObj\n"
  504. "0 OUT outObj\n";
  505. PikaObj* self = newRootObj("", New_PikaMain);
  506. pikaVM_runAsm(self, pikaAsm);
  507. Arg* newObj = obj_getArg(self, "newObj");
  508. Arg* outObj = obj_getArg(self, "outObj");
  509. void* newObj_ptr = arg_getPtr(newObj);
  510. void* outObj_ptr = arg_getPtr(outObj);
  511. EXPECT_EQ(newObj_ptr, outObj_ptr);
  512. int x = obj_getInt(self, "outObj.x");
  513. ArgType newObj_type = arg_getType(newObj);
  514. ArgType outObj_type = arg_getType(outObj);
  515. EXPECT_EQ(x, 1);
  516. EXPECT_EQ(newObj_type, ARG_TYPE_OBJECT);
  517. EXPECT_EQ(outObj_type, ARG_TYPE_OBJECT);
  518. obj_deinit(self);
  519. // obj_deinit(globals);
  520. EXPECT_EQ(pikaMemNow(), 0);
  521. }
  522. TEST(VM, RUN_DEF_NEW) {
  523. char* pikaAsm =
  524. "B0\n"
  525. "0 DEF testnew()\n"
  526. "0 JMP 1\n"
  527. "B1\n"
  528. "0 RUN TinyObj\n"
  529. "0 OUT newObj\n"
  530. "B1\n"
  531. "0 NUM 1\n"
  532. "0 OUT newObj.x\n"
  533. "B1\n"
  534. "0 NEW newObj\n"
  535. "0 RET \n"
  536. "B1\n"
  537. "0 RET \n"
  538. "B0\n"
  539. "0 RUN testnew\n"
  540. "0 OUT outobj\n";
  541. PikaObj* self = newRootObj("", New_PikaMain);
  542. pikaVM_runAsm(self, pikaAsm);
  543. /* assert */
  544. PikaObj* outobj = obj_getObj(self, "outobj");
  545. int x = obj_getInt(outobj, "x");
  546. EXPECT_EQ(x, 1);
  547. /* deinit */
  548. obj_deinit(self);
  549. // obj_deinit(globals);
  550. EXPECT_EQ(pikaMemNow(), 0);
  551. }
  552. TEST(VM, class_x_1) {
  553. char* line =
  554. "class Test(TinyObj):\n"
  555. " x = 1\n"
  556. "test = Test()\n"
  557. "print('test.x: ' + str(test.x))\n"
  558. "\n";
  559. Args* buffs = New_strBuff();
  560. char* pikaAsm = Parser_linesToAsm(buffs, line);
  561. printf("%s", pikaAsm);
  562. PikaObj* self = newRootObj("", New_PikaMain);
  563. pikaVM_runAsm(self, pikaAsm);
  564. PikaObj* test = obj_getObj(self, "test");
  565. Arg* test_arg = obj_getArg(self, "test");
  566. ArgType test_arg_type = arg_getType(test_arg);
  567. EXPECT_EQ(test_arg_type, ARG_TYPE_OBJECT);
  568. int x = obj_getInt(test, "x");
  569. EXPECT_EQ(x, 1);
  570. obj_deinit(self);
  571. args_deinit(buffs);
  572. EXPECT_EQ(pikaMemNow(), 0);
  573. }
  574. TEST(VM, nag_a) {
  575. char* line =
  576. "a = 1\n"
  577. "print(-a)\n"
  578. "b = 0.5\n"
  579. "print(-b)\n";
  580. Args* buffs = New_strBuff();
  581. char* pikaAsm = Parser_linesToAsm(buffs, line);
  582. printf("%s", pikaAsm);
  583. PikaObj* self = newRootObj("", New_PikaMain);
  584. __platform_printf("BEGIN\r\n");
  585. pikaVM_runAsm(self, pikaAsm);
  586. /* assert */
  587. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  588. EXPECT_STREQ(log_buff[1], "-1\r\n");
  589. EXPECT_STREQ(log_buff[0], "-0.500000\r\n");
  590. /* deinit */
  591. obj_deinit(self);
  592. args_deinit(buffs);
  593. /* check mem */
  594. EXPECT_EQ(pikaMemNow(), 0);
  595. }
  596. TEST(InstructUnit, base) {
  597. __platform_printf("BEGIN\r\n");
  598. InstructUnit ins_unit;
  599. instructUnit_init(&ins_unit);
  600. instructUnit_setBlockDeepth(&ins_unit, 2);
  601. instructUnit_setIsNewLine(&ins_unit, 1);
  602. instructUnit_setInvokeDeepth(&ins_unit, 3);
  603. instructUnit_setInstruct(&ins_unit, (Instruct)4);
  604. instructUnit_setConstPoolIndex(&ins_unit, 12);
  605. EXPECT_EQ(instructUnit_getBlockDeepth(&ins_unit), 2);
  606. EXPECT_EQ(instructUnit_getIsNewLine(&ins_unit), 1);
  607. EXPECT_EQ(instructUnit_getInvokeDeepth(&ins_unit), 3);
  608. EXPECT_EQ(instructUnit_getInstruct(&ins_unit), 4);
  609. EXPECT_EQ(instructUnit_getConstPoolIndex(&ins_unit), 12);
  610. instructUnit_print(&ins_unit);
  611. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  612. EXPECT_STREQ(log_buff[1], "B2\r\n");
  613. EXPECT_STREQ(log_buff[0], "3 OUT #12\r\n");
  614. EXPECT_EQ(pikaMemNow(), 0);
  615. }
  616. TEST(ConstPool, get) {
  617. __platform_printf("BEGIN\r\n");
  618. ConstPool cp;
  619. constPool_init(&cp);
  620. char* next_content;
  621. uint16_t next_offset;
  622. next_offset = constPool_getLastOffset(&cp);
  623. constPool_append(&cp, "hello");
  624. next_content = constPool_getByOffset(&cp, next_offset);
  625. EXPECT_STREQ(next_content, "hello");
  626. next_offset = constPool_getLastOffset(&cp);
  627. constPool_append(&cp, "world");
  628. next_content = constPool_getByOffset(&cp, next_offset);
  629. EXPECT_STREQ(next_content, "world");
  630. char* first = constPool_getNow(&cp);
  631. char* second = constPool_getNext(&cp);
  632. char* third = constPool_getNext(&cp);
  633. char* forth = constPool_getNext(&cp);
  634. EXPECT_STREQ(first, "");
  635. EXPECT_STREQ(second, "hello");
  636. EXPECT_STREQ(third, "world");
  637. EXPECT_EQ((uintptr_t)forth, (uintptr_t)NULL);
  638. EXPECT_STREQ(constPool_getByIndex(&cp, 0), "");
  639. EXPECT_STREQ(constPool_getByIndex(&cp, 1), "hello");
  640. EXPECT_STREQ(constPool_getByIndex(&cp, 2), "world");
  641. EXPECT_EQ((uintptr_t)constPool_getByIndex(&cp, 3), (uintptr_t)NULL);
  642. constPool_print(&cp);
  643. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  644. EXPECT_STREQ(log_buff[1], "1: hello\r\n");
  645. EXPECT_STREQ(log_buff[0], "7: world\r\n");
  646. constPool_deinit(&cp);
  647. EXPECT_EQ(pikaMemNow(), 0);
  648. }
  649. TEST(InstructArray, set) {
  650. __platform_printf("BEGIN\r\n");
  651. InstructUnit ins_unit;
  652. instructUnit_init(&ins_unit);
  653. instructUnit_setBlockDeepth(&ins_unit, 2);
  654. instructUnit_setIsNewLine(&ins_unit, 1);
  655. instructUnit_setInvokeDeepth(&ins_unit, 3);
  656. instructUnit_setInstruct(&ins_unit, (Instruct)4);
  657. instructUnit_setConstPoolIndex(&ins_unit, 12);
  658. InstructArray ins_array;
  659. instructArray_init(&ins_array);
  660. instructArray_append(&ins_array, &ins_unit);
  661. instructArray_print(&ins_array);
  662. instructArray_deinit(&ins_array);
  663. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  664. EXPECT_STREQ(log_buff[1], "B2\r\n");
  665. EXPECT_STREQ(log_buff[0], "3 OUT #12\r\n");
  666. EXPECT_EQ(pikaMemNow(), 0);
  667. }
  668. TEST(VM, bytecode_jjcc) {
  669. char* line = "a = (1 + 1.1) * 3 - 2 /4.0";
  670. Args* buffs = New_strBuff();
  671. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  672. printf("%s", pikaAsm);
  673. ByteCodeFrame byte_frame;
  674. byteCodeFrame_init(&byte_frame);
  675. byteCodeFrame_appendFromAsm(&byte_frame, pikaAsm);
  676. byteCodeFrame_print(&byte_frame);
  677. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  678. // pikaVM_runAsm(self, pikaAsm);
  679. pikaVM_runByteCodeFrame(self, &byte_frame);
  680. float res = obj_getFloat(self, "a");
  681. obj_deinit(self);
  682. args_deinit(buffs);
  683. byteCodeFrame_deinit(&byte_frame);
  684. ASSERT_FLOAT_EQ(res, 5.8);
  685. EXPECT_EQ(pikaMemNow(), 0);
  686. }
  687. TEST(VM, WHILE_byte) {
  688. pikaMemInfo.heapUsedMax = 0;
  689. Args* buffs = New_strBuff();
  690. char* lines =(char *)
  691. "a = 1\n"
  692. "b = 0\n"
  693. "while a:\n"
  694. " b = 1\n"
  695. " a = 0\n"
  696. "\n";
  697. printf("%s", lines);
  698. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  699. printf("%s", pikaAsm);
  700. pikaMemInfo.heapUsedMax = 0;
  701. PikaObj* self = New_TinyObj(NULL);
  702. pikaVM_run(self, lines);
  703. EXPECT_EQ(obj_getInt(self, "a"), 0);
  704. EXPECT_EQ(obj_getInt(self, "b"), 1);
  705. args_deinit(buffs);
  706. obj_deinit(self);
  707. EXPECT_EQ(pikaMemNow(), 0);
  708. }
  709. TEST(VM, for_break_byte) {
  710. pikaMemInfo.heapUsedMax = 0;
  711. Args* buffs = New_strBuff();
  712. char* lines =(char *)
  713. "a = 0\n"
  714. "for i in range(0, 10):\n"
  715. " if i == 5:\n"
  716. " break\n"
  717. " a = a + i\n"
  718. "\n";
  719. printf("%s", lines);
  720. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  721. printf("%s", pikaAsm);
  722. pikaMemInfo.heapUsedMax = 0;
  723. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  724. pikaVM_run(self, lines);
  725. /* assert */
  726. int a = obj_getInt(self, "a");
  727. EXPECT_EQ(a, 10);
  728. /* deinit */
  729. args_deinit(buffs);
  730. obj_deinit(self);
  731. EXPECT_EQ(pikaMemNow(), 0);
  732. }
  733. TEST(VM, load_static_bytes) {
  734. const uint8_t bytes[] = {
  735. 0x08, 0x02, /* instruct array size */
  736. 0x00, 0x82, 0x01, 0x00, 0x00, 0x04, 0x0d, 0x00, 0x00, 0x83, 0x11, 0x00,
  737. 0x00, 0x04, 0x16, 0x00, 0x00, 0x85, 0x1b, 0x00, 0x00, 0x04, 0x1d, 0x00,
  738. 0x20, 0x85, 0x1b, 0x00, 0x20, 0x01, 0x0d, 0x00, 0x10, 0x02, 0x28, 0x00,
  739. 0x00, 0x02, 0x2e, 0x00, 0x00, 0x04, 0x33, 0x00, 0x00, 0x01, 0x37, 0x00,
  740. 0x00, 0x01, 0x3b, 0x00, 0x00, 0x01, 0x3f, 0x00, 0x00, 0x04, 0x43, 0x00,
  741. 0x00, 0x04, 0x4a, 0x00, 0x00, 0x04, 0x51, 0x00, 0x00, 0x82, 0x58, 0x00,
  742. 0x00, 0x04, 0x65, 0x00, 0x00, 0x0d, 0x65, 0x00, 0x00, 0x07, 0x67, 0x00,
  743. 0x11, 0x81, 0x0d, 0x00, 0x11, 0x05, 0x1b, 0x00, 0x01, 0x08, 0x69, 0x00,
  744. 0x01, 0x07, 0x6c, 0x00, 0x02, 0x8e, 0x6c, 0x00, 0x11, 0x81, 0x1d, 0x00,
  745. 0x11, 0x05, 0x1b, 0x00, 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00,
  746. 0x12, 0x81, 0x6e, 0x00, 0x12, 0x01, 0x65, 0x00, 0x02, 0x02, 0x76, 0x00,
  747. 0x02, 0x04, 0x16, 0x00, 0x02, 0x85, 0x6c, 0x00, 0x02, 0x04, 0x1d, 0x00,
  748. 0x01, 0x8b, 0x6c, 0x00, 0x11, 0x01, 0x1d, 0x00, 0x11, 0x05, 0x6c, 0x00,
  749. 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00, 0x12, 0x81, 0x16, 0x00,
  750. 0x12, 0x03, 0x7e, 0x00, 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00,
  751. 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00,
  752. 0x03, 0x04, 0x85, 0x00, 0x03, 0x82, 0x85, 0x00, 0x03, 0x85, 0x1b, 0x00,
  753. 0x03, 0x04, 0x1d, 0x00, 0x02, 0x8b, 0x6c, 0x00, 0x12, 0x01, 0x16, 0x00,
  754. 0x12, 0x03, 0x8a, 0x00, 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00,
  755. 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00,
  756. 0x03, 0x04, 0x8a, 0x00, 0x03, 0x85, 0x67, 0x00, 0x03, 0x04, 0x1d, 0x00,
  757. 0x02, 0x8b, 0x6c, 0x00, 0x12, 0x01, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00,
  758. 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x6e, 0x00,
  759. 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00, 0x03, 0x04, 0x8f, 0x00,
  760. 0x03, 0x85, 0x67, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x01, 0x8b, 0x6c, 0x00,
  761. 0x11, 0x01, 0x1d, 0x00, 0x11, 0x05, 0x67, 0x00, 0x01, 0x08, 0x69, 0x00,
  762. 0x01, 0x07, 0x6c, 0x00, 0x12, 0x81, 0x16, 0x00, 0x12, 0x03, 0x8a, 0x00,
  763. 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00, 0x03, 0x82, 0x8a, 0x00,
  764. 0x03, 0x07, 0x6c, 0x00, 0x14, 0x81, 0x6e, 0x00, 0x14, 0x01, 0x65, 0x00,
  765. 0x04, 0x02, 0x76, 0x00, 0x04, 0x04, 0x85, 0x00, 0x04, 0x82, 0x85, 0x00,
  766. 0x03, 0x85, 0x1b, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x02, 0x8b, 0x6c, 0x00,
  767. 0x12, 0x01, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00, 0x02, 0x08, 0x69, 0x00,
  768. 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00,
  769. 0x03, 0x02, 0x76, 0x00, 0x03, 0x04, 0x85, 0x00, 0x03, 0x85, 0x99, 0x00,
  770. 0x03, 0x04, 0x1d, 0x00, 0x01, 0x8b, 0x6c, 0x00, 0x11, 0x01, 0x1d, 0x00,
  771. 0x11, 0x05, 0x99, 0x00, 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00,
  772. 0x12, 0x81, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00, 0x02, 0x08, 0x69, 0x00,
  773. 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x9b, 0x00, 0x23, 0x01, 0x6e, 0x00,
  774. 0x23, 0x01, 0x65, 0x00, 0x13, 0x02, 0x76, 0x00, 0x03, 0x08, 0xa2, 0x00,
  775. 0x03, 0x07, 0x6c, 0x00, 0x04, 0x82, 0x85, 0x00, 0x14, 0x81, 0x6e, 0x00,
  776. 0x14, 0x01, 0x65, 0x00, 0x24, 0x01, 0x9b, 0x00, 0x24, 0x01, 0x8f, 0x00,
  777. 0x14, 0x08, 0xa4, 0x00, 0x14, 0x03, 0x6e, 0x00, 0x04, 0x02, 0xa6, 0x00,
  778. 0x03, 0x85, 0x1b, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x00, 0x86, 0xae, 0x00,
  779. 0x00, 0x8c, 0x33, 0x00, /* instruct array */
  780. 0xb1, 0x00, /* const pool size */
  781. 0x00, 0x5f, 0x5f, 0x63, 0x61, 0x6c, 0x6c, 0x73, 0x2e, 0x6c, 0x65, 0x6e,
  782. 0x00, 0x6c, 0x65, 0x6e, 0x00, 0x6e, 0x6f, 0x6e, 0x65, 0x00, 0x6d, 0x6f,
  783. 0x64, 0x65, 0x00, 0x30, 0x00, 0x69, 0x6e, 0x66, 0x6f, 0x5f, 0x69, 0x6e,
  784. 0x64, 0x65, 0x78, 0x00, 0x72, 0x61, 0x6e, 0x67, 0x65, 0x00, 0x69, 0x74,
  785. 0x65, 0x72, 0x00, 0x5f, 0x6c, 0x30, 0x00, 0x5f, 0x72, 0x31, 0x00, 0x5f,
  786. 0x72, 0x32, 0x00, 0x5f, 0x72, 0x33, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x61,
  787. 0x31, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x61, 0x32, 0x00, 0x5f, 0x6c, 0x30,
  788. 0x2e, 0x61, 0x33, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x5f, 0x5f, 0x6e, 0x65,
  789. 0x78, 0x74, 0x5f, 0x5f, 0x00, 0x69, 0x00, 0x32, 0x00, 0x3d, 0x3d, 0x00,
  790. 0x31, 0x00, 0x5f, 0x5f, 0x63, 0x61, 0x6c, 0x6c, 0x73, 0x00, 0x5f, 0x5f,
  791. 0x67, 0x65, 0x74, 0x5f, 0x5f, 0x00, 0x61, 0x6c, 0x77, 0x61, 0x79, 0x73,
  792. 0x00, 0x74, 0x6f, 0x64, 0x6f, 0x00, 0x77, 0x68, 0x65, 0x6e, 0x00, 0x70,
  793. 0x65, 0x72, 0x69, 0x6f, 0x64, 0x5f, 0x6d, 0x73, 0x00, 0x33, 0x00, 0x5f,
  794. 0x5f, 0x74, 0x69, 0x63, 0x6b, 0x00, 0x3e, 0x00, 0x2b, 0x00, 0x5f, 0x5f,
  795. 0x73, 0x65, 0x74, 0x5f, 0x5f, 0x00, 0x2d, 0x31, 0x00, /* const pool */
  796. };
  797. ByteCodeFrame bytecode_frame = {.const_pool =
  798. {
  799. .arg_buff = NULL,
  800. .content_offset_now = 0,
  801. .size = 0,
  802. .content_start = NULL,
  803. .output_redirect_fun = NULL,
  804. .output_f = NULL,
  805. },
  806. .instruct_array = {
  807. .arg_buff = NULL,
  808. .content_offset_now = 0,
  809. .size = 0,
  810. .content_start = NULL,
  811. .output_redirect_fun = NULL,
  812. .output_f = NULL,
  813. }};
  814. byteCodeFrame_loadByteCode(&bytecode_frame, (uint8_t*)bytes);
  815. byteCodeFrame_print(&bytecode_frame);
  816. EXPECT_EQ(instructArray_getSize(&(bytecode_frame.instruct_array)), 520);
  817. EXPECT_EQ(bytecode_frame.const_pool.size, 177);
  818. EXPECT_EQ(pikaMemNow(), 0);
  819. }
  820. TEST(VM, multi_jian) {
  821. pikaMemInfo.heapUsedMax = 0;
  822. Args* buffs = New_strBuff();
  823. char* lines = "a = (3-4) - 4\n";
  824. printf("%s", lines);
  825. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  826. printf("%s", pikaAsm);
  827. pikaMemInfo.heapUsedMax = 0;
  828. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  829. pikaVM_run(self, lines);
  830. /* assert */
  831. int a = obj_getInt(self, "a");
  832. EXPECT_EQ(a, -5);
  833. /* deinit */
  834. args_deinit(buffs);
  835. obj_deinit(self);
  836. EXPECT_EQ(pikaMemNow(), 0);
  837. }
  838. TEST(VM, _3_p_3) {
  839. char* line = "-3+3";
  840. Args* buffs = New_strBuff();
  841. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  842. printf("%s", pikaAsm);
  843. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  844. __platform_printf("BEGIN\r\n");
  845. pikaVM_runAsm(self, pikaAsm);
  846. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  847. EXPECT_STREQ(log_buff[0], "0\r\n");
  848. obj_deinit(self);
  849. args_deinit(buffs);
  850. EXPECT_EQ(pikaMemNow(), 0);
  851. }
  852. TEST(VM, _3_3) {
  853. char* line = "-3-3";
  854. Args* buffs = New_strBuff();
  855. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  856. printf("%s", pikaAsm);
  857. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  858. __platform_printf("BEGIN\r\n");
  859. pikaVM_runAsm(self, pikaAsm);
  860. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  861. EXPECT_STREQ(log_buff[0], "-6\r\n");
  862. obj_deinit(self);
  863. args_deinit(buffs);
  864. EXPECT_EQ(pikaMemNow(), 0);
  865. }
  866. TEST(VM, a_3) {
  867. char* line = "a-3";
  868. Args* buffs = New_strBuff();
  869. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  870. printf("%s", pikaAsm);
  871. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  872. __platform_printf("BEGIN\r\n");
  873. pikaVM_runAsm(self, pikaAsm);
  874. EXPECT_STREQ(log_buff[3], "BEGIN\r\n");
  875. EXPECT_STREQ(log_buff[2], "NameError: name 'a' is not defined\r\n");
  876. obj_deinit(self);
  877. args_deinit(buffs);
  878. EXPECT_EQ(pikaMemNow(), 0);
  879. }
  880. TEST(VM, hex_str) {
  881. char* line = "a = '\\x33\\x35'";
  882. Args* buffs = New_strBuff();
  883. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  884. printf("%s", pikaAsm);
  885. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  886. __platform_printf("BEGIN\r\n");
  887. pikaVM_runAsm(self, pikaAsm);
  888. /* collect */
  889. char* a = obj_getStr(self, "a");
  890. /* assert */
  891. EXPECT_STREQ(a, "35");
  892. /* deinit */
  893. obj_deinit(self);
  894. args_deinit(buffs);
  895. EXPECT_EQ(pikaMemNow(), 0);
  896. }
  897. TEST(VM, hex_bytes) {
  898. char* line =
  899. "a = b'\\x03\\x05'\n"
  900. "a\n";
  901. Args* buffs = New_strBuff();
  902. char* pikaAsm = Parser_linesToAsm(buffs, line);
  903. printf("%s", pikaAsm);
  904. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  905. __platform_printf("BEGIN\r\n");
  906. pikaVM_runAsm(self, pikaAsm);
  907. /* collect */
  908. /* assert */
  909. EXPECT_STREQ(log_buff[0], "b'\\x03\\x05'\r\n");
  910. /* deinit */
  911. obj_deinit(self);
  912. args_deinit(buffs);
  913. EXPECT_EQ(pikaMemNow(), 0);
  914. }
  915. TEST(VM, bytes_equ) {
  916. char* line =
  917. "a = b'\\x03\\x05'\n"
  918. "b = b'\\x03\\x05'\n"
  919. "if a == b:\n"
  920. " c = 1\n"
  921. "b = b'\\x02\\x05'\n"
  922. "if a != b:\n"
  923. " d = 1\n";
  924. Args* buffs = New_strBuff();
  925. char* pikaAsm = Parser_linesToAsm(buffs, line);
  926. printf("%s", pikaAsm);
  927. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  928. __platform_printf("BEGIN\r\n");
  929. pikaVM_runAsm(self, pikaAsm);
  930. /* collect */
  931. /* assert */
  932. int c = obj_getInt(self, "c");
  933. int d = obj_getInt(self, "d");
  934. EXPECT_EQ(c, 1);
  935. EXPECT_EQ(d, 1);
  936. /* deinit */
  937. obj_deinit(self);
  938. args_deinit(buffs);
  939. EXPECT_EQ(pikaMemNow(), 0);
  940. }
  941. #if PIKA_BUILTIN_STRUCT_ENABLE
  942. TEST(VM, tuple_literal) {
  943. char* line = "(1,2,3)\n";
  944. Args* buffs = New_strBuff();
  945. char* pikaAsm = Parser_linesToAsm(buffs, line);
  946. printf("%s", pikaAsm);
  947. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  948. __platform_printf("BEGIN\r\n");
  949. pikaVM_runAsm(self, pikaAsm);
  950. /* collect */
  951. /* assert */
  952. EXPECT_STREQ(log_buff[0], "(1, 2, 3)\r\n");
  953. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  954. /* deinit */
  955. obj_deinit(self);
  956. args_deinit(buffs);
  957. EXPECT_EQ(pikaMemNow(), 0);
  958. }
  959. #endif
  960. TEST(VM, dvd_opt) {
  961. char* line = "a = 10%4\n";
  962. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  963. obj_run(self, line);
  964. /* collect */
  965. int a = obj_getInt(self, "a");
  966. /* assert */
  967. EXPECT_EQ(a, 2);
  968. /* deinit */
  969. obj_deinit(self);
  970. EXPECT_EQ(pikaMemNow(), 0);
  971. }
  972. TEST(VM, del_) {
  973. char* line =
  974. "a = 1\n"
  975. "print(a)\n"
  976. "del a\n"
  977. "print(a)\n";
  978. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  979. obj_run(self, line);
  980. /* collect */
  981. /* assert */
  982. EXPECT_EQ(obj_isArgExist(self, "a"), 0);
  983. /* deinit */
  984. obj_deinit(self);
  985. EXPECT_EQ(pikaMemNow(), 0);
  986. }
  987. TEST(VM, is) {
  988. char* line =
  989. "a = 1\n"
  990. "b = a\n"
  991. "res1 = a is b\n"
  992. "res2 = 1 is 1\n"
  993. "c = TinyObj()\n"
  994. "d = c\n"
  995. "res3 = c is d\n"
  996. "res4 = c is None\n";
  997. ;
  998. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  999. obj_run(self, line);
  1000. /* collect */
  1001. /* assert */
  1002. EXPECT_EQ(obj_getInt(self, "res1"), 1);
  1003. EXPECT_EQ(obj_getInt(self, "res2"), 1);
  1004. EXPECT_EQ(obj_getInt(self, "res3"), 1);
  1005. EXPECT_EQ(obj_getInt(self, "res4"), 0);
  1006. /* deinit */
  1007. obj_deinit(self);
  1008. EXPECT_EQ(pikaMemNow(), 0);
  1009. }
  1010. TEST(VM, in) {
  1011. char* line =
  1012. "res1 = 'a' in 'aaa'\n"
  1013. "res2 = 'aaa' in 'a'\n";
  1014. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1015. obj_run(self, line);
  1016. /* collect */
  1017. /* assert */
  1018. EXPECT_EQ(obj_getInt(self, "res1"), 1);
  1019. EXPECT_EQ(obj_getInt(self, "res2"), 0);
  1020. /* deinit */
  1021. obj_deinit(self);
  1022. EXPECT_EQ(pikaMemNow(), 0);
  1023. }
  1024. #if !PIKA_NANO_ENABLE
  1025. TEST(VM, assert) {
  1026. char* line =
  1027. "assert True\n"
  1028. "assert 1 == 1, 'testparser'\n"
  1029. "res1 = 0\n"
  1030. "try:\n"
  1031. " assert False\n"
  1032. "except:\n"
  1033. " res1 = 1\n";
  1034. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1035. obj_run(self, line);
  1036. /* collect */
  1037. int res1 = obj_getInt(self, "res1");
  1038. /* assert */
  1039. EXPECT_EQ(res1, 1);
  1040. /* deinit */
  1041. obj_deinit(self);
  1042. EXPECT_EQ(pikaMemNow(), 0);
  1043. }
  1044. #endif
  1045. TEST(VM, issue_I5LHJG) {
  1046. char* line =
  1047. "res = (a, 2, 3)\n"
  1048. "[a, 2, 3]\n"
  1049. "{'x': a, 'y': 2, 'z': 3}\n";
  1050. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1051. obj_run(self, line);
  1052. /* collect */
  1053. /* assert */
  1054. EXPECT_EQ(obj_isArgExist(self, "res"), 0);
  1055. /* deinit */
  1056. obj_deinit(self);
  1057. EXPECT_EQ(pikaMemNow(), 0);
  1058. }
  1059. TEST(VM, vars_runtime) {
  1060. char* line =
  1061. "def testvars(a, *b):\n"
  1062. " sum = 0\n"
  1063. " for i in b:\n"
  1064. " sum += i\n"
  1065. " return a * sum\n"
  1066. "res = testvars(6, 2, 3, 4, 5)\n";
  1067. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1068. obj_run(self, line);
  1069. /* collect */
  1070. int res = obj_getInt(self, "res");
  1071. /* assert */
  1072. EXPECT_EQ(res, 84);
  1073. /* deinit */
  1074. obj_deinit(self);
  1075. EXPECT_EQ(pikaMemNow(), 0);
  1076. }
  1077. #if PIKA_BUILTIN_STRUCT_ENABLE
  1078. TEST(VM, list_add) {
  1079. char* line = "[1, 2, 3] + [4, 5, 6]";
  1080. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1081. obj_run(self, line);
  1082. /* collect */
  1083. /* assert */
  1084. EXPECT_STREQ(log_buff[0], "[1, 2, 3, 4, 5, 6]\r\n");
  1085. /* deinit */
  1086. obj_deinit(self);
  1087. EXPECT_EQ(pikaMemNow(), 0);
  1088. }
  1089. #endif
  1090. TEST(VM, science_num) {
  1091. char* line =
  1092. "a = 1.0e-3\n"
  1093. "b = 2e-5\n"
  1094. "c = -3e-5\n"
  1095. "d = 0.4e2\n";
  1096. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1097. obj_run(self, line);
  1098. /* collect */
  1099. double a = obj_getFloat(self, "a");
  1100. double b = obj_getFloat(self, "b");
  1101. double c = obj_getFloat(self, "c");
  1102. double d = obj_getFloat(self, "d");
  1103. /* assert */
  1104. EXPECT_DOUBLE_EQ(a, 1.0e-3);
  1105. EXPECT_DOUBLE_EQ(b, 2.0e-5);
  1106. EXPECT_DOUBLE_EQ(c, -3e-5);
  1107. EXPECT_DOUBLE_EQ(d, 0.4e2);
  1108. /* deinit */
  1109. obj_deinit(self);
  1110. EXPECT_EQ(pikaMemNow(), 0);
  1111. }
  1112. TEST(VM, issue_I5OJQB) {
  1113. char* line = "s = '\\\\'";
  1114. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1115. obj_run(self, line);
  1116. /* collect */
  1117. char* s = obj_getStr(self, "s");
  1118. /* assert */
  1119. EXPECT_STREQ(s, "\\");
  1120. /* deinit */
  1121. obj_deinit(self);
  1122. EXPECT_EQ(pikaMemNow(), 0);
  1123. }
  1124. #if !PIKA_NANO_ENABLE
  1125. TEST(vm, keyword_2) {
  1126. char* line =
  1127. "def test(a, b):\n"
  1128. " print(__kwargs['a'], __kwargs['b'])\n"
  1129. "test(a=1, b= 2)";
  1130. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1131. obj_run(self, line);
  1132. /* collect */
  1133. /* assert */
  1134. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1135. /* deinit */
  1136. obj_deinit(self);
  1137. EXPECT_EQ(pikaMemNow(), 0);
  1138. }
  1139. TEST(vm, keyword_3) {
  1140. char* line =
  1141. "def test(**keys):\n"
  1142. " print(keys['a'], keys['b'])\n"
  1143. "test(a=1, b= 2)";
  1144. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1145. obj_run(self, line);
  1146. /* collect */
  1147. /* assert */
  1148. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1149. /* deinit */
  1150. obj_deinit(self);
  1151. EXPECT_EQ(pikaMemNow(), 0);
  1152. }
  1153. TEST(vm, keyword_4) {
  1154. char* line =
  1155. "def test(a, b, **keys):\n"
  1156. " print(keys['a'], keys['b'], a, b)\n"
  1157. "test(3, 4, a=1, b= 2)";
  1158. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1159. obj_run(self, line);
  1160. /* collect */
  1161. /* assert */
  1162. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  1163. /* deinit */
  1164. obj_deinit(self);
  1165. EXPECT_EQ(pikaMemNow(), 0);
  1166. }
  1167. TEST(vm, vars_keyward) {
  1168. /* init */
  1169. pikaMemInfo.heapUsedMax = 0;
  1170. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1171. extern unsigned char pikaModules_py_a[];
  1172. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1173. /* run */
  1174. __platform_printf("BEGIN\r\n");
  1175. pikaVM_runSingleFile(pikaMain, "../../examples/BuiltIn/function.py");
  1176. /* collect */
  1177. /* assert */
  1178. EXPECT_STREQ(log_buff[1], "1 2 3 4 5 10\r\n");
  1179. EXPECT_STREQ(log_buff[0], "1 2 3 4 5 12\r\n");
  1180. /* deinit */
  1181. obj_deinit(pikaMain);
  1182. EXPECT_EQ(pikaMemNow(), 0);
  1183. }
  1184. TEST(vm, cb_1) {
  1185. /* init */
  1186. pikaMemInfo.heapUsedMax = 0;
  1187. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1188. extern unsigned char pikaModules_py_a[];
  1189. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1190. /* run */
  1191. __platform_printf("BEGIN\r\n");
  1192. pikaVM_runSingleFile(pikaMain, "../../examples/Callback/test1.py");
  1193. /* collect */
  1194. /* assert */
  1195. EXPECT_STREQ(log_buff[3], "__init__\r\n");
  1196. EXPECT_STREQ(log_buff[1], "a\r\n");
  1197. EXPECT_STREQ(log_buff[0], "b\r\n");
  1198. /* deinit */
  1199. obj_deinit(pikaMain);
  1200. EXPECT_EQ(pikaMemNow(), 0);
  1201. }
  1202. TEST(vm, cb_2) {
  1203. /* init */
  1204. pikaMemInfo.heapUsedMax = 0;
  1205. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1206. extern unsigned char pikaModules_py_a[];
  1207. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1208. /* run */
  1209. __platform_printf("BEGIN\r\n");
  1210. pikaVM_runSingleFile(pikaMain, "../../examples/Callback/test2.py");
  1211. /* collect */
  1212. /* assert */
  1213. EXPECT_STREQ(log_buff[4], "__init__\r\n");
  1214. EXPECT_STREQ(log_buff[2], "a\r\n");
  1215. EXPECT_STREQ(log_buff[1], "b\r\n");
  1216. EXPECT_STREQ(log_buff[0], "ppp\r\n");
  1217. /* deinit */
  1218. obj_deinit(pikaMain);
  1219. EXPECT_EQ(pikaMemNow(), 0);
  1220. }
  1221. #endif
  1222. TEST(vm, default_no_input) {
  1223. /* init */
  1224. pikaMemInfo.heapUsedMax = 0;
  1225. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1226. extern unsigned char pikaModules_py_a[];
  1227. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1228. /* run */
  1229. __platform_printf("BEGIN\r\n");
  1230. obj_run(pikaMain,
  1231. "def test(a = 1):\n"
  1232. " print(a)\n"
  1233. "test()");
  1234. /* collect */
  1235. /* assert */
  1236. EXPECT_STREQ(log_buff[0], "1\r\n");
  1237. /* deinit */
  1238. obj_deinit(pikaMain);
  1239. EXPECT_EQ(pikaMemNow(), 0);
  1240. }
  1241. TEST(vm, default_1) {
  1242. /* init */
  1243. pikaMemInfo.heapUsedMax = 0;
  1244. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1245. extern unsigned char pikaModules_py_a[];
  1246. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1247. /* run */
  1248. __platform_printf("BEGIN\r\n");
  1249. obj_run(pikaMain,
  1250. "def test(a = 1):\n"
  1251. " print(a)\n"
  1252. "test(a = 2)");
  1253. /* collect */
  1254. /* assert */
  1255. EXPECT_STREQ(log_buff[0], "2\r\n");
  1256. /* deinit */
  1257. obj_deinit(pikaMain);
  1258. EXPECT_EQ(pikaMemNow(), 0);
  1259. }
  1260. TEST(vm, default_2) {
  1261. /* init */
  1262. pikaMemInfo.heapUsedMax = 0;
  1263. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1264. extern unsigned char pikaModules_py_a[];
  1265. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1266. /* run */
  1267. __platform_printf("BEGIN\r\n");
  1268. obj_run(pikaMain,
  1269. "def test(a, b = 1):\n"
  1270. " print(a, b)\n"
  1271. "test(1, b = 2)");
  1272. /* collect */
  1273. /* assert */
  1274. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1275. /* deinit */
  1276. obj_deinit(pikaMain);
  1277. EXPECT_EQ(pikaMemNow(), 0);
  1278. }
  1279. TEST(vm, default_3) {
  1280. /* init */
  1281. pikaMemInfo.heapUsedMax = 0;
  1282. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1283. extern unsigned char pikaModules_py_a[];
  1284. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1285. /* run */
  1286. __platform_printf("BEGIN\r\n");
  1287. obj_run(pikaMain,
  1288. "def myprint(end='\\n', *var):\n"
  1289. " for v in var:\n"
  1290. " print(v, end = end)\n"
  1291. "myprint(1, 2)\n"
  1292. "myprint(1, 2, end=';\\n')\n");
  1293. /* collect */
  1294. /* assert */
  1295. EXPECT_STREQ(log_buff[3], "1\n");
  1296. EXPECT_STREQ(log_buff[2], "2\n");
  1297. EXPECT_STREQ(log_buff[1], "1;\n");
  1298. EXPECT_STREQ(log_buff[0], "2;\n");
  1299. /* deinit */
  1300. obj_deinit(pikaMain);
  1301. EXPECT_EQ(pikaMemNow(), 0);
  1302. }
  1303. TEST(vm, default_4) {
  1304. /* init */
  1305. pikaMemInfo.heapUsedMax = 0;
  1306. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1307. extern unsigned char pikaModules_py_a[];
  1308. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1309. /* run */
  1310. __platform_printf("BEGIN\r\n");
  1311. obj_run(pikaMain,
  1312. "def test(end=5, *var):\n"
  1313. " res = end\n"
  1314. " print('end', end)\n"
  1315. " for v in var:\n"
  1316. " res += v\n"
  1317. " return res\n"
  1318. "test(1, 2, 3)\n"
  1319. "test(1, 2, 3, end=6)\n");
  1320. /* collect */
  1321. /* assert */
  1322. EXPECT_STREQ(log_buff[3], "end 5\r\n");
  1323. EXPECT_STREQ(log_buff[2], "11\r\n");
  1324. EXPECT_STREQ(log_buff[1], "end 6\r\n");
  1325. EXPECT_STREQ(log_buff[0], "12\r\n");
  1326. /* deinit */
  1327. obj_deinit(pikaMain);
  1328. EXPECT_EQ(pikaMemNow(), 0);
  1329. }
  1330. TEST(vm, none) {
  1331. /* init */
  1332. pikaMemInfo.heapUsedMax = 0;
  1333. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1334. extern unsigned char pikaModules_py_a[];
  1335. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1336. /* run */
  1337. __platform_printf("BEGIN\r\n");
  1338. obj_run(pikaMain, "GTestTask.test(1, None)");
  1339. /* collect */
  1340. /* assert */
  1341. EXPECT_STREQ(log_buff[0], "GTestTask_test: 1, 0\r\n");
  1342. /* deinit */
  1343. obj_deinit(pikaMain);
  1344. EXPECT_EQ(pikaMemNow(), 0);
  1345. }
  1346. #if !PIKA_NANO_ENABLE
  1347. TEST(vm, super_) {
  1348. /* init */
  1349. pikaMemInfo.heapUsedMax = 0;
  1350. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1351. extern unsigned char pikaModules_py_a[];
  1352. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1353. /* run */
  1354. __platform_printf("BEGIN\r\n");
  1355. obj_run(pikaMain,
  1356. "class test:\n"
  1357. " def __init__(self):\n"
  1358. " print('in test init')\n"
  1359. "class test2(test):\n"
  1360. " def __init__(self):\n"
  1361. " super().__init__()\n"
  1362. " print('in test2 init')\n"
  1363. "t = test2()");
  1364. /* collect */
  1365. /* assert */
  1366. EXPECT_STREQ(log_buff[0], "in test2 init\r\n");
  1367. EXPECT_STREQ(log_buff[1], "in test init\r\n");
  1368. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1369. /* deinit */
  1370. obj_deinit(pikaMain);
  1371. EXPECT_EQ(pikaMemNow(), 0);
  1372. }
  1373. TEST(vm, super_val) {
  1374. /* init */
  1375. pikaMemInfo.heapUsedMax = 0;
  1376. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1377. extern unsigned char pikaModules_py_a[];
  1378. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1379. /* run */
  1380. __platform_printf("BEGIN\r\n");
  1381. obj_run(pikaMain,
  1382. "class test:\n"
  1383. " def __init__(self):\n"
  1384. " self.a = 1\n"
  1385. " self.b = 2\n"
  1386. "class test2(test):\n"
  1387. " def __init__(self):\n"
  1388. " super().__init__()\n"
  1389. " self.c = 3\n"
  1390. " self.d = 4\n"
  1391. "t = test2()\n"
  1392. "print(t.a, t.b, t.c, t.d)\n");
  1393. /* collect */
  1394. /* assert */
  1395. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1396. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  1397. /* deinit */
  1398. obj_deinit(pikaMain);
  1399. EXPECT_EQ(pikaMemNow(), 0);
  1400. }
  1401. #endif
  1402. #if !PIKA_NANO_ENABLE
  1403. TEST(vm, multi_return) {
  1404. /* init */
  1405. pikaMemInfo.heapUsedMax = 0;
  1406. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1407. extern unsigned char pikaModules_py_a[];
  1408. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1409. /* run */
  1410. __platform_printf("BEGIN\r\n");
  1411. obj_run(pikaMain, "a,b = (1,2)");
  1412. /* collect */
  1413. int a = obj_getInt(pikaMain, "a");
  1414. int b = obj_getInt(pikaMain, "b");
  1415. /* assert */
  1416. EXPECT_EQ(a, 1);
  1417. EXPECT_EQ(b, 2);
  1418. /* deinit */
  1419. obj_deinit(pikaMain);
  1420. EXPECT_EQ(pikaMemNow(), 0);
  1421. }
  1422. #endif
  1423. #if !PIKA_NANO_ENABLE
  1424. TEST(vm, multi_return_fn) {
  1425. /* init */
  1426. pikaMemInfo.heapUsedMax = 0;
  1427. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1428. extern unsigned char pikaModules_py_a[];
  1429. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1430. /* run */
  1431. __platform_printf("BEGIN\r\n");
  1432. obj_run(pikaMain,
  1433. "def test():\n"
  1434. " return 1,2\n"
  1435. "a,b = test()");
  1436. /* collect */
  1437. int a = obj_getInt(pikaMain, "a");
  1438. int b = obj_getInt(pikaMain, "b");
  1439. /* assert */
  1440. EXPECT_EQ(a, 1);
  1441. EXPECT_EQ(b, 2);
  1442. /* deinit */
  1443. obj_deinit(pikaMain);
  1444. EXPECT_EQ(pikaMemNow(), 0);
  1445. }
  1446. #endif
  1447. TEST(vm, rang_1) {
  1448. /* init */
  1449. pikaMemInfo.heapUsedMax = 0;
  1450. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1451. extern unsigned char pikaModules_py_a[];
  1452. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1453. /* run */
  1454. __platform_printf("BEGIN\r\n");
  1455. obj_run(pikaMain,
  1456. "sum = 0\n"
  1457. "for i in range(10):\n"
  1458. " sum += i\n");
  1459. /* collect */
  1460. int sum = obj_getInt(pikaMain, "sum");
  1461. /* assert */
  1462. EXPECT_EQ(sum, 45);
  1463. /* deinit */
  1464. obj_deinit(pikaMain);
  1465. EXPECT_EQ(pikaMemNow(), 0);
  1466. }
  1467. TEST(vm, rang_3) {
  1468. /* init */
  1469. pikaMemInfo.heapUsedMax = 0;
  1470. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1471. extern unsigned char pikaModules_py_a[];
  1472. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1473. /* run */
  1474. __platform_printf("BEGIN\r\n");
  1475. obj_run(pikaMain,
  1476. "sum = 0\n"
  1477. "for i in range(-5, 20, 2):\n"
  1478. " sum += i\n");
  1479. /* collect */
  1480. int sum = obj_getInt(pikaMain, "sum");
  1481. /* assert */
  1482. EXPECT_EQ(sum, 91);
  1483. /* deinit */
  1484. obj_deinit(pikaMain);
  1485. EXPECT_EQ(pikaMemNow(), 0);
  1486. }
  1487. TEST(vm, test64) {
  1488. /* init */
  1489. pikaMemInfo.heapUsedMax = 0;
  1490. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1491. extern unsigned char pikaModules_py_a[];
  1492. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1493. /* run */
  1494. __platform_printf("BEGIN\r\n");
  1495. obj_run(pikaMain,
  1496. "import GTestTask\n"
  1497. "res = GTestTask.test64(4294967295, 20)\n");
  1498. /* collect */
  1499. int64_t res = obj_getInt(pikaMain, "res");
  1500. /* assert */
  1501. EXPECT_EQ(res, 4294967295 * 20);
  1502. /* deinit */
  1503. obj_deinit(pikaMain);
  1504. EXPECT_EQ(pikaMemNow(), 0);
  1505. }
  1506. #if !PIKA_NANO_ENABLE
  1507. TEST(vm, exit) {
  1508. /* init */
  1509. pikaMemInfo.heapUsedMax = 0;
  1510. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1511. extern unsigned char pikaModules_py_a[];
  1512. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1513. /* run */
  1514. __platform_printf("BEGIN\r\n");
  1515. obj_run(pikaMain,
  1516. "i = 0\n"
  1517. "while True:\n"
  1518. " i += 1\n"
  1519. " print(i)\n"
  1520. " if i == 10:\n"
  1521. " exit()\n");
  1522. /* collect */
  1523. int i = obj_getInt(pikaMain, "i");
  1524. /* assert */
  1525. EXPECT_EQ(i, 10);
  1526. /* deinit */
  1527. obj_deinit(pikaMain);
  1528. EXPECT_EQ(pikaMemNow(), 0);
  1529. }
  1530. TEST(vm, exit_fn) {
  1531. /* init */
  1532. pikaMemInfo.heapUsedMax = 0;
  1533. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1534. extern unsigned char pikaModules_py_a[];
  1535. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1536. /* run */
  1537. __platform_printf("BEGIN\r\n");
  1538. obj_run(pikaMain,
  1539. "i = 0\n"
  1540. "def test():\n"
  1541. " global i\n"
  1542. " while True:\n"
  1543. " i += 1\n"
  1544. " print(i)\n"
  1545. " if i == 10:\n"
  1546. " exit()\n"
  1547. "while True:\n"
  1548. " test()\n");
  1549. /* collect */
  1550. int i = obj_getInt(pikaMain, "i");
  1551. /* assert */
  1552. EXPECT_EQ(i, 10);
  1553. /* deinit */
  1554. obj_deinit(pikaMain);
  1555. EXPECT_EQ(pikaMemNow(), 0);
  1556. }
  1557. #endif
  1558. TEST(vm, pass_) {
  1559. /* init */
  1560. pikaMemInfo.heapUsedMax = 0;
  1561. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1562. extern unsigned char pikaModules_py_a[];
  1563. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1564. /* run */
  1565. __platform_printf("BEGIN\r\n");
  1566. obj_run(pikaMain,
  1567. "pass\n"
  1568. "def testpass():\n"
  1569. " pass\n"
  1570. "for i in range(10):\n"
  1571. " pass\n"
  1572. "testpass()\n"
  1573. "print('after pass')\n");
  1574. /* collect */
  1575. /* assert */
  1576. EXPECT_STREQ(log_buff[0], "after pass\r\n");
  1577. /* deinit */
  1578. obj_deinit(pikaMain);
  1579. EXPECT_EQ(pikaMemNow(), 0);
  1580. }
  1581. TEST(vm, test64_hex) {
  1582. /* init */
  1583. pikaMemInfo.heapUsedMax = 0;
  1584. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1585. extern unsigned char pikaModules_py_a[];
  1586. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1587. /* run */
  1588. __platform_printf("BEGIN\r\n");
  1589. obj_run(pikaMain,
  1590. "import GTestTask\n"
  1591. "res = GTestTask.test64(0xffffffff, 20)\n");
  1592. /* collect */
  1593. int64_t res = obj_getInt(pikaMain, "res");
  1594. /* assert */
  1595. EXPECT_EQ(res, 4294967295 * 20);
  1596. /* deinit */
  1597. obj_deinit(pikaMain);
  1598. EXPECT_EQ(pikaMemNow(), 0);
  1599. }
  1600. TEST(vm, test64_hex_print) {
  1601. /* init */
  1602. pikaMemInfo.heapUsedMax = 0;
  1603. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1604. extern unsigned char pikaModules_py_a[];
  1605. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1606. /* run */
  1607. __platform_printf("BEGIN\r\n");
  1608. obj_run(pikaMain,
  1609. "0xffffffff\n"
  1610. "print(0xffffffff)\n");
  1611. /* collect */
  1612. /* assert */
  1613. EXPECT_STREQ(log_buff[0], "4294967295\r\n");
  1614. EXPECT_STREQ(log_buff[1], "4294967295\r\n");
  1615. /* deinit */
  1616. obj_deinit(pikaMain);
  1617. EXPECT_EQ(pikaMemNow(), 0);
  1618. }