VM-test.cpp 65 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199
  1. #include "test_common.h"
  2. TEST_START
  3. extern "C" {
  4. extern char log_buff[LOG_BUFF_MAX][LOG_SIZE];
  5. /* test head */
  6. VMParameters* pikaVM_runAsmWithPars(PikaObj* self,
  7. VMParameters* locals,
  8. VMParameters* globals,
  9. char* pikaAsm);
  10. char* Parser_LineToAsm(Args* buffs, char* line, Stack* blockStack);
  11. char* Parser_linesToAsm(Args* outBuffs, char* multiLine);
  12. int32_t __clearInvokeStackes(VMParameters* locals);
  13. int32_t pikaVM_runAsmLine(PikaObj* self,
  14. VMParameters* locals,
  15. VMParameters* globals,
  16. char* pikaAsm,
  17. int32_t lineAddr);
  18. }
  19. TEST(VM, num1) {
  20. char* line = "1";
  21. Args* buffs = New_strBuff();
  22. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  23. printf("%s", pikaAsm);
  24. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  25. __platform_printf("BEGIN\r\n");
  26. pikaVM_runAsm(self, pikaAsm);
  27. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  28. EXPECT_STREQ(log_buff[0], "1\r\n");
  29. obj_deinit(self);
  30. args_deinit(buffs);
  31. EXPECT_EQ(pikaMemNow(), 0);
  32. }
  33. TEST(VM, a_1) {
  34. char* line = "a = 1";
  35. Args* buffs = New_strBuff();
  36. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  37. printf("%s", pikaAsm);
  38. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  39. pikaVM_runAsm(self, pikaAsm);
  40. ASSERT_EQ(args_getInt(self->list, "a"), 1);
  41. obj_deinit(self);
  42. args_deinit(buffs);
  43. EXPECT_EQ(pikaMemNow(), 0);
  44. }
  45. TEST(VM, a_1d1) {
  46. char* line = "a = 1.1";
  47. Args* buffs = New_strBuff();
  48. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  49. printf("%s", pikaAsm);
  50. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  51. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  52. ASSERT_FLOAT_EQ(args_getFloat(globals->list, "a"), 1.1);
  53. obj_deinit(self);
  54. // obj_deinit(globals);
  55. args_deinit(buffs);
  56. EXPECT_EQ(pikaMemNow(), 0);
  57. }
  58. TEST(VM, str_xy) {
  59. char* line = "a = 'xy'";
  60. Args* buffs = New_strBuff();
  61. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  62. printf("%s", pikaAsm);
  63. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  64. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  65. ASSERT_STREQ(args_getStr(globals->list, "a"), "xy");
  66. obj_deinit(self);
  67. // obj_deinit(globals);
  68. args_deinit(buffs);
  69. EXPECT_EQ(pikaMemNow(), 0);
  70. }
  71. TEST(VM, str_xy_space) {
  72. char* line = "a = 'xy '";
  73. Args* buffs = New_strBuff();
  74. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  75. printf("%s", pikaAsm);
  76. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  77. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  78. ASSERT_STREQ(args_getStr(globals->list, "a"), "xy ");
  79. obj_deinit(self);
  80. args_deinit(buffs);
  81. // obj_deinit(globals);
  82. EXPECT_EQ(pikaMemNow(), 0);
  83. }
  84. TEST(VM, ref_a_b) {
  85. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  86. Args* buffs = New_strBuff();
  87. pikaVM_runAsm(self, Parser_LineToAsm(buffs, "a = 'xy '", NULL));
  88. args_deinit(buffs);
  89. obj_deinit(self);
  90. // obj_deinit(globals);
  91. EXPECT_EQ(pikaMemNow(), 0);
  92. }
  93. TEST(VM, Run_add) {
  94. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  95. Args* buffs = New_strBuff();
  96. char* pikaAsm = Parser_LineToAsm(buffs, "a = plusInt(1,2)", NULL);
  97. __platform_printf("%s", pikaAsm);
  98. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  99. args_deinit(buffs);
  100. int a = args_getInt(globals->list, "a");
  101. ASSERT_EQ(a, 3);
  102. obj_deinit(self);
  103. // obj_deinit(globals);
  104. EXPECT_EQ(pikaMemNow(), 0);
  105. }
  106. TEST(VM, Run_add_multy) {
  107. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  108. Args* buffs = New_strBuff();
  109. pikaVM_runAsm(self, Parser_LineToAsm(buffs, "b = 2", NULL));
  110. args_deinit(buffs);
  111. obj_deinit(self);
  112. // obj_deinit(globals);
  113. EXPECT_EQ(pikaMemNow(), 0);
  114. }
  115. TEST(VM, Run_add_1_2_3) {
  116. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  117. Args* buffs = New_strBuff();
  118. char* pikaAsm =
  119. Parser_LineToAsm(buffs, "a = plusInt(1, plusInt(2,3) )", NULL);
  120. __platform_printf("%s", pikaAsm);
  121. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  122. args_deinit(buffs);
  123. int a = args_getInt(globals->list, "a");
  124. ASSERT_EQ(a, 6);
  125. obj_deinit(self);
  126. // obj_deinit(globals);
  127. EXPECT_EQ(pikaMemNow(), 0);
  128. }
  129. extern PikaMemInfo pikaMemInfo;
  130. TEST(VM, WHILE) {
  131. pikaMemInfo.heapUsedMax = 0;
  132. Args* buffs = New_strBuff();
  133. char* lines =(char *)
  134. "a = 1\n"
  135. "b = 0\n"
  136. "while a:\n"
  137. " b = 1\n"
  138. " a = 0\n"
  139. "\n";
  140. printf("%s", lines);
  141. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  142. printf("%s", pikaAsm);
  143. pikaMemInfo.heapUsedMax = 0;
  144. PikaObj* self = New_TinyObj(NULL);
  145. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  146. EXPECT_EQ(args_getInt(globals->list, "a"), 0);
  147. EXPECT_EQ(args_getInt(globals->list, "b"), 1);
  148. // obj_deinit(globals);
  149. args_deinit(buffs);
  150. obj_deinit(self);
  151. EXPECT_EQ(pikaMemNow(), 0);
  152. }
  153. TEST(VM, a_1_1) {
  154. char* line = "a = 1 + 1";
  155. Args* buffs = New_strBuff();
  156. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  157. printf("%s", pikaAsm);
  158. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  159. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  160. int res = args_getInt(globals->list, "a");
  161. obj_deinit(self);
  162. args_deinit(buffs);
  163. ASSERT_EQ(res, 2);
  164. EXPECT_EQ(pikaMemNow(), 0);
  165. }
  166. TEST(VM, a_1_1d1) {
  167. char* line = "a = 1 + 1.1";
  168. Args* buffs = New_strBuff();
  169. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  170. printf("%s", pikaAsm);
  171. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  172. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  173. float res = args_getFloat(globals->list, "a");
  174. obj_deinit(self);
  175. // obj_deinit(globals);
  176. args_deinit(buffs);
  177. ASSERT_FLOAT_EQ(res, 2.1);
  178. EXPECT_EQ(pikaMemNow(), 0);
  179. }
  180. TEST(VM, a_jjcc) {
  181. char* line = "a = (1 + 1.1) * 3 - 2 /4.0";
  182. printf("%s\r\n", line);
  183. Args* buffs = New_strBuff();
  184. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  185. printf("%s", pikaAsm);
  186. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  187. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  188. float res = args_getFloat(globals->list, "a");
  189. obj_deinit(self);
  190. args_deinit(buffs);
  191. // obj_deinit(globals);
  192. ASSERT_FLOAT_EQ(res, 5.8);
  193. EXPECT_EQ(pikaMemNow(), 0);
  194. }
  195. TEST(VM, while_a_1to10) {
  196. char* line =
  197. "a = 0\n"
  198. "while a < 10:\n"
  199. " a += 1\n"
  200. " print(a)\n"
  201. "\n";
  202. Args* buffs = New_strBuff();
  203. char* pikaAsm = Parser_linesToAsm(buffs, line);
  204. printf("%s", pikaAsm);
  205. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  206. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  207. int res = args_getInt(globals->list, "a");
  208. obj_deinit(self);
  209. args_deinit(buffs);
  210. // obj_deinit(globals);
  211. ASSERT_FLOAT_EQ(res, 10);
  212. EXPECT_EQ(pikaMemNow(), 0);
  213. }
  214. TEST(VM, mem_x) {
  215. char* line =
  216. "mem = PikaStdLib.MemChecker()\n"
  217. "mem.x = 1\n"
  218. "print(mem.x)\n"
  219. "mem.x = 2\n"
  220. "print(mem.x)\n"
  221. "\n";
  222. Args* buffs = New_strBuff();
  223. char* pikaAsm = Parser_linesToAsm(buffs, line);
  224. printf("%s", pikaAsm);
  225. PikaObj* self = newRootObj("", New_PikaMain);
  226. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  227. int res = obj_getInt(globals, "mem.x");
  228. EXPECT_EQ(2, res);
  229. obj_deinit(self);
  230. // obj_deinit(globals);
  231. args_deinit(buffs);
  232. EXPECT_EQ(pikaMemNow(), 0);
  233. }
  234. TEST(VM, RET_instruct) {
  235. char* pikaAsm =
  236. "B1\n"
  237. "0 NUM 13\n"
  238. "0 RET \n"
  239. "0 NUM 2\n"
  240. "0 RET \n"
  241. "B0\n";
  242. PikaObj* self = New_TinyObj(NULL);
  243. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  244. Arg* return_arg = args_getArg(globals->list, "@rt");
  245. int num = arg_getInt(return_arg);
  246. EXPECT_EQ(num, 13);
  247. obj_deinit(self);
  248. // obj_deinit(globals);
  249. EXPECT_EQ(pikaMemNow(), 0);
  250. }
  251. TEST(VM, RUN_DEF) {
  252. char* pikaAsm =
  253. "B0\n"
  254. "0 DEF test()\n"
  255. "0 JMP 1\n"
  256. "B1\n"
  257. "0 NUM 1\n"
  258. "0 RET \n"
  259. "B0\n"
  260. "0 RUN test\n"
  261. "0 OUT a\n";
  262. PikaObj* self = New_TinyObj(NULL);
  263. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  264. int num = obj_getInt(globals, "a");
  265. EXPECT_EQ(num, 1);
  266. obj_deinit(self);
  267. // obj_deinit(globals);
  268. EXPECT_EQ(pikaMemNow(), 0);
  269. }
  270. TEST(VM, RUN_global) {
  271. char* pikaAsm =
  272. "B0\n"
  273. "0 NUM 1\n"
  274. "0 OUT a\n"
  275. "B0\n"
  276. "0 DEF test()\n"
  277. "0 JMP 1\n"
  278. "B1\n"
  279. "0 REF a\n"
  280. "0 RET \n"
  281. "B0\n"
  282. "0 RUN test\n"
  283. "0 OUT b\n";
  284. PikaObj* self = New_TinyObj(NULL);
  285. pikaVM_runAsm(self, pikaAsm);
  286. int a = obj_getInt(self, "a");
  287. int b = obj_getInt(self, "b");
  288. EXPECT_EQ(a, 1);
  289. EXPECT_EQ(b, 1);
  290. obj_deinit(self);
  291. // obj_deinit(globals);
  292. EXPECT_EQ(pikaMemNow(), 0);
  293. }
  294. TEST(VM, RUN_local_b) {
  295. char* pikaAsm =
  296. "B0\n"
  297. "0 NUM 1\n"
  298. "0 OUT a\n"
  299. "B0\n"
  300. "0 DEF test()\n"
  301. "0 JMP 1\n"
  302. "B1\n"
  303. "0 REF a\n"
  304. "0 OUT b\n"
  305. "1 REF b\n"
  306. "1 REF a\n"
  307. "0 OPT +\n"
  308. "0 RET \n"
  309. "B0\n"
  310. "0 RUN test\n"
  311. "0 OUT c\n";
  312. PikaObj* self = newRootObj("", New_BaseObj);
  313. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  314. int a = obj_getInt(globals, "a");
  315. int b = obj_getInt(globals, "b");
  316. int c = obj_getInt(globals, "c");
  317. EXPECT_EQ(a, 1);
  318. /* b is local, should not be exist in globals */
  319. EXPECT_EQ(b, -999999999);
  320. EXPECT_EQ(c, 2);
  321. obj_deinit(self);
  322. // obj_deinit(globals);
  323. EXPECT_EQ(pikaMemNow(), 0);
  324. }
  325. TEST(VM, RUN_DEF_add) {
  326. char* pikaAsm =
  327. "B0\n"
  328. "0 DEF add(a,b)\n"
  329. "0 JMP 1\n"
  330. "B1\n"
  331. "1 REF b\n"
  332. "1 REF a\n"
  333. "0 OPT +\n"
  334. "0 RET \n"
  335. "B0\n"
  336. "1 NUM 1\n"
  337. "1 NUM 2\n"
  338. "0 RUN add\n"
  339. "0 OUT c\n";
  340. PikaObj* self = newRootObj("", New_BaseObj);
  341. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  342. int a = obj_getInt(globals, "a");
  343. int b = obj_getInt(globals, "b");
  344. int c = obj_getInt(globals, "c");
  345. /* a is local, should not be exist in globals */
  346. EXPECT_EQ(a, -999999999);
  347. /* b is local, should not be exist in globals */
  348. EXPECT_EQ(b, -999999999);
  349. EXPECT_EQ(c, 3);
  350. obj_deinit(self);
  351. // obj_deinit(globals);
  352. EXPECT_EQ(pikaMemNow(), 0);
  353. }
  354. TEST(VM, run_def_add) {
  355. char* line =
  356. "def add(a, b):\n"
  357. " return a + b\n"
  358. "c = add(1,3)\n"
  359. "\n";
  360. Args* buffs = New_strBuff();
  361. char* pikaAsm = Parser_linesToAsm(buffs, line);
  362. printf("%s", pikaAsm);
  363. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  364. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  365. int c = args_getInt(globals->list, "c");
  366. obj_deinit(self);
  367. args_deinit(buffs);
  368. // obj_deinit(globals);
  369. ASSERT_FLOAT_EQ(c, 4);
  370. EXPECT_EQ(pikaMemNow(), 0);
  371. }
  372. TEST(VM, equ) {
  373. char* line =
  374. "if -1 == -1:\n"
  375. " a = 1\n"
  376. "\n";
  377. Args* buffs = New_strBuff();
  378. char* pikaAsm = Parser_linesToAsm(buffs, line);
  379. printf("%s", pikaAsm);
  380. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  381. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  382. int c = args_getInt(globals->list, "a");
  383. obj_deinit(self);
  384. args_deinit(buffs);
  385. // obj_deinit(globals);
  386. ASSERT_FLOAT_EQ(c, 1);
  387. EXPECT_EQ(pikaMemNow(), 0);
  388. }
  389. TEST(VM, if_elif) {
  390. char* line =
  391. "a = 2\n"
  392. "b = 0\n"
  393. "if a > 1:\n"
  394. " b = 1\n"
  395. "elif a > 0:\n"
  396. " b = 2\n"
  397. "\n";
  398. Args* buffs = New_strBuff();
  399. char* pikaAsm = Parser_linesToAsm(buffs, line);
  400. printf("%s", pikaAsm);
  401. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  402. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  403. int b = args_getInt(globals->list, "b");
  404. obj_deinit(self);
  405. args_deinit(buffs);
  406. ASSERT_FLOAT_EQ(b, 1);
  407. EXPECT_EQ(pikaMemNow(), 0);
  408. }
  409. TEST(VM, if_else) {
  410. char* line =
  411. "a = 0\n"
  412. "b = 0\n"
  413. "if a > 1:\n"
  414. " b = 1\n"
  415. "else:\n"
  416. " b = 2\n"
  417. "\n";
  418. Args* buffs = New_strBuff();
  419. char* pikaAsm = Parser_linesToAsm(buffs, line);
  420. printf("%s", pikaAsm);
  421. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  422. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  423. int b = args_getInt(globals->list, "b");
  424. obj_deinit(self);
  425. args_deinit(buffs);
  426. // obj_deinit(globals);
  427. ASSERT_FLOAT_EQ(b, 2);
  428. EXPECT_EQ(pikaMemNow(), 0);
  429. }
  430. TEST(VM, if_else_) {
  431. char* line =
  432. "a = 2\n"
  433. "b = 0\n"
  434. "if a > 1:\n"
  435. " b = 1\n"
  436. "else:\n"
  437. " b = 2\n"
  438. "\n";
  439. Args* buffs = New_strBuff();
  440. char* pikaAsm = Parser_linesToAsm(buffs, line);
  441. printf("%s", pikaAsm);
  442. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  443. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  444. int b = args_getInt(globals->list, "b");
  445. obj_deinit(self);
  446. args_deinit(buffs);
  447. // obj_deinit(globals);
  448. ASSERT_FLOAT_EQ(b, 1);
  449. EXPECT_EQ(pikaMemNow(), 0);
  450. }
  451. TEST(VM, EST) {
  452. char* pikaAsm =
  453. "B0\n"
  454. "0 EST a\n"
  455. "0 OUT b\n";
  456. PikaObj* self = newRootObj("", New_BaseObj);
  457. pikaVM_runAsm(self, pikaAsm);
  458. int a = obj_getInt(self, "a");
  459. int b = obj_getInt(self, "b");
  460. /* a is local, should not be exist in globals */
  461. EXPECT_EQ(a, -999999999);
  462. /* b is local, should not be exist in globals */
  463. EXPECT_EQ(b, 0);
  464. obj_deinit(self);
  465. // obj_deinit(globals);
  466. EXPECT_EQ(pikaMemNow(), 0);
  467. }
  468. TEST(VM, RUN_AS) {
  469. char* pikaAsm =
  470. "B0\n"
  471. "0 RUN TinyObj\n"
  472. "0 OUT as\n"
  473. "B0\n"
  474. "0 RAS as\n"
  475. "B0\n"
  476. "0 NUM 1\n"
  477. "0 OUT x\n"
  478. "B0\n"
  479. "0 RAS $origin\n"
  480. "B0\n"
  481. "0 NUM 2\n"
  482. "0 OUT x\n";
  483. PikaObj* self = newRootObj("", New_PikaMain);
  484. pikaVM_runAsm(self, pikaAsm);
  485. PikaObj* as = obj_getObj(self, "as");
  486. int x_as_ = obj_getInt(as, "x");
  487. int x_as = obj_getInt(self, "as.x");
  488. int x_origin = obj_getInt(self, "x");
  489. EXPECT_EQ(x_as_, 1);
  490. EXPECT_EQ(x_as, 1);
  491. EXPECT_EQ(x_origin, 2);
  492. obj_deinit(self);
  493. // obj_deinit(globals);
  494. EXPECT_EQ(pikaMemNow(), 0);
  495. }
  496. TEST(VM, RUN_NEW) {
  497. char* pikaAsm =
  498. "B0\n"
  499. "0 RUN TinyObj\n"
  500. "0 OUT newObj\n"
  501. "0 NUM 1\n"
  502. "0 OUT newObj.x\n"
  503. "B0\n"
  504. "0 NEW newObj\n"
  505. "0 OUT outObj\n";
  506. PikaObj* self = newRootObj("", New_PikaMain);
  507. pikaVM_runAsm(self, pikaAsm);
  508. Arg* newObj = obj_getArg(self, "newObj");
  509. Arg* outObj = obj_getArg(self, "outObj");
  510. void* newObj_ptr = arg_getPtr(newObj);
  511. void* outObj_ptr = arg_getPtr(outObj);
  512. EXPECT_EQ(newObj_ptr, outObj_ptr);
  513. int x = obj_getInt(self, "outObj.x");
  514. ArgType newObj_type = arg_getType(newObj);
  515. ArgType outObj_type = arg_getType(outObj);
  516. EXPECT_EQ(x, 1);
  517. EXPECT_EQ(newObj_type, ARG_TYPE_OBJECT);
  518. EXPECT_EQ(outObj_type, ARG_TYPE_OBJECT);
  519. obj_deinit(self);
  520. // obj_deinit(globals);
  521. EXPECT_EQ(pikaMemNow(), 0);
  522. }
  523. TEST(VM, RUN_DEF_NEW) {
  524. char* pikaAsm =
  525. "B0\n"
  526. "0 DEF testnew()\n"
  527. "0 JMP 1\n"
  528. "B1\n"
  529. "0 RUN TinyObj\n"
  530. "0 OUT newObj\n"
  531. "B1\n"
  532. "0 NUM 1\n"
  533. "0 OUT newObj.x\n"
  534. "B1\n"
  535. "0 NEW newObj\n"
  536. "0 RET \n"
  537. "B1\n"
  538. "0 RET \n"
  539. "B0\n"
  540. "0 RUN testnew\n"
  541. "0 OUT outobj\n";
  542. PikaObj* self = newRootObj("", New_PikaMain);
  543. pikaVM_runAsm(self, pikaAsm);
  544. /* assert */
  545. PikaObj* outobj = obj_getObj(self, "outobj");
  546. int x = obj_getInt(outobj, "x");
  547. EXPECT_EQ(x, 1);
  548. /* deinit */
  549. obj_deinit(self);
  550. // obj_deinit(globals);
  551. EXPECT_EQ(pikaMemNow(), 0);
  552. }
  553. TEST(VM, class_x_1) {
  554. char* line =
  555. "class Test(TinyObj):\n"
  556. " x = 1\n"
  557. "test = Test()\n"
  558. "print('test.x: ' + str(test.x))\n"
  559. "\n";
  560. Args* buffs = New_strBuff();
  561. char* pikaAsm = Parser_linesToAsm(buffs, line);
  562. printf("%s", pikaAsm);
  563. PikaObj* self = newRootObj("", New_PikaMain);
  564. pikaVM_runAsm(self, pikaAsm);
  565. PikaObj* test = obj_getObj(self, "test");
  566. Arg* test_arg = obj_getArg(self, "test");
  567. ArgType test_arg_type = arg_getType(test_arg);
  568. EXPECT_EQ(test_arg_type, ARG_TYPE_OBJECT);
  569. int x = obj_getInt(test, "x");
  570. EXPECT_EQ(x, 1);
  571. obj_deinit(self);
  572. args_deinit(buffs);
  573. EXPECT_EQ(pikaMemNow(), 0);
  574. }
  575. TEST(VM, nag_a) {
  576. char* line =
  577. "a = 1\n"
  578. "print(-a)\n"
  579. "b = 0.5\n"
  580. "print(-b)\n";
  581. Args* buffs = New_strBuff();
  582. char* pikaAsm = Parser_linesToAsm(buffs, line);
  583. printf("%s", pikaAsm);
  584. PikaObj* self = newRootObj("", New_PikaMain);
  585. __platform_printf("BEGIN\r\n");
  586. pikaVM_runAsm(self, pikaAsm);
  587. /* assert */
  588. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  589. EXPECT_STREQ(log_buff[1], "-1\r\n");
  590. EXPECT_STREQ(log_buff[0], "-0.500000\r\n");
  591. /* deinit */
  592. obj_deinit(self);
  593. args_deinit(buffs);
  594. /* check mem */
  595. EXPECT_EQ(pikaMemNow(), 0);
  596. }
  597. TEST(InstructUnit, base) {
  598. __platform_printf("BEGIN\r\n");
  599. InstructUnit ins_unit;
  600. instructUnit_init(&ins_unit);
  601. instructUnit_setBlockDeepth(&ins_unit, 2);
  602. instructUnit_setIsNewLine(&ins_unit, 1);
  603. instructUnit_setInvokeDeepth(&ins_unit, 3);
  604. instructUnit_setInstruct(&ins_unit, (Instruct)4);
  605. instructUnit_setConstPoolIndex(&ins_unit, 12);
  606. EXPECT_EQ(instructUnit_getBlockDeepth(&ins_unit), 2);
  607. EXPECT_EQ(instructUnit_getIsNewLine(&ins_unit), 1);
  608. EXPECT_EQ(instructUnit_getInvokeDeepth(&ins_unit), 3);
  609. EXPECT_EQ(instructUnit_getInstruct(&ins_unit), 4);
  610. EXPECT_EQ(instructUnit_getConstPoolIndex(&ins_unit), 12);
  611. instructUnit_print(&ins_unit);
  612. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  613. EXPECT_STREQ(log_buff[1], "B2\r\n");
  614. EXPECT_STREQ(log_buff[0], "3 OUT #12\r\n");
  615. EXPECT_EQ(pikaMemNow(), 0);
  616. }
  617. TEST(ConstPool, get) {
  618. __platform_printf("BEGIN\r\n");
  619. ConstPool cp;
  620. constPool_init(&cp);
  621. char* next_content;
  622. uint16_t next_offset;
  623. next_offset = constPool_getLastOffset(&cp);
  624. constPool_append(&cp, "hello");
  625. next_content = constPool_getByOffset(&cp, next_offset);
  626. EXPECT_STREQ(next_content, "hello");
  627. next_offset = constPool_getLastOffset(&cp);
  628. constPool_append(&cp, "world");
  629. next_content = constPool_getByOffset(&cp, next_offset);
  630. EXPECT_STREQ(next_content, "world");
  631. char* first = constPool_getNow(&cp);
  632. char* second = constPool_getNext(&cp);
  633. char* third = constPool_getNext(&cp);
  634. char* forth = constPool_getNext(&cp);
  635. EXPECT_STREQ(first, "");
  636. EXPECT_STREQ(second, "hello");
  637. EXPECT_STREQ(third, "world");
  638. EXPECT_EQ((uintptr_t)forth, (uintptr_t)NULL);
  639. EXPECT_STREQ(constPool_getByIndex(&cp, 0), "");
  640. EXPECT_STREQ(constPool_getByIndex(&cp, 1), "hello");
  641. EXPECT_STREQ(constPool_getByIndex(&cp, 2), "world");
  642. EXPECT_EQ((uintptr_t)constPool_getByIndex(&cp, 3), (uintptr_t)NULL);
  643. constPool_print(&cp);
  644. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  645. EXPECT_STREQ(log_buff[1], "1: hello\r\n");
  646. EXPECT_STREQ(log_buff[0], "7: world\r\n");
  647. constPool_deinit(&cp);
  648. EXPECT_EQ(pikaMemNow(), 0);
  649. }
  650. TEST(InstructArray, set) {
  651. __platform_printf("BEGIN\r\n");
  652. InstructUnit ins_unit;
  653. instructUnit_init(&ins_unit);
  654. instructUnit_setBlockDeepth(&ins_unit, 2);
  655. instructUnit_setIsNewLine(&ins_unit, 1);
  656. instructUnit_setInvokeDeepth(&ins_unit, 3);
  657. instructUnit_setInstruct(&ins_unit, (Instruct)4);
  658. instructUnit_setConstPoolIndex(&ins_unit, 12);
  659. InstructArray ins_array;
  660. instructArray_init(&ins_array);
  661. instructArray_append(&ins_array, &ins_unit);
  662. instructArray_print(&ins_array);
  663. instructArray_deinit(&ins_array);
  664. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  665. EXPECT_STREQ(log_buff[1], "B2\r\n");
  666. EXPECT_STREQ(log_buff[0], "3 OUT #12\r\n");
  667. EXPECT_EQ(pikaMemNow(), 0);
  668. }
  669. TEST(VM, bytecode_jjcc) {
  670. char* line = "a = (1 + 1.1) * 3 - 2 /4.0";
  671. Args* buffs = New_strBuff();
  672. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  673. printf("%s", pikaAsm);
  674. ByteCodeFrame byte_frame;
  675. byteCodeFrame_init(&byte_frame);
  676. byteCodeFrame_appendFromAsm(&byte_frame, pikaAsm);
  677. byteCodeFrame_print(&byte_frame);
  678. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  679. // pikaVM_runAsm(self, pikaAsm);
  680. pikaVM_runByteCodeFrame(self, &byte_frame);
  681. float res = obj_getFloat(self, "a");
  682. obj_deinit(self);
  683. args_deinit(buffs);
  684. byteCodeFrame_deinit(&byte_frame);
  685. ASSERT_FLOAT_EQ(res, 5.8);
  686. EXPECT_EQ(pikaMemNow(), 0);
  687. }
  688. TEST(VM, WHILE_byte) {
  689. pikaMemInfo.heapUsedMax = 0;
  690. Args* buffs = New_strBuff();
  691. char* lines =(char *)
  692. "a = 1\n"
  693. "b = 0\n"
  694. "while a:\n"
  695. " b = 1\n"
  696. " a = 0\n"
  697. "\n";
  698. printf("%s", lines);
  699. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  700. printf("%s", pikaAsm);
  701. pikaMemInfo.heapUsedMax = 0;
  702. PikaObj* self = New_TinyObj(NULL);
  703. pikaVM_run(self, lines);
  704. EXPECT_EQ(obj_getInt(self, "a"), 0);
  705. EXPECT_EQ(obj_getInt(self, "b"), 1);
  706. args_deinit(buffs);
  707. obj_deinit(self);
  708. EXPECT_EQ(pikaMemNow(), 0);
  709. }
  710. TEST(VM, for_break_byte) {
  711. pikaMemInfo.heapUsedMax = 0;
  712. Args* buffs = New_strBuff();
  713. char* lines =(char *)
  714. "a = 0\n"
  715. "for i in range(0, 10):\n"
  716. " if i == 5:\n"
  717. " break\n"
  718. " a = a + i\n"
  719. "\n";
  720. printf("%s", lines);
  721. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  722. printf("%s", pikaAsm);
  723. pikaMemInfo.heapUsedMax = 0;
  724. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  725. pikaVM_run(self, lines);
  726. /* assert */
  727. int a = obj_getInt(self, "a");
  728. EXPECT_EQ(a, 10);
  729. /* deinit */
  730. args_deinit(buffs);
  731. obj_deinit(self);
  732. EXPECT_EQ(pikaMemNow(), 0);
  733. }
  734. TEST(VM, load_static_bytes) {
  735. const uint8_t bytes[] = {
  736. 0x08, 0x02, 0x00, 0x00, /* instruct array size */
  737. 0x00, 0x82, 0x01, 0x00, 0x00, 0x04, 0x0d, 0x00, 0x00, 0x83, 0x11, 0x00,
  738. 0x00, 0x04, 0x16, 0x00, 0x00, 0x85, 0x1b, 0x00, 0x00, 0x04, 0x1d, 0x00,
  739. 0x20, 0x85, 0x1b, 0x00, 0x20, 0x01, 0x0d, 0x00, 0x10, 0x02, 0x28, 0x00,
  740. 0x00, 0x02, 0x2e, 0x00, 0x00, 0x04, 0x33, 0x00, 0x00, 0x01, 0x37, 0x00,
  741. 0x00, 0x01, 0x3b, 0x00, 0x00, 0x01, 0x3f, 0x00, 0x00, 0x04, 0x43, 0x00,
  742. 0x00, 0x04, 0x4a, 0x00, 0x00, 0x04, 0x51, 0x00, 0x00, 0x82, 0x58, 0x00,
  743. 0x00, 0x04, 0x65, 0x00, 0x00, 0x0d, 0x65, 0x00, 0x00, 0x07, 0x67, 0x00,
  744. 0x11, 0x81, 0x0d, 0x00, 0x11, 0x05, 0x1b, 0x00, 0x01, 0x08, 0x69, 0x00,
  745. 0x01, 0x07, 0x6c, 0x00, 0x02, 0x8e, 0x6c, 0x00, 0x11, 0x81, 0x1d, 0x00,
  746. 0x11, 0x05, 0x1b, 0x00, 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00,
  747. 0x12, 0x81, 0x6e, 0x00, 0x12, 0x01, 0x65, 0x00, 0x02, 0x02, 0x76, 0x00,
  748. 0x02, 0x04, 0x16, 0x00, 0x02, 0x85, 0x6c, 0x00, 0x02, 0x04, 0x1d, 0x00,
  749. 0x01, 0x8b, 0x6c, 0x00, 0x11, 0x01, 0x1d, 0x00, 0x11, 0x05, 0x6c, 0x00,
  750. 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00, 0x12, 0x81, 0x16, 0x00,
  751. 0x12, 0x03, 0x7e, 0x00, 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00,
  752. 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00,
  753. 0x03, 0x04, 0x85, 0x00, 0x03, 0x82, 0x85, 0x00, 0x03, 0x85, 0x1b, 0x00,
  754. 0x03, 0x04, 0x1d, 0x00, 0x02, 0x8b, 0x6c, 0x00, 0x12, 0x01, 0x16, 0x00,
  755. 0x12, 0x03, 0x8a, 0x00, 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00,
  756. 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00,
  757. 0x03, 0x04, 0x8a, 0x00, 0x03, 0x85, 0x67, 0x00, 0x03, 0x04, 0x1d, 0x00,
  758. 0x02, 0x8b, 0x6c, 0x00, 0x12, 0x01, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00,
  759. 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x6e, 0x00,
  760. 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00, 0x03, 0x04, 0x8f, 0x00,
  761. 0x03, 0x85, 0x67, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x01, 0x8b, 0x6c, 0x00,
  762. 0x11, 0x01, 0x1d, 0x00, 0x11, 0x05, 0x67, 0x00, 0x01, 0x08, 0x69, 0x00,
  763. 0x01, 0x07, 0x6c, 0x00, 0x12, 0x81, 0x16, 0x00, 0x12, 0x03, 0x8a, 0x00,
  764. 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00, 0x03, 0x82, 0x8a, 0x00,
  765. 0x03, 0x07, 0x6c, 0x00, 0x14, 0x81, 0x6e, 0x00, 0x14, 0x01, 0x65, 0x00,
  766. 0x04, 0x02, 0x76, 0x00, 0x04, 0x04, 0x85, 0x00, 0x04, 0x82, 0x85, 0x00,
  767. 0x03, 0x85, 0x1b, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x02, 0x8b, 0x6c, 0x00,
  768. 0x12, 0x01, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00, 0x02, 0x08, 0x69, 0x00,
  769. 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00,
  770. 0x03, 0x02, 0x76, 0x00, 0x03, 0x04, 0x85, 0x00, 0x03, 0x85, 0x99, 0x00,
  771. 0x03, 0x04, 0x1d, 0x00, 0x01, 0x8b, 0x6c, 0x00, 0x11, 0x01, 0x1d, 0x00,
  772. 0x11, 0x05, 0x99, 0x00, 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00,
  773. 0x12, 0x81, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00, 0x02, 0x08, 0x69, 0x00,
  774. 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x9b, 0x00, 0x23, 0x01, 0x6e, 0x00,
  775. 0x23, 0x01, 0x65, 0x00, 0x13, 0x02, 0x76, 0x00, 0x03, 0x08, 0xa2, 0x00,
  776. 0x03, 0x07, 0x6c, 0x00, 0x04, 0x82, 0x85, 0x00, 0x14, 0x81, 0x6e, 0x00,
  777. 0x14, 0x01, 0x65, 0x00, 0x24, 0x01, 0x9b, 0x00, 0x24, 0x01, 0x8f, 0x00,
  778. 0x14, 0x08, 0xa4, 0x00, 0x14, 0x03, 0x6e, 0x00, 0x04, 0x02, 0xa6, 0x00,
  779. 0x03, 0x85, 0x1b, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x00, 0x86, 0xae, 0x00,
  780. 0x00, 0x8c, 0x33, 0x00, /* instruct array */
  781. 0xb1, 0x00, 0x00, 0x00, /* const pool size */
  782. 0x00, 0x5f, 0x5f, 0x63, 0x61, 0x6c, 0x6c, 0x73, 0x2e, 0x6c, 0x65, 0x6e,
  783. 0x00, 0x6c, 0x65, 0x6e, 0x00, 0x6e, 0x6f, 0x6e, 0x65, 0x00, 0x6d, 0x6f,
  784. 0x64, 0x65, 0x00, 0x30, 0x00, 0x69, 0x6e, 0x66, 0x6f, 0x5f, 0x69, 0x6e,
  785. 0x64, 0x65, 0x78, 0x00, 0x72, 0x61, 0x6e, 0x67, 0x65, 0x00, 0x69, 0x74,
  786. 0x65, 0x72, 0x00, 0x5f, 0x6c, 0x30, 0x00, 0x5f, 0x72, 0x31, 0x00, 0x5f,
  787. 0x72, 0x32, 0x00, 0x5f, 0x72, 0x33, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x61,
  788. 0x31, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x61, 0x32, 0x00, 0x5f, 0x6c, 0x30,
  789. 0x2e, 0x61, 0x33, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x5f, 0x5f, 0x6e, 0x65,
  790. 0x78, 0x74, 0x5f, 0x5f, 0x00, 0x69, 0x00, 0x32, 0x00, 0x3d, 0x3d, 0x00,
  791. 0x31, 0x00, 0x5f, 0x5f, 0x63, 0x61, 0x6c, 0x6c, 0x73, 0x00, 0x5f, 0x5f,
  792. 0x67, 0x65, 0x74, 0x5f, 0x5f, 0x00, 0x61, 0x6c, 0x77, 0x61, 0x79, 0x73,
  793. 0x00, 0x74, 0x6f, 0x64, 0x6f, 0x00, 0x77, 0x68, 0x65, 0x6e, 0x00, 0x70,
  794. 0x65, 0x72, 0x69, 0x6f, 0x64, 0x5f, 0x6d, 0x73, 0x00, 0x33, 0x00, 0x5f,
  795. 0x5f, 0x74, 0x69, 0x63, 0x6b, 0x00, 0x3e, 0x00, 0x2b, 0x00, 0x5f, 0x5f,
  796. 0x73, 0x65, 0x74, 0x5f, 0x5f, 0x00, 0x2d, 0x31, 0x00, /* const pool */
  797. };
  798. ByteCodeFrame bytecode_frame = {.const_pool =
  799. {
  800. .arg_buff = NULL,
  801. .content_offset_now = 0,
  802. .size = 0,
  803. .content_start = NULL,
  804. .output_redirect_fun = NULL,
  805. .output_f = NULL,
  806. },
  807. .instruct_array = {
  808. .arg_buff = NULL,
  809. .content_offset_now = 0,
  810. .size = 0,
  811. .content_start = NULL,
  812. .output_redirect_fun = NULL,
  813. .output_f = NULL,
  814. }};
  815. byteCodeFrame_loadByteCode(&bytecode_frame, (uint8_t*)bytes);
  816. byteCodeFrame_print(&bytecode_frame);
  817. EXPECT_EQ(instructArray_getSize(&(bytecode_frame.instruct_array)), 520);
  818. EXPECT_EQ(bytecode_frame.const_pool.size, 177);
  819. EXPECT_EQ(pikaMemNow(), 0);
  820. }
  821. TEST(VM, multi_jian) {
  822. pikaMemInfo.heapUsedMax = 0;
  823. Args* buffs = New_strBuff();
  824. char* lines = "a = (3-4) - 4\n";
  825. printf("%s", lines);
  826. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  827. printf("%s", pikaAsm);
  828. pikaMemInfo.heapUsedMax = 0;
  829. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  830. pikaVM_run(self, lines);
  831. /* assert */
  832. int a = obj_getInt(self, "a");
  833. EXPECT_EQ(a, -5);
  834. /* deinit */
  835. args_deinit(buffs);
  836. obj_deinit(self);
  837. EXPECT_EQ(pikaMemNow(), 0);
  838. }
  839. TEST(VM, _3_p_3) {
  840. char* line = "-3+3";
  841. Args* buffs = New_strBuff();
  842. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  843. printf("%s", pikaAsm);
  844. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  845. __platform_printf("BEGIN\r\n");
  846. pikaVM_runAsm(self, pikaAsm);
  847. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  848. EXPECT_STREQ(log_buff[0], "0\r\n");
  849. obj_deinit(self);
  850. args_deinit(buffs);
  851. EXPECT_EQ(pikaMemNow(), 0);
  852. }
  853. TEST(VM, _3_3) {
  854. char* line = "-3-3";
  855. Args* buffs = New_strBuff();
  856. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  857. printf("%s", pikaAsm);
  858. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  859. __platform_printf("BEGIN\r\n");
  860. pikaVM_runAsm(self, pikaAsm);
  861. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  862. EXPECT_STREQ(log_buff[0], "-6\r\n");
  863. obj_deinit(self);
  864. args_deinit(buffs);
  865. EXPECT_EQ(pikaMemNow(), 0);
  866. }
  867. TEST(VM, a_3) {
  868. char* line = "a-3";
  869. Args* buffs = New_strBuff();
  870. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  871. printf("%s", pikaAsm);
  872. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  873. __platform_printf("BEGIN\r\n");
  874. pikaVM_runAsm(self, pikaAsm);
  875. EXPECT_STREQ(log_buff[3], "BEGIN\r\n");
  876. EXPECT_STREQ(log_buff[2], "NameError: name 'a' is not defined\r\n");
  877. obj_deinit(self);
  878. args_deinit(buffs);
  879. EXPECT_EQ(pikaMemNow(), 0);
  880. }
  881. TEST(VM, hex_str) {
  882. char* line = "a = '\\x33\\x35'";
  883. Args* buffs = New_strBuff();
  884. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  885. printf("%s", pikaAsm);
  886. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  887. __platform_printf("BEGIN\r\n");
  888. pikaVM_runAsm(self, pikaAsm);
  889. /* collect */
  890. char* a = obj_getStr(self, "a");
  891. /* assert */
  892. EXPECT_STREQ(a, "35");
  893. /* deinit */
  894. obj_deinit(self);
  895. args_deinit(buffs);
  896. EXPECT_EQ(pikaMemNow(), 0);
  897. }
  898. TEST(VM, hex_bytes) {
  899. char* line =
  900. "a = b'\\x03\\x05'\n"
  901. "a\n";
  902. Args* buffs = New_strBuff();
  903. char* pikaAsm = Parser_linesToAsm(buffs, line);
  904. printf("%s", pikaAsm);
  905. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  906. __platform_printf("BEGIN\r\n");
  907. pikaVM_runAsm(self, pikaAsm);
  908. /* collect */
  909. /* assert */
  910. EXPECT_STREQ(log_buff[0], "b'\\x03\\x05'\r\n");
  911. /* deinit */
  912. obj_deinit(self);
  913. args_deinit(buffs);
  914. EXPECT_EQ(pikaMemNow(), 0);
  915. }
  916. TEST(VM, bytes_equ) {
  917. char* line =
  918. "a = b'\\x03\\x05'\n"
  919. "b = b'\\x03\\x05'\n"
  920. "if a == b:\n"
  921. " c = 1\n"
  922. "b = b'\\x02\\x05'\n"
  923. "if a != b:\n"
  924. " d = 1\n";
  925. Args* buffs = New_strBuff();
  926. char* pikaAsm = Parser_linesToAsm(buffs, line);
  927. printf("%s", pikaAsm);
  928. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  929. __platform_printf("BEGIN\r\n");
  930. pikaVM_runAsm(self, pikaAsm);
  931. /* collect */
  932. /* assert */
  933. int c = obj_getInt(self, "c");
  934. int d = obj_getInt(self, "d");
  935. EXPECT_EQ(c, 1);
  936. EXPECT_EQ(d, 1);
  937. /* deinit */
  938. obj_deinit(self);
  939. args_deinit(buffs);
  940. EXPECT_EQ(pikaMemNow(), 0);
  941. }
  942. #if PIKA_BUILTIN_STRUCT_ENABLE
  943. TEST(VM, tuple_literal) {
  944. char* line = "(1,2,3)\n";
  945. Args* buffs = New_strBuff();
  946. char* pikaAsm = Parser_linesToAsm(buffs, line);
  947. printf("%s", pikaAsm);
  948. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  949. __platform_printf("BEGIN\r\n");
  950. pikaVM_runAsm(self, pikaAsm);
  951. /* collect */
  952. /* assert */
  953. EXPECT_STREQ(log_buff[0], "(1, 2, 3)\r\n");
  954. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  955. /* deinit */
  956. obj_deinit(self);
  957. args_deinit(buffs);
  958. EXPECT_EQ(pikaMemNow(), 0);
  959. }
  960. #endif
  961. TEST(VM, dvd_opt) {
  962. char* line = "a = 10%4\n";
  963. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  964. obj_run(self, line);
  965. /* collect */
  966. int a = obj_getInt(self, "a");
  967. /* assert */
  968. EXPECT_EQ(a, 2);
  969. /* deinit */
  970. obj_deinit(self);
  971. EXPECT_EQ(pikaMemNow(), 0);
  972. }
  973. TEST(VM, del_) {
  974. char* line =
  975. "a = 1\n"
  976. "print(a)\n"
  977. "del a\n"
  978. "print(a)\n";
  979. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  980. obj_run(self, line);
  981. /* collect */
  982. /* assert */
  983. EXPECT_EQ(obj_isArgExist(self, "a"), 0);
  984. /* deinit */
  985. obj_deinit(self);
  986. EXPECT_EQ(pikaMemNow(), 0);
  987. }
  988. TEST(VM, is) {
  989. char* line =
  990. "a = 1\n"
  991. "b = a\n"
  992. "res1 = a is b\n"
  993. "res2 = 1 is 1\n"
  994. "c = TinyObj()\n"
  995. "d = c\n"
  996. "res3 = c is d\n"
  997. "res4 = c is None\n";
  998. ;
  999. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1000. obj_run(self, line);
  1001. /* collect */
  1002. /* assert */
  1003. EXPECT_EQ(obj_getInt(self, "res1"), 1);
  1004. EXPECT_EQ(obj_getInt(self, "res2"), 1);
  1005. EXPECT_EQ(obj_getInt(self, "res3"), 1);
  1006. EXPECT_EQ(obj_getInt(self, "res4"), 0);
  1007. /* deinit */
  1008. obj_deinit(self);
  1009. EXPECT_EQ(pikaMemNow(), 0);
  1010. }
  1011. TEST(VM, in) {
  1012. char* line =
  1013. "res1 = 'a' in 'aaa'\n"
  1014. "res2 = 'aaa' in 'a'\n";
  1015. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1016. obj_run(self, line);
  1017. /* collect */
  1018. /* assert */
  1019. EXPECT_EQ(obj_getInt(self, "res1"), 1);
  1020. EXPECT_EQ(obj_getInt(self, "res2"), 0);
  1021. /* deinit */
  1022. obj_deinit(self);
  1023. EXPECT_EQ(pikaMemNow(), 0);
  1024. }
  1025. #if !PIKA_NANO_ENABLE
  1026. TEST(VM, assert) {
  1027. char* line =
  1028. "assert True\n"
  1029. "assert 1 == 1, 'testparser'\n"
  1030. "res1 = 0\n"
  1031. "try:\n"
  1032. " assert False\n"
  1033. "except:\n"
  1034. " res1 = 1\n";
  1035. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1036. obj_run(self, line);
  1037. /* collect */
  1038. int res1 = obj_getInt(self, "res1");
  1039. /* assert */
  1040. EXPECT_EQ(res1, 1);
  1041. /* deinit */
  1042. obj_deinit(self);
  1043. EXPECT_EQ(pikaMemNow(), 0);
  1044. }
  1045. #endif
  1046. TEST(VM, issue_I5LHJG) {
  1047. char* line =
  1048. "res = (a, 2, 3)\n"
  1049. "[a, 2, 3]\n"
  1050. "{'x': a, 'y': 2, 'z': 3}\n";
  1051. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1052. obj_run(self, line);
  1053. /* collect */
  1054. /* assert */
  1055. EXPECT_EQ(obj_isArgExist(self, "res"), 0);
  1056. /* deinit */
  1057. obj_deinit(self);
  1058. EXPECT_EQ(pikaMemNow(), 0);
  1059. }
  1060. TEST(VM, vars_runtime) {
  1061. char* line =
  1062. "def testvars(a, *b):\n"
  1063. " sum = 0\n"
  1064. " for i in b:\n"
  1065. " sum += i\n"
  1066. " return a * sum\n"
  1067. "res = testvars(6, 2, 3, 4, 5)\n";
  1068. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1069. obj_run(self, line);
  1070. /* collect */
  1071. int res = obj_getInt(self, "res");
  1072. /* assert */
  1073. EXPECT_EQ(res, 84);
  1074. /* deinit */
  1075. obj_deinit(self);
  1076. EXPECT_EQ(pikaMemNow(), 0);
  1077. }
  1078. #if PIKA_BUILTIN_STRUCT_ENABLE
  1079. TEST(VM, list_add) {
  1080. char* line = "[1, 2, 3] + [4, 5, 6]";
  1081. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1082. obj_run(self, line);
  1083. /* collect */
  1084. /* assert */
  1085. EXPECT_STREQ(log_buff[0], "[1, 2, 3, 4, 5, 6]\r\n");
  1086. /* deinit */
  1087. obj_deinit(self);
  1088. EXPECT_EQ(pikaMemNow(), 0);
  1089. }
  1090. #endif
  1091. TEST(VM, science_num) {
  1092. char* line =
  1093. "a = 1.0e-3\n"
  1094. "b = 2e-5\n"
  1095. "c = -3e-5\n"
  1096. "d = 0.4e2\n";
  1097. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1098. obj_run(self, line);
  1099. /* collect */
  1100. double a = obj_getFloat(self, "a");
  1101. double b = obj_getFloat(self, "b");
  1102. double c = obj_getFloat(self, "c");
  1103. double d = obj_getFloat(self, "d");
  1104. /* assert */
  1105. #if PIKA_FLOAT_TYPE_DOUBLE
  1106. EXPECT_DOUBLE_EQ(a, 1.0e-3);
  1107. EXPECT_DOUBLE_EQ(b, 2.0e-5);
  1108. EXPECT_DOUBLE_EQ(c, -3e-5);
  1109. EXPECT_DOUBLE_EQ(d, 0.4e2);
  1110. #else
  1111. EXPECT_FLOAT_EQ(a, 1.0e-3);
  1112. EXPECT_FLOAT_EQ(b, 2.0e-5);
  1113. EXPECT_FLOAT_EQ(c, -3e-5);
  1114. EXPECT_FLOAT_EQ(d, 0.4e2);
  1115. #endif
  1116. /* deinit */
  1117. obj_deinit(self);
  1118. EXPECT_EQ(pikaMemNow(), 0);
  1119. }
  1120. TEST(VM, issue_I5OJQB) {
  1121. char* line = "s = '\\\\'";
  1122. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1123. obj_run(self, line);
  1124. /* collect */
  1125. char* s = obj_getStr(self, "s");
  1126. /* assert */
  1127. EXPECT_STREQ(s, "\\");
  1128. /* deinit */
  1129. obj_deinit(self);
  1130. EXPECT_EQ(pikaMemNow(), 0);
  1131. }
  1132. #if !PIKA_NANO_ENABLE
  1133. TEST(vm, keyword_2) {
  1134. char* line =
  1135. "def test(a, b):\n"
  1136. " print(__kwargs['a'], __kwargs['b'])\n"
  1137. "test(a=1, b= 2)";
  1138. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1139. obj_run(self, line);
  1140. /* collect */
  1141. /* assert */
  1142. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1143. /* deinit */
  1144. obj_deinit(self);
  1145. EXPECT_EQ(pikaMemNow(), 0);
  1146. }
  1147. TEST(vm, keyword_3) {
  1148. char* line =
  1149. "def test(**keys):\n"
  1150. " print(keys['a'], keys['b'])\n"
  1151. "test(a=1, b= 2)";
  1152. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1153. obj_run(self, line);
  1154. /* collect */
  1155. /* assert */
  1156. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1157. /* deinit */
  1158. obj_deinit(self);
  1159. EXPECT_EQ(pikaMemNow(), 0);
  1160. }
  1161. TEST(vm, keyword_4) {
  1162. char* line =
  1163. "def test(a, b, **keys):\n"
  1164. " print(keys['a'], keys['b'], a, b)\n"
  1165. "test(3, 4, a=1, b= 2)";
  1166. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1167. obj_run(self, line);
  1168. /* collect */
  1169. /* assert */
  1170. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  1171. /* deinit */
  1172. obj_deinit(self);
  1173. EXPECT_EQ(pikaMemNow(), 0);
  1174. }
  1175. TEST(vm, class_keyword) {
  1176. char* line =
  1177. "class Test:\n"
  1178. " def __init__(self, a, b = 1, c =2):\n"
  1179. " print(a, b, c)\n"
  1180. "t = Test(0, b = 3)\n";
  1181. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1182. obj_run(self, line);
  1183. /* collect */
  1184. /* assert */
  1185. EXPECT_STREQ(log_buff[0], "0 3 2\r\n");
  1186. /* deinit */
  1187. obj_deinit(self);
  1188. EXPECT_EQ(pikaMemNow(), 0);
  1189. }
  1190. TEST(vm, class_keyword_mqtt) {
  1191. char* line =
  1192. "class MQTT:\n"
  1193. " def __init__(self,\n"
  1194. " ip: str,\n"
  1195. " port=1883,\n"
  1196. " clinetID='mac',\n"
  1197. " username='',\n"
  1198. " password='',\n"
  1199. " version='3.1.1',\n"
  1200. " ca='',\n"
  1201. " keepalive=60):\n"
  1202. " print('ip:', ip)\n"
  1203. " print('port:', port)\n"
  1204. "c = MQTT('test')\n";
  1205. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1206. obj_run(self, line);
  1207. /* collect */
  1208. /* assert */
  1209. EXPECT_STREQ(log_buff[1], "ip: test\r\n");
  1210. EXPECT_STREQ(log_buff[0], "port: 1883\r\n");
  1211. /* deinit */
  1212. obj_deinit(self);
  1213. EXPECT_EQ(pikaMemNow(), 0);
  1214. }
  1215. TEST(vm, vars_keyward) {
  1216. /* init */
  1217. pikaMemInfo.heapUsedMax = 0;
  1218. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1219. extern unsigned char pikaModules_py_a[];
  1220. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1221. /* run */
  1222. __platform_printf("BEGIN\r\n");
  1223. pikaVM_runSingleFile(pikaMain, "../../examples/BuiltIn/function.py");
  1224. /* collect */
  1225. /* assert */
  1226. EXPECT_STREQ(log_buff[1], "1 2 3 4 5 10\r\n");
  1227. EXPECT_STREQ(log_buff[0], "1 2 3 4 5 12\r\n");
  1228. /* deinit */
  1229. obj_deinit(pikaMain);
  1230. EXPECT_EQ(pikaMemNow(), 0);
  1231. }
  1232. TEST(vm, cb_1) {
  1233. /* init */
  1234. pikaMemInfo.heapUsedMax = 0;
  1235. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1236. extern unsigned char pikaModules_py_a[];
  1237. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1238. /* run */
  1239. __platform_printf("BEGIN\r\n");
  1240. pikaVM_runSingleFile(pikaMain, "../../examples/Callback/test1.py");
  1241. /* collect */
  1242. /* assert */
  1243. EXPECT_STREQ(log_buff[3], "__init__\r\n");
  1244. EXPECT_STREQ(log_buff[1], "a\r\n");
  1245. EXPECT_STREQ(log_buff[0], "b\r\n");
  1246. /* deinit */
  1247. obj_deinit(pikaMain);
  1248. EXPECT_EQ(pikaMemNow(), 0);
  1249. }
  1250. TEST(vm, cb_2) {
  1251. /* init */
  1252. pikaMemInfo.heapUsedMax = 0;
  1253. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1254. extern unsigned char pikaModules_py_a[];
  1255. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1256. /* run */
  1257. __platform_printf("BEGIN\r\n");
  1258. pikaVM_runSingleFile(pikaMain, "../../examples/Callback/test2.py");
  1259. /* collect */
  1260. /* assert */
  1261. EXPECT_STREQ(log_buff[4], "__init__\r\n");
  1262. EXPECT_STREQ(log_buff[2], "a\r\n");
  1263. EXPECT_STREQ(log_buff[1], "b\r\n");
  1264. EXPECT_STREQ(log_buff[0], "ppp\r\n");
  1265. /* deinit */
  1266. obj_deinit(pikaMain);
  1267. EXPECT_EQ(pikaMemNow(), 0);
  1268. }
  1269. #endif
  1270. #if !PIKA_NANO_ENABLE
  1271. TEST(vm, default_no_input) {
  1272. /* init */
  1273. pikaMemInfo.heapUsedMax = 0;
  1274. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1275. extern unsigned char pikaModules_py_a[];
  1276. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1277. /* run */
  1278. __platform_printf("BEGIN\r\n");
  1279. obj_run(pikaMain,
  1280. "def test(a = 1):\n"
  1281. " print(a)\n"
  1282. "test()");
  1283. /* collect */
  1284. /* assert */
  1285. EXPECT_STREQ(log_buff[0], "1\r\n");
  1286. /* deinit */
  1287. obj_deinit(pikaMain);
  1288. EXPECT_EQ(pikaMemNow(), 0);
  1289. }
  1290. TEST(vm, default_1) {
  1291. /* init */
  1292. pikaMemInfo.heapUsedMax = 0;
  1293. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1294. extern unsigned char pikaModules_py_a[];
  1295. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1296. /* run */
  1297. __platform_printf("BEGIN\r\n");
  1298. obj_run(pikaMain,
  1299. "def test(a = 1):\n"
  1300. " print(a)\n"
  1301. "test(a = 2)");
  1302. /* collect */
  1303. /* assert */
  1304. EXPECT_STREQ(log_buff[0], "2\r\n");
  1305. /* deinit */
  1306. obj_deinit(pikaMain);
  1307. EXPECT_EQ(pikaMemNow(), 0);
  1308. }
  1309. TEST(vm, default_2) {
  1310. /* init */
  1311. pikaMemInfo.heapUsedMax = 0;
  1312. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1313. extern unsigned char pikaModules_py_a[];
  1314. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1315. /* run */
  1316. __platform_printf("BEGIN\r\n");
  1317. obj_run(pikaMain,
  1318. "def test(a, b = 1):\n"
  1319. " print(a, b)\n"
  1320. "test(1, b = 2)");
  1321. /* collect */
  1322. /* assert */
  1323. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1324. /* deinit */
  1325. obj_deinit(pikaMain);
  1326. EXPECT_EQ(pikaMemNow(), 0);
  1327. }
  1328. TEST(vm, default_3) {
  1329. /* init */
  1330. pikaMemInfo.heapUsedMax = 0;
  1331. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1332. extern unsigned char pikaModules_py_a[];
  1333. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1334. /* run */
  1335. __platform_printf("BEGIN\r\n");
  1336. obj_run(pikaMain,
  1337. "def myprint(end='\\n', *var):\n"
  1338. " for v in var:\n"
  1339. " print(v, end = end)\n"
  1340. "myprint(1, 2)\n"
  1341. "myprint(1, 2, end=';\\n')\n");
  1342. /* collect */
  1343. /* assert */
  1344. EXPECT_STREQ(log_buff[3], "1\n");
  1345. EXPECT_STREQ(log_buff[2], "2\n");
  1346. EXPECT_STREQ(log_buff[1], "1;\n");
  1347. EXPECT_STREQ(log_buff[0], "2;\n");
  1348. /* deinit */
  1349. obj_deinit(pikaMain);
  1350. EXPECT_EQ(pikaMemNow(), 0);
  1351. }
  1352. TEST(vm, default_4) {
  1353. /* init */
  1354. pikaMemInfo.heapUsedMax = 0;
  1355. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1356. extern unsigned char pikaModules_py_a[];
  1357. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1358. /* run */
  1359. __platform_printf("BEGIN\r\n");
  1360. obj_run(pikaMain,
  1361. "def test(end=5, *var):\n"
  1362. " res = end\n"
  1363. " print('end', end)\n"
  1364. " for v in var:\n"
  1365. " res += v\n"
  1366. " return res\n"
  1367. "test(1, 2, 3)\n"
  1368. "test(1, 2, 3, end=6)\n");
  1369. /* collect */
  1370. /* assert */
  1371. EXPECT_STREQ(log_buff[3], "end 5\r\n");
  1372. EXPECT_STREQ(log_buff[2], "11\r\n");
  1373. EXPECT_STREQ(log_buff[1], "end 6\r\n");
  1374. EXPECT_STREQ(log_buff[0], "12\r\n");
  1375. /* deinit */
  1376. obj_deinit(pikaMain);
  1377. EXPECT_EQ(pikaMemNow(), 0);
  1378. }
  1379. TEST(vm, default_no_kw) {
  1380. /* init */
  1381. pikaMemInfo.heapUsedMax = 0;
  1382. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1383. extern unsigned char pikaModules_py_a[];
  1384. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1385. /* run */
  1386. __platform_printf("BEGIN\r\n");
  1387. obj_run(pikaMain,
  1388. "def test(a, b=3):\n"
  1389. " print(a ,b)\n"
  1390. "test(1, 2)\n"
  1391. "test(1, b=2)\n"
  1392. "test(1)\n");
  1393. /* collect */
  1394. /* assert */
  1395. EXPECT_STREQ(log_buff[2], "1 2\r\n");
  1396. EXPECT_STREQ(log_buff[1], "1 2\r\n");
  1397. EXPECT_STREQ(log_buff[0], "1 3\r\n");
  1398. /* deinit */
  1399. obj_deinit(pikaMain);
  1400. EXPECT_EQ(pikaMemNow(), 0);
  1401. }
  1402. #endif
  1403. TEST(vm, none) {
  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, "GTestTask.test(1, None)");
  1412. /* collect */
  1413. /* assert */
  1414. EXPECT_STREQ(log_buff[0], "GTestTask_test: 1, 0\r\n");
  1415. /* deinit */
  1416. obj_deinit(pikaMain);
  1417. EXPECT_EQ(pikaMemNow(), 0);
  1418. }
  1419. #if !PIKA_NANO_ENABLE
  1420. TEST(vm, super_) {
  1421. /* init */
  1422. pikaMemInfo.heapUsedMax = 0;
  1423. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1424. extern unsigned char pikaModules_py_a[];
  1425. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1426. /* run */
  1427. __platform_printf("BEGIN\r\n");
  1428. obj_run(pikaMain,
  1429. "class test:\n"
  1430. " def __init__(self):\n"
  1431. " print('in test init')\n"
  1432. "class test2(test):\n"
  1433. " def __init__(self):\n"
  1434. " super().__init__()\n"
  1435. " print('in test2 init')\n"
  1436. "t = test2()");
  1437. /* collect */
  1438. /* assert */
  1439. EXPECT_STREQ(log_buff[0], "in test2 init\r\n");
  1440. EXPECT_STREQ(log_buff[1], "in test init\r\n");
  1441. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1442. /* deinit */
  1443. obj_deinit(pikaMain);
  1444. EXPECT_EQ(pikaMemNow(), 0);
  1445. }
  1446. TEST(vm, super_val) {
  1447. /* init */
  1448. pikaMemInfo.heapUsedMax = 0;
  1449. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1450. extern unsigned char pikaModules_py_a[];
  1451. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1452. /* run */
  1453. __platform_printf("BEGIN\r\n");
  1454. obj_run(pikaMain,
  1455. "class test:\n"
  1456. " def __init__(self):\n"
  1457. " self.a = 1\n"
  1458. " self.b = 2\n"
  1459. "class test2(test):\n"
  1460. " def __init__(self):\n"
  1461. " super().__init__()\n"
  1462. " self.c = 3\n"
  1463. " self.d = 4\n"
  1464. "t = test2()\n"
  1465. "print(t.a, t.b, t.c, t.d)\n");
  1466. /* collect */
  1467. /* assert */
  1468. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1469. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  1470. /* deinit */
  1471. obj_deinit(pikaMain);
  1472. EXPECT_EQ(pikaMemNow(), 0);
  1473. }
  1474. TEST(vm, super_val_) {
  1475. /* init */
  1476. pikaMemInfo.heapUsedMax = 0;
  1477. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1478. extern unsigned char pikaModules_py_a[];
  1479. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1480. /* run */
  1481. __platform_printf("BEGIN\r\n");
  1482. obj_run(pikaMain,
  1483. "class test:\n"
  1484. " def test(self, a):\n"
  1485. " self.a = a\n"
  1486. "class test2(test):\n"
  1487. " def test(self, a):\n"
  1488. " super().test(str(a))\n"
  1489. "t1 = test()\n"
  1490. "t2 = test2()\n"
  1491. "t1.test(1)\n"
  1492. "t2.test(1)\n");
  1493. /* collect */
  1494. int t1_a = obj_getInt(pikaMain, "t1.a");
  1495. char* t2_a = obj_getStr(pikaMain, "t2.a");
  1496. /* assert */
  1497. EXPECT_EQ(t1_a, 1);
  1498. EXPECT_STREQ(t2_a, "1");
  1499. /* deinit */
  1500. obj_deinit(pikaMain);
  1501. EXPECT_EQ(pikaMemNow(), 0);
  1502. }
  1503. #endif
  1504. #if !PIKA_NANO_ENABLE
  1505. TEST(vm, multi_return) {
  1506. /* init */
  1507. pikaMemInfo.heapUsedMax = 0;
  1508. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1509. extern unsigned char pikaModules_py_a[];
  1510. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1511. /* run */
  1512. __platform_printf("BEGIN\r\n");
  1513. obj_run(pikaMain, "a,b = (1,2)");
  1514. /* collect */
  1515. int a = obj_getInt(pikaMain, "a");
  1516. int b = obj_getInt(pikaMain, "b");
  1517. /* assert */
  1518. EXPECT_EQ(a, 1);
  1519. EXPECT_EQ(b, 2);
  1520. /* deinit */
  1521. obj_deinit(pikaMain);
  1522. EXPECT_EQ(pikaMemNow(), 0);
  1523. }
  1524. #endif
  1525. #if !PIKA_NANO_ENABLE
  1526. TEST(vm, multi_return_fn) {
  1527. /* init */
  1528. pikaMemInfo.heapUsedMax = 0;
  1529. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1530. extern unsigned char pikaModules_py_a[];
  1531. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1532. /* run */
  1533. __platform_printf("BEGIN\r\n");
  1534. obj_run(pikaMain,
  1535. "def test():\n"
  1536. " return 1,2\n"
  1537. "a,b = test()");
  1538. /* collect */
  1539. int a = obj_getInt(pikaMain, "a");
  1540. int b = obj_getInt(pikaMain, "b");
  1541. /* assert */
  1542. EXPECT_EQ(a, 1);
  1543. EXPECT_EQ(b, 2);
  1544. /* deinit */
  1545. obj_deinit(pikaMain);
  1546. EXPECT_EQ(pikaMemNow(), 0);
  1547. }
  1548. #endif
  1549. TEST(vm, range_1) {
  1550. /* init */
  1551. pikaMemInfo.heapUsedMax = 0;
  1552. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1553. extern unsigned char pikaModules_py_a[];
  1554. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1555. /* run */
  1556. __platform_printf("BEGIN\r\n");
  1557. obj_run(pikaMain,
  1558. "sum = 0\n"
  1559. "for i in range(10):\n"
  1560. " sum += i\n");
  1561. /* collect */
  1562. int sum = obj_getInt(pikaMain, "sum");
  1563. /* assert */
  1564. EXPECT_EQ(sum, 45);
  1565. /* deinit */
  1566. obj_deinit(pikaMain);
  1567. EXPECT_EQ(pikaMemNow(), 0);
  1568. }
  1569. TEST(vm, rang_3) {
  1570. /* init */
  1571. pikaMemInfo.heapUsedMax = 0;
  1572. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1573. extern unsigned char pikaModules_py_a[];
  1574. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1575. /* run */
  1576. __platform_printf("BEGIN\r\n");
  1577. obj_run(pikaMain,
  1578. "sum = 0\n"
  1579. "for i in range(-5, 20, 2):\n"
  1580. " sum += i\n");
  1581. /* collect */
  1582. int sum = obj_getInt(pikaMain, "sum");
  1583. /* assert */
  1584. EXPECT_EQ(sum, 91);
  1585. /* deinit */
  1586. obj_deinit(pikaMain);
  1587. EXPECT_EQ(pikaMemNow(), 0);
  1588. }
  1589. TEST(vm, test64) {
  1590. /* init */
  1591. pikaMemInfo.heapUsedMax = 0;
  1592. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1593. extern unsigned char pikaModules_py_a[];
  1594. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1595. /* run */
  1596. __platform_printf("BEGIN\r\n");
  1597. obj_run(pikaMain,
  1598. "import GTestTask\n"
  1599. "res = GTestTask.test64(4294967295, 20)\n");
  1600. /* collect */
  1601. int64_t res = obj_getInt(pikaMain, "res");
  1602. /* assert */
  1603. EXPECT_EQ(res, 4294967295 * 20);
  1604. /* deinit */
  1605. obj_deinit(pikaMain);
  1606. EXPECT_EQ(pikaMemNow(), 0);
  1607. }
  1608. #if !PIKA_NANO_ENABLE
  1609. TEST(vm, exit) {
  1610. /* init */
  1611. pikaMemInfo.heapUsedMax = 0;
  1612. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1613. extern unsigned char pikaModules_py_a[];
  1614. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1615. /* run */
  1616. __platform_printf("BEGIN\r\n");
  1617. obj_run(pikaMain,
  1618. "i = 0\n"
  1619. "while True:\n"
  1620. " i += 1\n"
  1621. " print(i)\n"
  1622. " if i == 10:\n"
  1623. " exit()\n");
  1624. /* collect */
  1625. int i = obj_getInt(pikaMain, "i");
  1626. /* assert */
  1627. EXPECT_EQ(i, 10);
  1628. /* deinit */
  1629. obj_deinit(pikaMain);
  1630. EXPECT_EQ(pikaMemNow(), 0);
  1631. }
  1632. TEST(vm, exit_fn) {
  1633. /* init */
  1634. pikaMemInfo.heapUsedMax = 0;
  1635. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1636. extern unsigned char pikaModules_py_a[];
  1637. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1638. /* run */
  1639. __platform_printf("BEGIN\r\n");
  1640. obj_run(pikaMain,
  1641. "i = 0\n"
  1642. "PikaStdLib.MemChecker.now()\n"
  1643. "def test():\n"
  1644. " global i\n"
  1645. " while True:\n"
  1646. " i += 1\n"
  1647. " print(i)\n"
  1648. " if i == 10:\n"
  1649. " exit()\n"
  1650. "while True:\n"
  1651. " test()\n");
  1652. obj_run(pikaMain, "PikaStdLib.MemChecker.now()\n");
  1653. /* collect */
  1654. int i = obj_getInt(pikaMain, "i");
  1655. /* assert */
  1656. EXPECT_EQ(i, 10);
  1657. /* deinit */
  1658. obj_deinit(pikaMain);
  1659. EXPECT_EQ(pikaMemNow(), 0);
  1660. }
  1661. #endif
  1662. #if PIKA_INSTRUCT_HOOK_ENABLE
  1663. typedef void (*hook_func)(void);
  1664. extern volatile int g_hook_cnt;
  1665. extern volatile hook_func g_hook_func;
  1666. extern "C" {
  1667. volatile int g_hook_cnt_triggle;
  1668. void hook_func_exit_issue_1(void) {
  1669. if (g_hook_cnt >= g_hook_cnt_triggle) {
  1670. pks_vm_exit();
  1671. }
  1672. }
  1673. void __gtest_hook_default_(void) {
  1674. return;
  1675. }
  1676. }
  1677. void _vm_exit_fn_issue_1_item(int hook_cnt) {
  1678. g_hook_cnt_triggle = hook_cnt;
  1679. /* init */
  1680. pikaMemInfo.heapUsedMax = 0;
  1681. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1682. extern unsigned char pikaModules_py_a[];
  1683. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1684. /* run */
  1685. __platform_printf("BEGIN\r\n");
  1686. obj_run(pikaMain,
  1687. "PikaStdLib.MemChecker.now()\n"
  1688. "def task1():\n"
  1689. " print('task1 hello')\n"
  1690. "\n"
  1691. "def task2():\n"
  1692. " print('task2 hello')\n"
  1693. "\n"
  1694. "def fake_runtask():\n"
  1695. " i = 0\n"
  1696. " while True:\n"
  1697. " i += 1\n"
  1698. " print(i)\n"
  1699. " task1()\n"
  1700. " task2()\n"
  1701. "fake_runtask()\n"
  1702. "fake_runtask()\n"
  1703. "\n");
  1704. obj_run(pikaMain, "PikaStdLib.MemChecker.now()\n");
  1705. /* collect */
  1706. /* assert */
  1707. obj_deinit(pikaMain);
  1708. EXPECT_EQ(pikaMemNow(), 0);
  1709. }
  1710. TEST(vm, exit_fn_issue_1) {
  1711. g_hook_func = hook_func_exit_issue_1;
  1712. for (int i = 0; i < 50; i++) {
  1713. g_hook_cnt = 0;
  1714. _vm_exit_fn_issue_1_item(i);
  1715. }
  1716. g_hook_func = __gtest_hook_default_;
  1717. }
  1718. #endif
  1719. TEST(vm, pass_) {
  1720. /* init */
  1721. pikaMemInfo.heapUsedMax = 0;
  1722. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1723. extern unsigned char pikaModules_py_a[];
  1724. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1725. /* run */
  1726. __platform_printf("BEGIN\r\n");
  1727. obj_run(pikaMain,
  1728. "pass\n"
  1729. "def testpass():\n"
  1730. " pass\n"
  1731. "for i in range(10):\n"
  1732. " pass\n"
  1733. "testpass()\n"
  1734. "print('after pass')\n");
  1735. /* collect */
  1736. /* assert */
  1737. EXPECT_STREQ(log_buff[0], "after pass\r\n");
  1738. /* deinit */
  1739. obj_deinit(pikaMain);
  1740. EXPECT_EQ(pikaMemNow(), 0);
  1741. }
  1742. TEST(vm, test64_hex) {
  1743. /* init */
  1744. pikaMemInfo.heapUsedMax = 0;
  1745. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1746. extern unsigned char pikaModules_py_a[];
  1747. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1748. /* run */
  1749. __platform_printf("BEGIN\r\n");
  1750. obj_run(pikaMain,
  1751. "import GTestTask\n"
  1752. "res = GTestTask.test64(0xffffffff, 20)\n");
  1753. /* collect */
  1754. int64_t res = obj_getInt(pikaMain, "res");
  1755. /* assert */
  1756. EXPECT_EQ(res, 4294967295 * 20);
  1757. /* deinit */
  1758. obj_deinit(pikaMain);
  1759. EXPECT_EQ(pikaMemNow(), 0);
  1760. }
  1761. TEST(vm, test64_hex_print) {
  1762. /* init */
  1763. pikaMemInfo.heapUsedMax = 0;
  1764. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1765. extern unsigned char pikaModules_py_a[];
  1766. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1767. /* run */
  1768. __platform_printf("BEGIN\r\n");
  1769. obj_run(pikaMain,
  1770. "0xffffffff\n"
  1771. "print(0xffffffff)\n");
  1772. /* collect */
  1773. /* assert */
  1774. EXPECT_STREQ(log_buff[0], "4294967295\r\n");
  1775. EXPECT_STREQ(log_buff[1], "4294967295\r\n");
  1776. /* deinit */
  1777. obj_deinit(pikaMain);
  1778. EXPECT_EQ(pikaMemNow(), 0);
  1779. }
  1780. #if !PIKA_NANO_ENABLE
  1781. TEST(vm, call_dict_err) {
  1782. /* init */
  1783. pikaMemInfo.heapUsedMax = 0;
  1784. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1785. extern unsigned char pikaModules_py_a[];
  1786. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1787. /* run */
  1788. __platform_printf("BEGIN\r\n");
  1789. obj_run(pikaMain,
  1790. "a = {'a': 1}\n"
  1791. "a('a')\n");
  1792. /* collect */
  1793. /* assert */
  1794. EXPECT_STREQ(log_buff[4], "TypeError: 'a' object is not callable\r\n");
  1795. /* deinit */
  1796. obj_deinit(pikaMain);
  1797. EXPECT_EQ(pikaMemNow(), 0);
  1798. }
  1799. TEST(vm, getattribute) {
  1800. /* init */
  1801. pikaMemInfo.heapUsedMax = 0;
  1802. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1803. extern unsigned char pikaModules_py_a[];
  1804. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1805. /* run */
  1806. __platform_printf("BEGIN\r\n");
  1807. obj_run(pikaMain,
  1808. "class test:\n"
  1809. " def __getattribute__(self, name):\n"
  1810. " return name\n"
  1811. "t = test()\n"
  1812. "t.a\n");
  1813. /* collect */
  1814. /* assert */
  1815. EXPECT_STREQ(log_buff[0], "'a'\r\n");
  1816. /* deinit */
  1817. obj_deinit(pikaMain);
  1818. EXPECT_EQ(pikaMemNow(), 0);
  1819. }
  1820. TEST(vm, getattr) {
  1821. /* init */
  1822. pikaMemInfo.heapUsedMax = 0;
  1823. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1824. extern unsigned char pikaModules_py_a[];
  1825. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1826. /* run */
  1827. __platform_printf("BEGIN\r\n");
  1828. obj_run(pikaMain,
  1829. "class test:\n"
  1830. " a = 1\n"
  1831. " def __getattr__(self, name):\n"
  1832. " return name\n"
  1833. "t = test()\n"
  1834. "t.a\n"
  1835. "t.b\n");
  1836. /* collect */
  1837. /* assert */
  1838. EXPECT_STREQ(log_buff[0], "'b'\r\n");
  1839. EXPECT_STREQ(log_buff[1], "1\r\n");
  1840. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1841. /* deinit */
  1842. obj_deinit(pikaMain);
  1843. EXPECT_EQ(pikaMemNow(), 0);
  1844. }
  1845. TEST(vm, setattr) {
  1846. /* init */
  1847. pikaMemInfo.heapUsedMax = 0;
  1848. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1849. extern unsigned char pikaModules_py_a[];
  1850. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1851. /* run */
  1852. __platform_printf("BEGIN\r\n");
  1853. obj_run(pikaMain,
  1854. "class test:\n"
  1855. " a = 1\n"
  1856. " def __setattr__(self, name, val):\n"
  1857. " print((name, val))\n"
  1858. "t = test()\n"
  1859. "t.a = 1\n"
  1860. "t.b = 'test'\n");
  1861. /* collect */
  1862. /* assert */
  1863. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1864. EXPECT_STREQ(log_buff[1], "('a', 1)\r\n");
  1865. EXPECT_STREQ(log_buff[0], "('b', 'test')\r\n");
  1866. /* deinit */
  1867. obj_deinit(pikaMain);
  1868. EXPECT_EQ(pikaMemNow(), 0);
  1869. }
  1870. TEST(vm, c_module_get_set_attr) {
  1871. /* init */
  1872. pikaMemInfo.heapUsedMax = 0;
  1873. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1874. extern unsigned char pikaModules_py_a[];
  1875. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1876. /* run */
  1877. __platform_printf("BEGIN\r\n");
  1878. obj_run(pikaMain,
  1879. "t = GTestTask.ProxyTest()\n"
  1880. "t.a\n"
  1881. "t.b\n"
  1882. "t.a = 'test1'\n"
  1883. "t.b = 'test2'\n");
  1884. /* collect */
  1885. /* assert */
  1886. EXPECT_STREQ(log_buff[3], "'a'\r\n");
  1887. EXPECT_STREQ(log_buff[2], "'b'\r\n");
  1888. EXPECT_STREQ(log_buff[1], "GTestTask_ProxyTest___setattr__: a, test1\r\n");
  1889. EXPECT_STREQ(log_buff[0], "GTestTask_ProxyTest___setattr__: b, test2\r\n");
  1890. /* deinit */
  1891. obj_deinit(pikaMain);
  1892. EXPECT_EQ(pikaMemNow(), 0);
  1893. }
  1894. TEST(vm, class_attr_ref) {
  1895. /* init */
  1896. pikaMemInfo.heapUsedMax = 0;
  1897. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1898. extern unsigned char pikaModules_py_a[];
  1899. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1900. /* run */
  1901. __platform_printf("BEGIN\r\n");
  1902. obj_run(pikaMain,
  1903. "class test:\n"
  1904. " a = 1\n"
  1905. " b = a\n"
  1906. "t = test()\n");
  1907. /* collect */
  1908. /* assert */
  1909. /* deinit */
  1910. obj_deinit(pikaMain);
  1911. EXPECT_EQ(pikaMemNow(), 0);
  1912. }
  1913. TEST(vm, getattr_native) {
  1914. /* init */
  1915. pikaMemInfo.heapUsedMax = 0;
  1916. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1917. extern unsigned char pikaModules_py_a[];
  1918. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1919. /* run */
  1920. __platform_printf("BEGIN\r\n");
  1921. obj_run(pikaMain,
  1922. "mem = PikaStdLib.MemChecker()\n"
  1923. "if hasattr(mem, 'max'):\n"
  1924. " max = getattr(mem, 'max')\n"
  1925. " max()\n");
  1926. /* collect */
  1927. /* assert */
  1928. /* deinit */
  1929. obj_deinit(pikaMain);
  1930. EXPECT_EQ(pikaMemNow(), 0);
  1931. }
  1932. TEST(vm, issue_dict_update) {
  1933. /* init */
  1934. pikaMemInfo.heapUsedMax = 0;
  1935. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1936. extern unsigned char pikaModules_py_a[];
  1937. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1938. /* run */
  1939. __platform_printf("BEGIN\r\n");
  1940. pikaVM_runSingleFile(pikaMain, "test/python/issue/issue_dict_update.py");
  1941. /* collect */
  1942. /* assert */
  1943. EXPECT_STREQ(obj_getStr(pikaMain, "inner"), "widget5");
  1944. /* deinit */
  1945. obj_deinit(pikaMain);
  1946. EXPECT_EQ(pikaMemNow(), 0);
  1947. }
  1948. TEST(vm, issue_big_dict_update) {
  1949. /* init */
  1950. pikaMemInfo.heapUsedMax = 0;
  1951. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1952. extern unsigned char pikaModules_py_a[];
  1953. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1954. /* run */
  1955. __platform_printf("BEGIN\r\n");
  1956. pikaVM_runSingleFile(pikaMain,
  1957. "test/python/issue/issue_big_dict_update.py");
  1958. /* collect */
  1959. /* assert */
  1960. /* deinit */
  1961. obj_deinit(pikaMain);
  1962. EXPECT_EQ(pikaMemNow(), 0);
  1963. }
  1964. #endif
  1965. TEST(vm, i_pp) {
  1966. /* init */
  1967. pikaMemInfo.heapUsedMax = 0;
  1968. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1969. extern unsigned char pikaModules_py_a[];
  1970. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1971. /* run */
  1972. __platform_printf("BEGIN\r\n");
  1973. /* clang-format off */
  1974. PIKA_PYTHON(
  1975. i = 0
  1976. while i < 10000:
  1977. i += 1
  1978. )
  1979. /* clang-format on */
  1980. const uint8_t bytes[] = {
  1981. 0x30, 0x00, 0x00, 0x00, /* instruct array size */
  1982. 0x00, 0x85, 0x01, 0x00, 0x00, 0x04, 0x03, 0x00, 0x10, 0x81, 0x03, 0x00,
  1983. 0x10, 0x05, 0x05, 0x00, 0x00, 0x08, 0x0b, 0x00, 0x00, 0x07, 0x0d, 0x00,
  1984. 0x11, 0x81, 0x03, 0x00, 0x21, 0x05, 0x0f, 0x00, 0x11, 0x02, 0x00, 0x00,
  1985. 0x01, 0x08, 0x11, 0x00, 0x01, 0x04, 0x03, 0x00, 0x00, 0x86, 0x13, 0x00,
  1986. /* instruct array */
  1987. 0x16, 0x00, 0x00, 0x00, /* const pool size */
  1988. 0x00, 0x30, 0x00, 0x69, 0x00, 0x31, 0x30, 0x30, 0x30, 0x30, 0x00, 0x3c,
  1989. 0x00, 0x32, 0x00, 0x31, 0x00, 0x2b, 0x00, 0x2d, 0x31,
  1990. 0x00, /* const pool */
  1991. };
  1992. pikaVM_runByteCode(pikaMain, (uint8_t*)bytes);
  1993. EXPECT_EQ(obj_getInt(pikaMain, "i"), 10000);
  1994. /* deinit */
  1995. obj_deinit(pikaMain);
  1996. EXPECT_EQ(pikaMemNow(), 0);
  1997. }
  1998. TEST(vm, benchmark) {
  1999. /* init */
  2000. pikaMemInfo.heapUsedMax = 0;
  2001. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2002. extern unsigned char pikaModules_py_a[];
  2003. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2004. /* run */
  2005. __platform_printf("BEGIN\r\n");
  2006. /* clang-format off */
  2007. PIKA_PYTHON(
  2008. num = 0
  2009. i = 2
  2010. while i < 1000:
  2011. is_prime = 1
  2012. j = 2
  2013. while j < i:
  2014. if i%j==0 :
  2015. is_prime = 0
  2016. break
  2017. j += 1
  2018. if is_prime:
  2019. num = num + i
  2020. i += 1
  2021. )
  2022. /* clang-format on */
  2023. const uint8_t bytes[] = {
  2024. 0xac, 0x00, 0x00, 0x00, /* instruct array size */
  2025. 0x00, 0x85, 0x01, 0x00, 0x00, 0x04, 0x03, 0x00, 0x00, 0x85, 0x07, 0x00,
  2026. 0x00, 0x04, 0x09, 0x00, 0x10, 0x81, 0x09, 0x00, 0x10, 0x05, 0x0b, 0x00,
  2027. 0x00, 0x08, 0x10, 0x00, 0x00, 0x07, 0x07, 0x00, 0x01, 0x85, 0x12, 0x00,
  2028. 0x01, 0x04, 0x14, 0x00, 0x01, 0x85, 0x07, 0x00, 0x01, 0x04, 0x1d, 0x00,
  2029. 0x11, 0x81, 0x1d, 0x00, 0x11, 0x01, 0x09, 0x00, 0x01, 0x08, 0x10, 0x00,
  2030. 0x01, 0x07, 0x07, 0x00, 0x22, 0x81, 0x09, 0x00, 0x22, 0x01, 0x1d, 0x00,
  2031. 0x12, 0x08, 0x1f, 0x00, 0x12, 0x05, 0x01, 0x00, 0x02, 0x08, 0x21, 0x00,
  2032. 0x02, 0x07, 0x12, 0x00, 0x03, 0x85, 0x01, 0x00, 0x03, 0x04, 0x14, 0x00,
  2033. 0x03, 0x8e, 0x00, 0x00, 0x12, 0x81, 0x1d, 0x00, 0x22, 0x05, 0x12, 0x00,
  2034. 0x12, 0x02, 0x00, 0x00, 0x02, 0x08, 0x24, 0x00, 0x02, 0x04, 0x1d, 0x00,
  2035. 0x01, 0x86, 0x26, 0x00, 0x01, 0x81, 0x14, 0x00, 0x01, 0x07, 0x12, 0x00,
  2036. 0x12, 0x81, 0x03, 0x00, 0x12, 0x01, 0x09, 0x00, 0x02, 0x08, 0x24, 0x00,
  2037. 0x02, 0x04, 0x03, 0x00, 0x11, 0x81, 0x09, 0x00, 0x21, 0x05, 0x12, 0x00,
  2038. 0x11, 0x02, 0x00, 0x00, 0x01, 0x08, 0x24, 0x00, 0x01, 0x04, 0x09, 0x00,
  2039. 0x00, 0x86, 0x26, 0x00, /* instruct array */
  2040. 0x29, 0x00, 0x00, 0x00, /* const pool size */
  2041. 0x00, 0x30, 0x00, 0x6e, 0x75, 0x6d, 0x00, 0x32, 0x00, 0x69, 0x00, 0x31,
  2042. 0x30, 0x30, 0x30, 0x00, 0x3c, 0x00, 0x31, 0x00, 0x69, 0x73, 0x5f, 0x70,
  2043. 0x72, 0x69, 0x6d, 0x65, 0x00, 0x6a, 0x00, 0x25, 0x00, 0x3d, 0x3d, 0x00,
  2044. 0x2b, 0x00, 0x2d, 0x31, 0x00, /* const pool */
  2045. };
  2046. pikaVM_runByteCode(pikaMain, (uint8_t*)bytes);
  2047. /* deinit */
  2048. obj_deinit(pikaMain);
  2049. EXPECT_EQ(pikaMemNow(), 0);
  2050. }
  2051. TEST(VM, print_None) {
  2052. char* line = "print(None)\n";
  2053. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  2054. obj_run(self, line);
  2055. /* collect */
  2056. /* assert */
  2057. EXPECT_STREQ(log_buff[0], "None\r\n");
  2058. /* deinit */
  2059. obj_deinit(self);
  2060. EXPECT_EQ(pikaMemNow(), 0);
  2061. }
  2062. TEST_END