VM-test.cpp 84 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874
  1. #include "test_common.h"
  2. TEST_START
  3. TEST(VM, num1) {
  4. char* line = "1";
  5. Args* buffs = New_strBuff();
  6. char* pikaAsm = pika_lines2Asm(buffs, line);
  7. printf("%s", pikaAsm);
  8. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  9. __platform_printf("BEGIN\r\n");
  10. pikaVM_runAsm(self, pikaAsm);
  11. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  12. EXPECT_STREQ(log_buff[0], "1\r\n");
  13. obj_deinit(self);
  14. args_deinit(buffs);
  15. EXPECT_EQ(pikaMemNow(), 0);
  16. }
  17. TEST(VM, a_1) {
  18. char* line = "a = 1";
  19. Args* buffs = New_strBuff();
  20. char* pikaAsm = pika_lines2Asm(buffs, line);
  21. printf("%s", pikaAsm);
  22. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  23. pikaVM_runAsm(self, pikaAsm);
  24. ASSERT_EQ(args_getInt(self->list, "a"), 1);
  25. obj_deinit(self);
  26. args_deinit(buffs);
  27. EXPECT_EQ(pikaMemNow(), 0);
  28. }
  29. TEST(VM, a_1d1) {
  30. char* line = "a = 1.1";
  31. Args* buffs = New_strBuff();
  32. char* pikaAsm = pika_lines2Asm(buffs, line);
  33. printf("%s", pikaAsm);
  34. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  35. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  36. ASSERT_FLOAT_EQ(args_getFloat(globals->list, "a"), 1.1);
  37. obj_deinit(self);
  38. // obj_deinit(globals);
  39. args_deinit(buffs);
  40. EXPECT_EQ(pikaMemNow(), 0);
  41. }
  42. TEST(VM, str_xy) {
  43. char* line = "a = 'xy'";
  44. Args* buffs = New_strBuff();
  45. char* pikaAsm = pika_lines2Asm(buffs, line);
  46. printf("%s", pikaAsm);
  47. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  48. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  49. ASSERT_STREQ(args_getStr(globals->list, "a"), "xy");
  50. obj_deinit(self);
  51. // obj_deinit(globals);
  52. args_deinit(buffs);
  53. EXPECT_EQ(pikaMemNow(), 0);
  54. }
  55. TEST(VM, str_xy_space) {
  56. char* line = "a = 'xy '";
  57. Args* buffs = New_strBuff();
  58. char* pikaAsm = pika_lines2Asm(buffs, line);
  59. printf("%s", pikaAsm);
  60. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  61. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  62. ASSERT_STREQ(args_getStr(globals->list, "a"), "xy ");
  63. obj_deinit(self);
  64. args_deinit(buffs);
  65. // obj_deinit(globals);
  66. EXPECT_EQ(pikaMemNow(), 0);
  67. }
  68. TEST(VM, ref_a_b) {
  69. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  70. Args* buffs = New_strBuff();
  71. pikaVM_runAsm(self, pika_lines2Asm(buffs, "a = 'xy '"));
  72. args_deinit(buffs);
  73. obj_deinit(self);
  74. // obj_deinit(globals);
  75. EXPECT_EQ(pikaMemNow(), 0);
  76. }
  77. TEST(VM, Run_add) {
  78. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  79. Args* buffs = New_strBuff();
  80. char* pikaAsm = pika_lines2Asm(buffs, "a = plusInt(1,2)");
  81. __platform_printf("%s", pikaAsm);
  82. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  83. args_deinit(buffs);
  84. int a = args_getInt(globals->list, "a");
  85. ASSERT_EQ(a, 3);
  86. obj_deinit(self);
  87. // obj_deinit(globals);
  88. EXPECT_EQ(pikaMemNow(), 0);
  89. }
  90. TEST(VM, Run_add_multy) {
  91. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  92. Args* buffs = New_strBuff();
  93. pikaVM_runAsm(self, pika_lines2Asm(buffs, "b = 2"));
  94. args_deinit(buffs);
  95. obj_deinit(self);
  96. // obj_deinit(globals);
  97. EXPECT_EQ(pikaMemNow(), 0);
  98. }
  99. TEST(VM, Run_add_1_2_3) {
  100. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  101. Args* buffs = New_strBuff();
  102. char* pikaAsm = pika_lines2Asm(buffs, "a = plusInt(1, plusInt(2,3) )");
  103. __platform_printf("%s", pikaAsm);
  104. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  105. args_deinit(buffs);
  106. int a = args_getInt(globals->list, "a");
  107. ASSERT_EQ(a, 6);
  108. obj_deinit(self);
  109. // obj_deinit(globals);
  110. EXPECT_EQ(pikaMemNow(), 0);
  111. }
  112. extern PikaMemInfo g_PikaMemInfo;
  113. TEST(VM, WHILE) {
  114. g_PikaMemInfo.heapUsedMax = 0;
  115. Args* buffs = New_strBuff();
  116. char* lines =(char *)
  117. "a = 1\n"
  118. "b = 0\n"
  119. "while a:\n"
  120. " b = 1\n"
  121. " a = 0\n"
  122. "\n";
  123. printf("%s", lines);
  124. char* pikaAsm = pika_lines2Asm(buffs, lines);
  125. printf("%s", pikaAsm);
  126. g_PikaMemInfo.heapUsedMax = 0;
  127. PikaObj* self = newRootObj("root", New_TinyObj);
  128. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  129. EXPECT_EQ(args_getInt(globals->list, "a"), 0);
  130. EXPECT_EQ(args_getInt(globals->list, "b"), 1);
  131. // obj_deinit(globals);
  132. args_deinit(buffs);
  133. obj_deinit(self);
  134. EXPECT_EQ(pikaMemNow(), 0);
  135. }
  136. TEST(VM, a_1_1) {
  137. char* line = "a = 1 + 1";
  138. Args* buffs = New_strBuff();
  139. char* pikaAsm = pika_lines2Asm(buffs, line);
  140. printf("%s", pikaAsm);
  141. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  142. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  143. int res = args_getInt(globals->list, "a");
  144. obj_deinit(self);
  145. args_deinit(buffs);
  146. ASSERT_EQ(res, 2);
  147. EXPECT_EQ(pikaMemNow(), 0);
  148. }
  149. TEST(VM, a_1_1d1) {
  150. char* line = "a = 1 + 1.1";
  151. Args* buffs = New_strBuff();
  152. char* pikaAsm = pika_lines2Asm(buffs, line);
  153. printf("%s", pikaAsm);
  154. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  155. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  156. float res = args_getFloat(globals->list, "a");
  157. obj_deinit(self);
  158. // obj_deinit(globals);
  159. args_deinit(buffs);
  160. ASSERT_FLOAT_EQ(res, 2.1);
  161. EXPECT_EQ(pikaMemNow(), 0);
  162. }
  163. TEST(VM, a_jjcc) {
  164. char* line = "a = (1 + 1.1) * 3 - 2 /4.0";
  165. printf("%s\r\n", line);
  166. Args* buffs = New_strBuff();
  167. char* pikaAsm = pika_lines2Asm(buffs, line);
  168. printf("%s", pikaAsm);
  169. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  170. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  171. float res = args_getFloat(globals->list, "a");
  172. obj_deinit(self);
  173. args_deinit(buffs);
  174. // obj_deinit(globals);
  175. ASSERT_FLOAT_EQ(res, 5.8);
  176. EXPECT_EQ(pikaMemNow(), 0);
  177. }
  178. TEST(VM, while_a_1to10) {
  179. char* line =
  180. "a = 0\n"
  181. "while a < 10:\n"
  182. " a += 1\n"
  183. " print(a)\n"
  184. "\n";
  185. Args* buffs = New_strBuff();
  186. char* pikaAsm = pika_lines2Asm(buffs, line);
  187. printf("%s", pikaAsm);
  188. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  189. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  190. int res = args_getInt(globals->list, "a");
  191. obj_deinit(self);
  192. args_deinit(buffs);
  193. // obj_deinit(globals);
  194. ASSERT_FLOAT_EQ(res, 10);
  195. EXPECT_EQ(pikaMemNow(), 0);
  196. }
  197. TEST(VM, mem_x) {
  198. char* line =
  199. "mem = PikaStdLib.MemChecker()\n"
  200. "mem.x = 1\n"
  201. "print(mem.x)\n"
  202. "mem.x = 2\n"
  203. "print(mem.x)\n"
  204. "\n";
  205. Args* buffs = New_strBuff();
  206. char* pikaAsm = pika_lines2Asm(buffs, line);
  207. printf("%s", pikaAsm);
  208. PikaObj* self = newRootObj("", New_PikaMain);
  209. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  210. int res = obj_getInt(globals, "mem.x");
  211. EXPECT_EQ(2, res);
  212. obj_deinit(self);
  213. // obj_deinit(globals);
  214. args_deinit(buffs);
  215. EXPECT_EQ(pikaMemNow(), 0);
  216. }
  217. TEST(VM, RET_instruct) {
  218. char* pikaAsm =
  219. "B1\n"
  220. "0 NUM 13\n"
  221. "0 RET \n"
  222. "0 NUM 2\n"
  223. "0 RET \n"
  224. "B0\n";
  225. PikaObj* self = New_TinyObj(NULL);
  226. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  227. Arg* return_arg = args_getArg(globals->list, "@rt");
  228. int num = arg_getInt(return_arg);
  229. EXPECT_EQ(num, 13);
  230. obj_deinit(self);
  231. // obj_deinit(globals);
  232. EXPECT_EQ(pikaMemNow(), 0);
  233. }
  234. TEST(VM, RUN_DEF) {
  235. char* pikaAsm =
  236. "B0\n"
  237. "0 DEF test()\n"
  238. "0 JMP 1\n"
  239. "B1\n"
  240. "0 NUM 1\n"
  241. "0 RET \n"
  242. "B0\n"
  243. "0 RUN test\n"
  244. "0 OUT a\n";
  245. PikaObj* self = New_TinyObj(NULL);
  246. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  247. int num = obj_getInt(globals, "a");
  248. EXPECT_EQ(num, 1);
  249. obj_deinit(self);
  250. // obj_deinit(globals);
  251. EXPECT_EQ(pikaMemNow(), 0);
  252. }
  253. TEST(VM, RUN_global) {
  254. char* pikaAsm =
  255. "B0\n"
  256. "0 NUM 1\n"
  257. "0 OUT a\n"
  258. "B0\n"
  259. "0 DEF test()\n"
  260. "0 JMP 1\n"
  261. "B1\n"
  262. "0 REF a\n"
  263. "0 RET \n"
  264. "B0\n"
  265. "0 RUN test\n"
  266. "0 OUT b\n";
  267. PikaObj* self = New_TinyObj(NULL);
  268. pikaVM_runAsm(self, pikaAsm);
  269. int a = obj_getInt(self, "a");
  270. int b = obj_getInt(self, "b");
  271. EXPECT_EQ(a, 1);
  272. EXPECT_EQ(b, 1);
  273. obj_deinit(self);
  274. // obj_deinit(globals);
  275. EXPECT_EQ(pikaMemNow(), 0);
  276. }
  277. TEST(VM, RUN_local_b) {
  278. char* pikaAsm =
  279. "B0\n"
  280. "0 NUM 1\n"
  281. "0 OUT a\n"
  282. "B0\n"
  283. "0 DEF test()\n"
  284. "0 JMP 1\n"
  285. "B1\n"
  286. "0 REF a\n"
  287. "0 OUT b\n"
  288. "1 REF b\n"
  289. "1 REF a\n"
  290. "0 OPT +\n"
  291. "0 RET \n"
  292. "B0\n"
  293. "0 RUN test\n"
  294. "0 OUT c\n";
  295. PikaObj* self = newRootObj("", New_BaseObj);
  296. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  297. int a = obj_getInt(globals, "a");
  298. int b = obj_getInt(globals, "b");
  299. int c = obj_getInt(globals, "c");
  300. EXPECT_EQ(a, 1);
  301. /* b is local, should not be exist in globals */
  302. EXPECT_EQ(b, _PIKA_INT_ERR);
  303. EXPECT_EQ(c, 2);
  304. obj_deinit(self);
  305. // obj_deinit(globals);
  306. EXPECT_EQ(pikaMemNow(), 0);
  307. }
  308. TEST(VM, RUN_DEF_add) {
  309. char* pikaAsm =
  310. "B0\n"
  311. "0 DEF add(a,b)\n"
  312. "0 JMP 1\n"
  313. "B1\n"
  314. "1 REF b\n"
  315. "1 REF a\n"
  316. "0 OPT +\n"
  317. "0 RET \n"
  318. "B0\n"
  319. "1 NUM 1\n"
  320. "1 NUM 2\n"
  321. "0 RUN add\n"
  322. "0 OUT c\n";
  323. PikaObj* self = newRootObj("", New_BaseObj);
  324. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  325. int a = obj_getInt(globals, "a");
  326. int b = obj_getInt(globals, "b");
  327. int c = obj_getInt(globals, "c");
  328. /* a is local, should not be exist in globals */
  329. EXPECT_EQ(a, _PIKA_INT_ERR);
  330. /* b is local, should not be exist in globals */
  331. EXPECT_EQ(b, _PIKA_INT_ERR);
  332. EXPECT_EQ(c, 3);
  333. obj_deinit(self);
  334. // obj_deinit(globals);
  335. EXPECT_EQ(pikaMemNow(), 0);
  336. }
  337. TEST(VM, run_def_add) {
  338. char* line =
  339. "def add(a, b):\n"
  340. " return a + b\n"
  341. "c = add(1,3)\n"
  342. "\n";
  343. Args* buffs = New_strBuff();
  344. char* pikaAsm = pika_lines2Asm(buffs, line);
  345. printf("%s", pikaAsm);
  346. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  347. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  348. int c = args_getInt(globals->list, "c");
  349. obj_deinit(self);
  350. args_deinit(buffs);
  351. // obj_deinit(globals);
  352. ASSERT_FLOAT_EQ(c, 4);
  353. EXPECT_EQ(pikaMemNow(), 0);
  354. }
  355. TEST(VM, equ) {
  356. char* line =
  357. "if -1 == -1:\n"
  358. " a = 1\n"
  359. "\n";
  360. Args* buffs = New_strBuff();
  361. char* pikaAsm = pika_lines2Asm(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, "a");
  366. obj_deinit(self);
  367. args_deinit(buffs);
  368. // obj_deinit(globals);
  369. ASSERT_FLOAT_EQ(c, 1);
  370. EXPECT_EQ(pikaMemNow(), 0);
  371. }
  372. TEST(VM, if_elif) {
  373. char* line =
  374. "a = 2\n"
  375. "b = 0\n"
  376. "if a > 1:\n"
  377. " b = 1\n"
  378. "elif a > 0:\n"
  379. " b = 2\n"
  380. "\n";
  381. Args* buffs = New_strBuff();
  382. char* pikaAsm = pika_lines2Asm(buffs, line);
  383. printf("%s", pikaAsm);
  384. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  385. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  386. int b = args_getInt(globals->list, "b");
  387. obj_deinit(self);
  388. args_deinit(buffs);
  389. ASSERT_FLOAT_EQ(b, 1);
  390. EXPECT_EQ(pikaMemNow(), 0);
  391. }
  392. TEST(VM, if_else) {
  393. char* line =
  394. "a = 0\n"
  395. "b = 0\n"
  396. "if a > 1:\n"
  397. " b = 1\n"
  398. "else:\n"
  399. " b = 2\n"
  400. "\n";
  401. Args* buffs = New_strBuff();
  402. char* pikaAsm = pika_lines2Asm(buffs, line);
  403. printf("%s", pikaAsm);
  404. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  405. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  406. int b = args_getInt(globals->list, "b");
  407. obj_deinit(self);
  408. args_deinit(buffs);
  409. // obj_deinit(globals);
  410. ASSERT_FLOAT_EQ(b, 2);
  411. EXPECT_EQ(pikaMemNow(), 0);
  412. }
  413. TEST(VM, if_else_) {
  414. char* line =
  415. "a = 2\n"
  416. "b = 0\n"
  417. "if a > 1:\n"
  418. " b = 1\n"
  419. "else:\n"
  420. " b = 2\n"
  421. "\n";
  422. Args* buffs = New_strBuff();
  423. char* pikaAsm = pika_lines2Asm(buffs, line);
  424. printf("%s", pikaAsm);
  425. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  426. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  427. int b = args_getInt(globals->list, "b");
  428. obj_deinit(self);
  429. args_deinit(buffs);
  430. // obj_deinit(globals);
  431. ASSERT_FLOAT_EQ(b, 1);
  432. EXPECT_EQ(pikaMemNow(), 0);
  433. }
  434. TEST(VM, EST) {
  435. char* pikaAsm =
  436. "B0\n"
  437. "0 EST a\n"
  438. "0 OUT b\n";
  439. PikaObj* self = newRootObj("", New_BaseObj);
  440. pikaVM_runAsm(self, pikaAsm);
  441. int a = obj_getInt(self, "a");
  442. int b = obj_getInt(self, "b");
  443. /* a is local, should not be exist in globals */
  444. EXPECT_EQ(a, _PIKA_INT_ERR);
  445. /* b is local, should not be exist in globals */
  446. EXPECT_EQ(b, 0);
  447. obj_deinit(self);
  448. // obj_deinit(globals);
  449. EXPECT_EQ(pikaMemNow(), 0);
  450. }
  451. TEST(VM, RUN_AS) {
  452. char* pikaAsm =
  453. "B0\n"
  454. "0 RUN TinyObj\n"
  455. "0 OUT as\n"
  456. "B0\n"
  457. "0 RAS as\n"
  458. "B0\n"
  459. "0 NUM 1\n"
  460. "0 OUT x\n"
  461. "B0\n"
  462. "0 RAS $origin\n"
  463. "B0\n"
  464. "0 NUM 2\n"
  465. "0 OUT x\n";
  466. PikaObj* self = newRootObj("", New_PikaMain);
  467. pikaVM_runAsm(self, pikaAsm);
  468. PikaObj* as = obj_getObj(self, "as");
  469. int x_as_ = obj_getInt(as, "x");
  470. int x_as = obj_getInt(self, "as.x");
  471. int x_origin = obj_getInt(self, "x");
  472. EXPECT_EQ(x_as_, 1);
  473. EXPECT_EQ(x_as, 1);
  474. EXPECT_EQ(x_origin, 2);
  475. obj_deinit(self);
  476. // obj_deinit(globals);
  477. EXPECT_EQ(pikaMemNow(), 0);
  478. }
  479. TEST(VM, RUN_NEW) {
  480. char* pikaAsm =
  481. "B0\n"
  482. "0 RUN TinyObj\n"
  483. "0 OUT newObj\n"
  484. "0 NUM 1\n"
  485. "0 OUT newObj.x\n"
  486. "B0\n"
  487. "0 NEW newObj\n"
  488. "0 OUT outObj\n";
  489. PikaObj* self = newRootObj("", New_PikaMain);
  490. pikaVM_runAsm(self, pikaAsm);
  491. Arg* newObj = obj_getArg(self, "newObj");
  492. Arg* outObj = obj_getArg(self, "outObj");
  493. void* newObj_ptr = arg_getPtr(newObj);
  494. void* outObj_ptr = arg_getPtr(outObj);
  495. EXPECT_EQ(newObj_ptr, outObj_ptr);
  496. int x = obj_getInt(self, "outObj.x");
  497. ArgType newObj_type = arg_getType(newObj);
  498. ArgType outObj_type = arg_getType(outObj);
  499. EXPECT_EQ(x, 1);
  500. EXPECT_EQ(newObj_type, ARG_TYPE_OBJECT);
  501. EXPECT_EQ(outObj_type, ARG_TYPE_OBJECT);
  502. obj_deinit(self);
  503. // obj_deinit(globals);
  504. EXPECT_EQ(pikaMemNow(), 0);
  505. }
  506. TEST(VM, RUN_DEF_NEW) {
  507. char* pikaAsm =
  508. "B0\n"
  509. "0 DEF testnew()\n"
  510. "0 JMP 1\n"
  511. "B1\n"
  512. "0 RUN TinyObj\n"
  513. "0 OUT newObj\n"
  514. "B1\n"
  515. "0 NUM 1\n"
  516. "0 OUT newObj.x\n"
  517. "B1\n"
  518. "0 NEW newObj\n"
  519. "0 RET \n"
  520. "B1\n"
  521. "0 RET \n"
  522. "B0\n"
  523. "0 RUN testnew\n"
  524. "0 OUT outobj\n";
  525. PikaObj* self = newRootObj("", New_PikaMain);
  526. pikaVM_runAsm(self, pikaAsm);
  527. /* assert */
  528. PikaObj* outobj = obj_getObj(self, "outobj");
  529. int x = obj_getInt(outobj, "x");
  530. EXPECT_EQ(x, 1);
  531. /* deinit */
  532. obj_deinit(self);
  533. // obj_deinit(globals);
  534. EXPECT_EQ(pikaMemNow(), 0);
  535. }
  536. TEST(VM, class_x_1) {
  537. char* line =
  538. "class Test(TinyObj):\n"
  539. " x = 1\n"
  540. "test = Test()\n"
  541. "print('test.x: ' + str(test.x))\n"
  542. "\n";
  543. Args* buffs = New_strBuff();
  544. char* pikaAsm = pika_lines2Asm(buffs, line);
  545. printf("%s", pikaAsm);
  546. PikaObj* self = newRootObj("", New_PikaMain);
  547. pikaVM_runAsm(self, pikaAsm);
  548. PikaObj* test = obj_getObj(self, "test");
  549. Arg* test_arg = obj_getArg(self, "test");
  550. ArgType test_arg_type = arg_getType(test_arg);
  551. EXPECT_EQ(test_arg_type, ARG_TYPE_OBJECT);
  552. int x = obj_getInt(test, "x");
  553. EXPECT_EQ(x, 1);
  554. obj_deinit(self);
  555. args_deinit(buffs);
  556. EXPECT_EQ(pikaMemNow(), 0);
  557. }
  558. TEST(VM, nag_a) {
  559. char* line =
  560. "a = 1\n"
  561. "print(-a)\n"
  562. "b = 0.5\n"
  563. "print(-b)\n";
  564. Args* buffs = New_strBuff();
  565. char* pikaAsm = pika_lines2Asm(buffs, line);
  566. printf("%s", pikaAsm);
  567. PikaObj* self = newRootObj("", New_PikaMain);
  568. __platform_printf("BEGIN\r\n");
  569. pikaVM_runAsm(self, pikaAsm);
  570. /* assert */
  571. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  572. EXPECT_STREQ(log_buff[1], "-1\r\n");
  573. EXPECT_STREQ(log_buff[0], "-0.500000\r\n");
  574. /* deinit */
  575. obj_deinit(self);
  576. args_deinit(buffs);
  577. /* check mem */
  578. EXPECT_EQ(pikaMemNow(), 0);
  579. }
  580. TEST(InstructUnit, base) {
  581. __platform_printf("BEGIN\r\n");
  582. InstructUnit ins_unit;
  583. instructUnit_init(&ins_unit);
  584. instructUnit_setBlockDeepth(&ins_unit, 2);
  585. instructUnit_setIsNewLine(&ins_unit, 1);
  586. instructUnit_setInvokeDeepth(&ins_unit, 3);
  587. instructUnit_setInstruct(&ins_unit, (InstructIndex)4);
  588. instructUnit_setConstPoolIndex(&ins_unit, 12);
  589. EXPECT_EQ(instructUnit_getBlockDeepth(&ins_unit), 2);
  590. EXPECT_EQ(instructUnit_getIsNewLine(&ins_unit), 1);
  591. EXPECT_EQ(instructUnit_getInvokeDeepth(&ins_unit), 3);
  592. EXPECT_EQ(instructUnit_getInstructIndex(&ins_unit), 4);
  593. EXPECT_EQ(instructUnit_getConstPoolIndex(&ins_unit), 12);
  594. instructUnit_print(&ins_unit);
  595. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  596. EXPECT_STREQ(log_buff[1], "B2\r\n");
  597. EXPECT_STREQ(log_buff[0], "3 OUT #12\r\n");
  598. EXPECT_EQ(pikaMemNow(), 0);
  599. }
  600. TEST(ConstPool, get) {
  601. __platform_printf("BEGIN\r\n");
  602. ConstPool cp;
  603. constPool_init(&cp);
  604. char* next_content;
  605. uint16_t next_offset;
  606. next_offset = constPool_getLastOffset(&cp);
  607. constPool_append(&cp, "hello");
  608. next_content = constPool_getByOffset(&cp, next_offset);
  609. EXPECT_STREQ(next_content, "hello");
  610. next_offset = constPool_getLastOffset(&cp);
  611. constPool_append(&cp, "world");
  612. next_content = constPool_getByOffset(&cp, next_offset);
  613. EXPECT_STREQ(next_content, "world");
  614. char* first = constPool_getNow(&cp);
  615. char* second = constPool_getNext(&cp);
  616. char* third = constPool_getNext(&cp);
  617. char* forth = constPool_getNext(&cp);
  618. EXPECT_STREQ(first, "");
  619. EXPECT_STREQ(second, "hello");
  620. EXPECT_STREQ(third, "world");
  621. EXPECT_EQ((uintptr_t)forth, (uintptr_t)NULL);
  622. EXPECT_STREQ(constPool_getByIndex(&cp, 0), "");
  623. EXPECT_STREQ(constPool_getByIndex(&cp, 1), "hello");
  624. EXPECT_STREQ(constPool_getByIndex(&cp, 2), "world");
  625. EXPECT_EQ((uintptr_t)constPool_getByIndex(&cp, 3), (uintptr_t)NULL);
  626. constPool_print(&cp);
  627. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  628. EXPECT_STREQ(log_buff[1], "1: hello\r\n");
  629. EXPECT_STREQ(log_buff[0], "7: world\r\n");
  630. constPool_deinit(&cp);
  631. EXPECT_EQ(pikaMemNow(), 0);
  632. }
  633. TEST(InstructArray, set) {
  634. __platform_printf("BEGIN\r\n");
  635. InstructUnit ins_unit;
  636. instructUnit_init(&ins_unit);
  637. instructUnit_setBlockDeepth(&ins_unit, 2);
  638. instructUnit_setIsNewLine(&ins_unit, 1);
  639. instructUnit_setInvokeDeepth(&ins_unit, 3);
  640. instructUnit_setInstruct(&ins_unit, (InstructIndex)4);
  641. instructUnit_setConstPoolIndex(&ins_unit, 12);
  642. InstructArray ins_array;
  643. instructArray_init(&ins_array);
  644. instructArray_append(&ins_array, &ins_unit);
  645. instructArray_print(&ins_array);
  646. instructArray_deinit(&ins_array);
  647. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  648. EXPECT_STREQ(log_buff[1], "B2\r\n");
  649. EXPECT_STREQ(log_buff[0], "3 OUT #12\r\n");
  650. EXPECT_EQ(pikaMemNow(), 0);
  651. }
  652. TEST(VM, bytecode_jjcc) {
  653. char* line = "a = (1 + 1.1) * 3 - 2 /4.0";
  654. Args* buffs = New_strBuff();
  655. char* pikaAsm = pika_lines2Asm(buffs, line);
  656. printf("%s", pikaAsm);
  657. ByteCodeFrame byte_frame;
  658. byteCodeFrame_init(&byte_frame);
  659. byteCodeFrame_appendFromAsm(&byte_frame, pikaAsm);
  660. byteCodeFrame_print(&byte_frame);
  661. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  662. // pikaVM_runAsm(self, pikaAsm);
  663. pikaVM_runByteCodeFrame(self, &byte_frame);
  664. float res = obj_getFloat(self, "a");
  665. obj_deinit(self);
  666. args_deinit(buffs);
  667. byteCodeFrame_deinit(&byte_frame);
  668. ASSERT_FLOAT_EQ(res, 5.8);
  669. EXPECT_EQ(pikaMemNow(), 0);
  670. }
  671. TEST(VM, WHILE_byte) {
  672. g_PikaMemInfo.heapUsedMax = 0;
  673. Args* buffs = New_strBuff();
  674. char* lines =(char *)
  675. "a = 1\n"
  676. "b = 0\n"
  677. "while a:\n"
  678. " b = 1\n"
  679. " a = 0\n"
  680. "\n";
  681. printf("%s", lines);
  682. char* pikaAsm = pika_lines2Asm(buffs, lines);
  683. printf("%s", pikaAsm);
  684. g_PikaMemInfo.heapUsedMax = 0;
  685. PikaObj* self = newRootObj("root", New_TinyObj);
  686. pikaVM_run(self, lines);
  687. EXPECT_EQ(obj_getInt(self, "a"), 0);
  688. EXPECT_EQ(obj_getInt(self, "b"), 1);
  689. args_deinit(buffs);
  690. obj_deinit(self);
  691. EXPECT_EQ(pikaMemNow(), 0);
  692. }
  693. TEST(VM, for_break_byte) {
  694. g_PikaMemInfo.heapUsedMax = 0;
  695. Args* buffs = New_strBuff();
  696. char* lines =(char *)
  697. "a = 0\n"
  698. "for i in range(0, 10):\n"
  699. " if i == 5:\n"
  700. " break\n"
  701. " a = a + i\n"
  702. "\n";
  703. printf("%s", lines);
  704. char* pikaAsm = pika_lines2Asm(buffs, lines);
  705. printf("%s", pikaAsm);
  706. g_PikaMemInfo.heapUsedMax = 0;
  707. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  708. pikaVM_run(self, lines);
  709. /* assert */
  710. int a = obj_getInt(self, "a");
  711. EXPECT_EQ(a, 10);
  712. /* deinit */
  713. args_deinit(buffs);
  714. obj_deinit(self);
  715. EXPECT_EQ(pikaMemNow(), 0);
  716. }
  717. TEST(VM, load_static_bytes) {
  718. const uint8_t bytes[] = {
  719. 0x08, 0x02, 0x00, 0x00, /* instruct array size */
  720. 0x00, 0x82, 0x01, 0x00, 0x00, 0x04, 0x0d, 0x00, 0x00, 0x83, 0x11, 0x00,
  721. 0x00, 0x04, 0x16, 0x00, 0x00, 0x85, 0x1b, 0x00, 0x00, 0x04, 0x1d, 0x00,
  722. 0x20, 0x85, 0x1b, 0x00, 0x20, 0x01, 0x0d, 0x00, 0x10, 0x02, 0x28, 0x00,
  723. 0x00, 0x02, 0x2e, 0x00, 0x00, 0x04, 0x33, 0x00, 0x00, 0x01, 0x37, 0x00,
  724. 0x00, 0x01, 0x3b, 0x00, 0x00, 0x01, 0x3f, 0x00, 0x00, 0x04, 0x43, 0x00,
  725. 0x00, 0x04, 0x4a, 0x00, 0x00, 0x04, 0x51, 0x00, 0x00, 0x82, 0x58, 0x00,
  726. 0x00, 0x04, 0x65, 0x00, 0x00, 0x0d, 0x65, 0x00, 0x00, 0x07, 0x67, 0x00,
  727. 0x11, 0x81, 0x0d, 0x00, 0x11, 0x05, 0x1b, 0x00, 0x01, 0x08, 0x69, 0x00,
  728. 0x01, 0x07, 0x6c, 0x00, 0x02, 0x8e, 0x6c, 0x00, 0x11, 0x81, 0x1d, 0x00,
  729. 0x11, 0x05, 0x1b, 0x00, 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00,
  730. 0x12, 0x81, 0x6e, 0x00, 0x12, 0x01, 0x65, 0x00, 0x02, 0x02, 0x76, 0x00,
  731. 0x02, 0x04, 0x16, 0x00, 0x02, 0x85, 0x6c, 0x00, 0x02, 0x04, 0x1d, 0x00,
  732. 0x01, 0x8b, 0x6c, 0x00, 0x11, 0x01, 0x1d, 0x00, 0x11, 0x05, 0x6c, 0x00,
  733. 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00, 0x12, 0x81, 0x16, 0x00,
  734. 0x12, 0x03, 0x7e, 0x00, 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00,
  735. 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00,
  736. 0x03, 0x04, 0x85, 0x00, 0x03, 0x82, 0x85, 0x00, 0x03, 0x85, 0x1b, 0x00,
  737. 0x03, 0x04, 0x1d, 0x00, 0x02, 0x8b, 0x6c, 0x00, 0x12, 0x01, 0x16, 0x00,
  738. 0x12, 0x03, 0x8a, 0x00, 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00,
  739. 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00,
  740. 0x03, 0x04, 0x8a, 0x00, 0x03, 0x85, 0x67, 0x00, 0x03, 0x04, 0x1d, 0x00,
  741. 0x02, 0x8b, 0x6c, 0x00, 0x12, 0x01, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00,
  742. 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x6e, 0x00,
  743. 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00, 0x03, 0x04, 0x8f, 0x00,
  744. 0x03, 0x85, 0x67, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x01, 0x8b, 0x6c, 0x00,
  745. 0x11, 0x01, 0x1d, 0x00, 0x11, 0x05, 0x67, 0x00, 0x01, 0x08, 0x69, 0x00,
  746. 0x01, 0x07, 0x6c, 0x00, 0x12, 0x81, 0x16, 0x00, 0x12, 0x03, 0x8a, 0x00,
  747. 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00, 0x03, 0x82, 0x8a, 0x00,
  748. 0x03, 0x07, 0x6c, 0x00, 0x14, 0x81, 0x6e, 0x00, 0x14, 0x01, 0x65, 0x00,
  749. 0x04, 0x02, 0x76, 0x00, 0x04, 0x04, 0x85, 0x00, 0x04, 0x82, 0x85, 0x00,
  750. 0x03, 0x85, 0x1b, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x02, 0x8b, 0x6c, 0x00,
  751. 0x12, 0x01, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00, 0x02, 0x08, 0x69, 0x00,
  752. 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00,
  753. 0x03, 0x02, 0x76, 0x00, 0x03, 0x04, 0x85, 0x00, 0x03, 0x85, 0x99, 0x00,
  754. 0x03, 0x04, 0x1d, 0x00, 0x01, 0x8b, 0x6c, 0x00, 0x11, 0x01, 0x1d, 0x00,
  755. 0x11, 0x05, 0x99, 0x00, 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00,
  756. 0x12, 0x81, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00, 0x02, 0x08, 0x69, 0x00,
  757. 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x9b, 0x00, 0x23, 0x01, 0x6e, 0x00,
  758. 0x23, 0x01, 0x65, 0x00, 0x13, 0x02, 0x76, 0x00, 0x03, 0x08, 0xa2, 0x00,
  759. 0x03, 0x07, 0x6c, 0x00, 0x04, 0x82, 0x85, 0x00, 0x14, 0x81, 0x6e, 0x00,
  760. 0x14, 0x01, 0x65, 0x00, 0x24, 0x01, 0x9b, 0x00, 0x24, 0x01, 0x8f, 0x00,
  761. 0x14, 0x08, 0xa4, 0x00, 0x14, 0x03, 0x6e, 0x00, 0x04, 0x02, 0xa6, 0x00,
  762. 0x03, 0x85, 0x1b, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x00, 0x86, 0xae, 0x00,
  763. 0x00, 0x8c, 0x33, 0x00, /* instruct array */
  764. 0xb1, 0x00, 0x00, 0x00, /* const pool size */
  765. 0x00, 0x5f, 0x5f, 0x63, 0x61, 0x6c, 0x6c, 0x73, 0x2e, 0x6c, 0x65, 0x6e,
  766. 0x00, 0x6c, 0x65, 0x6e, 0x00, 0x6e, 0x6f, 0x6e, 0x65, 0x00, 0x6d, 0x6f,
  767. 0x64, 0x65, 0x00, 0x30, 0x00, 0x69, 0x6e, 0x66, 0x6f, 0x5f, 0x69, 0x6e,
  768. 0x64, 0x65, 0x78, 0x00, 0x72, 0x61, 0x6e, 0x67, 0x65, 0x00, 0x69, 0x74,
  769. 0x65, 0x72, 0x00, 0x5f, 0x6c, 0x30, 0x00, 0x5f, 0x72, 0x31, 0x00, 0x5f,
  770. 0x72, 0x32, 0x00, 0x5f, 0x72, 0x33, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x61,
  771. 0x31, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x61, 0x32, 0x00, 0x5f, 0x6c, 0x30,
  772. 0x2e, 0x61, 0x33, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x5f, 0x5f, 0x6e, 0x65,
  773. 0x78, 0x74, 0x5f, 0x5f, 0x00, 0x69, 0x00, 0x32, 0x00, 0x3d, 0x3d, 0x00,
  774. 0x31, 0x00, 0x5f, 0x5f, 0x63, 0x61, 0x6c, 0x6c, 0x73, 0x00, 0x5f, 0x5f,
  775. 0x67, 0x65, 0x74, 0x5f, 0x5f, 0x00, 0x61, 0x6c, 0x77, 0x61, 0x79, 0x73,
  776. 0x00, 0x74, 0x6f, 0x64, 0x6f, 0x00, 0x77, 0x68, 0x65, 0x6e, 0x00, 0x70,
  777. 0x65, 0x72, 0x69, 0x6f, 0x64, 0x5f, 0x6d, 0x73, 0x00, 0x33, 0x00, 0x5f,
  778. 0x5f, 0x74, 0x69, 0x63, 0x6b, 0x00, 0x3e, 0x00, 0x2b, 0x00, 0x5f, 0x5f,
  779. 0x73, 0x65, 0x74, 0x5f, 0x5f, 0x00, 0x2d, 0x31, 0x00, /* const pool */
  780. };
  781. ByteCodeFrame bytecode_frame = {.const_pool =
  782. {
  783. .arg_buff = NULL,
  784. .content_offset_now = 0,
  785. .size = 0,
  786. .content_start = NULL,
  787. .output_redirect_fun = NULL,
  788. .output_f = NULL,
  789. },
  790. .instruct_array = {
  791. .arg_buff = NULL,
  792. .content_offset_now = 0,
  793. .size = 0,
  794. .content_start = NULL,
  795. .output_redirect_fun = NULL,
  796. .output_f = NULL,
  797. }};
  798. byteCodeFrame_loadByteCode(&bytecode_frame, (uint8_t*)bytes);
  799. byteCodeFrame_print(&bytecode_frame);
  800. EXPECT_EQ(instructArray_getSize(&(bytecode_frame.instruct_array)), 520);
  801. EXPECT_EQ(bytecode_frame.const_pool.size, 177);
  802. byteCodeFrame_deinit(&bytecode_frame);
  803. EXPECT_EQ(pikaMemNow(), 0);
  804. }
  805. TEST(VM, multi_jian) {
  806. g_PikaMemInfo.heapUsedMax = 0;
  807. Args* buffs = New_strBuff();
  808. char* lines = "a = (3-4) - 4\n";
  809. printf("%s", lines);
  810. char* pikaAsm = pika_lines2Asm(buffs, lines);
  811. printf("%s", pikaAsm);
  812. g_PikaMemInfo.heapUsedMax = 0;
  813. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  814. pikaVM_run(self, lines);
  815. /* assert */
  816. int a = obj_getInt(self, "a");
  817. EXPECT_EQ(a, -5);
  818. /* deinit */
  819. args_deinit(buffs);
  820. obj_deinit(self);
  821. EXPECT_EQ(pikaMemNow(), 0);
  822. }
  823. TEST(VM, _3_p_3) {
  824. char* line = "-3+3";
  825. Args* buffs = New_strBuff();
  826. char* pikaAsm = pika_lines2Asm(buffs, line);
  827. printf("%s", pikaAsm);
  828. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  829. __platform_printf("BEGIN\r\n");
  830. pikaVM_runAsm(self, pikaAsm);
  831. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  832. EXPECT_STREQ(log_buff[0], "0\r\n");
  833. obj_deinit(self);
  834. args_deinit(buffs);
  835. EXPECT_EQ(pikaMemNow(), 0);
  836. }
  837. TEST(VM, _3_3) {
  838. char* line = "-3-3";
  839. Args* buffs = New_strBuff();
  840. char* pikaAsm = pika_lines2Asm(buffs, line);
  841. printf("%s", pikaAsm);
  842. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  843. __platform_printf("BEGIN\r\n");
  844. pikaVM_runAsm(self, pikaAsm);
  845. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  846. EXPECT_STREQ(log_buff[0], "-6\r\n");
  847. obj_deinit(self);
  848. args_deinit(buffs);
  849. EXPECT_EQ(pikaMemNow(), 0);
  850. }
  851. TEST(VM, a_3) {
  852. char* line = "a-3";
  853. Args* buffs = New_strBuff();
  854. char* pikaAsm = pika_lines2Asm(buffs, line);
  855. printf("%s", pikaAsm);
  856. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  857. __platform_printf("BEGIN\r\n");
  858. pikaVM_runAsm(self, pikaAsm);
  859. EXPECT_STREQ(log_buff[3], "BEGIN\r\n");
  860. EXPECT_STREQ(log_buff[2], "NameError: name 'a' is not defined\r\n");
  861. obj_deinit(self);
  862. args_deinit(buffs);
  863. EXPECT_EQ(pikaMemNow(), 0);
  864. }
  865. TEST(VM, hex_str) {
  866. char* line = "a = '\\x33\\x35'";
  867. Args* buffs = New_strBuff();
  868. char* pikaAsm = pika_lines2Asm(buffs, line);
  869. printf("%s", pikaAsm);
  870. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  871. __platform_printf("BEGIN\r\n");
  872. pikaVM_runAsm(self, pikaAsm);
  873. /* collect */
  874. char* a = obj_getStr(self, "a");
  875. /* assert */
  876. EXPECT_STREQ(a, "35");
  877. /* deinit */
  878. obj_deinit(self);
  879. args_deinit(buffs);
  880. EXPECT_EQ(pikaMemNow(), 0);
  881. }
  882. TEST(VM, hex_bytes) {
  883. char* line =
  884. "a = b'\\x03\\x05'\n"
  885. "a\n";
  886. Args* buffs = New_strBuff();
  887. char* pikaAsm = pika_lines2Asm(buffs, line);
  888. printf("%s", pikaAsm);
  889. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  890. __platform_printf("BEGIN\r\n");
  891. pikaVM_runAsm(self, pikaAsm);
  892. /* collect */
  893. /* assert */
  894. EXPECT_STREQ(log_buff[0], "b'\\x03\\x05'\r\n");
  895. /* deinit */
  896. obj_deinit(self);
  897. args_deinit(buffs);
  898. EXPECT_EQ(pikaMemNow(), 0);
  899. }
  900. TEST(VM, bytes_equ) {
  901. char* line =
  902. "a = b'\\x03\\x05'\n"
  903. "b = b'\\x03\\x05'\n"
  904. "if a == b:\n"
  905. " c = 1\n"
  906. "b = b'\\x02\\x05'\n"
  907. "if a != b:\n"
  908. " d = 1\n";
  909. Args* buffs = New_strBuff();
  910. char* pikaAsm = pika_lines2Asm(buffs, line);
  911. printf("%s", pikaAsm);
  912. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  913. __platform_printf("BEGIN\r\n");
  914. pikaVM_runAsm(self, pikaAsm);
  915. /* collect */
  916. /* assert */
  917. int c = obj_getInt(self, "c");
  918. int d = obj_getInt(self, "d");
  919. EXPECT_EQ(c, 1);
  920. EXPECT_EQ(d, 1);
  921. /* deinit */
  922. obj_deinit(self);
  923. args_deinit(buffs);
  924. EXPECT_EQ(pikaMemNow(), 0);
  925. }
  926. #if PIKA_BUILTIN_STRUCT_ENABLE
  927. TEST(VM, tuple_literal) {
  928. char* line = "(1,2,3)\n";
  929. Args* buffs = New_strBuff();
  930. char* pikaAsm = pika_lines2Asm(buffs, line);
  931. printf("%s", pikaAsm);
  932. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  933. __platform_printf("BEGIN\r\n");
  934. pikaVM_runAsm(self, pikaAsm);
  935. /* collect */
  936. /* assert */
  937. EXPECT_STREQ(log_buff[0], "(1, 2, 3)\r\n");
  938. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  939. /* deinit */
  940. obj_deinit(self);
  941. args_deinit(buffs);
  942. EXPECT_EQ(pikaMemNow(), 0);
  943. }
  944. #endif
  945. TEST(VM, dvd_opt) {
  946. char* line = "a = 10%4\n";
  947. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  948. obj_run(self, line);
  949. /* collect */
  950. int a = obj_getInt(self, "a");
  951. /* assert */
  952. EXPECT_EQ(a, 2);
  953. /* deinit */
  954. obj_deinit(self);
  955. EXPECT_EQ(pikaMemNow(), 0);
  956. }
  957. TEST(VM, del_) {
  958. char* line =
  959. "a = 1\n"
  960. "print(a)\n"
  961. "del a\n"
  962. "print(a)\n";
  963. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  964. obj_run(self, line);
  965. /* collect */
  966. /* assert */
  967. EXPECT_EQ(obj_isArgExist(self, "a"), 0);
  968. /* deinit */
  969. obj_deinit(self);
  970. EXPECT_EQ(pikaMemNow(), 0);
  971. }
  972. TEST(VM, is) {
  973. char* line =
  974. "a = 1\n"
  975. "b = a\n"
  976. "res1 = a is b\n"
  977. "res2 = 1 is 1\n"
  978. "c = TinyObj()\n"
  979. "d = c\n"
  980. "res3 = c is d\n"
  981. "res4 = c is None\n";
  982. ;
  983. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  984. obj_run(self, line);
  985. /* collect */
  986. /* assert */
  987. EXPECT_EQ(obj_getBool(self, "res1"), PIKA_TRUE);
  988. EXPECT_EQ(obj_getBool(self, "res2"), PIKA_TRUE);
  989. EXPECT_EQ(obj_getBool(self, "res3"), PIKA_TRUE);
  990. EXPECT_EQ(obj_getBool(self, "res4"), PIKA_FALSE);
  991. /* deinit */
  992. obj_deinit(self);
  993. EXPECT_EQ(pikaMemNow(), 0);
  994. }
  995. TEST(VM, in) {
  996. char* line =
  997. "res1 = 'a' in 'aaa'\n"
  998. "res2 = 'aaa' in 'a'\n";
  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"), 0);
  1005. /* deinit */
  1006. obj_deinit(self);
  1007. EXPECT_EQ(pikaMemNow(), 0);
  1008. }
  1009. #if !PIKA_NANO_ENABLE
  1010. TEST(VM, assert) {
  1011. char* line =
  1012. "assert True\n"
  1013. "assert 1 == 1, 'testparser'\n"
  1014. "res1 = 0\n"
  1015. "try:\n"
  1016. " assert False\n"
  1017. "except:\n"
  1018. " res1 = 1\n";
  1019. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1020. obj_run(self, line);
  1021. /* collect */
  1022. int res1 = obj_getInt(self, "res1");
  1023. /* assert */
  1024. EXPECT_EQ(res1, 1);
  1025. /* deinit */
  1026. obj_deinit(self);
  1027. EXPECT_EQ(pikaMemNow(), 0);
  1028. }
  1029. #endif
  1030. TEST(VM, issue_I5LHJG) {
  1031. char* line =
  1032. "res = (a, 2, 3)\n"
  1033. "[a, 2, 3]\n"
  1034. "{'x': a, 'y': 2, 'z': 3}\n";
  1035. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1036. obj_run(self, line);
  1037. /* collect */
  1038. /* assert */
  1039. EXPECT_EQ(obj_isArgExist(self, "res"), 0);
  1040. /* deinit */
  1041. obj_deinit(self);
  1042. EXPECT_EQ(pikaMemNow(), 0);
  1043. }
  1044. TEST(VM, vars_runtime) {
  1045. char* line =
  1046. "def testvars(a, *b):\n"
  1047. " sum = 0\n"
  1048. " for i in b:\n"
  1049. " sum += i\n"
  1050. " return a * sum\n"
  1051. "res = testvars(6, 2, 3, 4, 5)\n";
  1052. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1053. obj_run(self, line);
  1054. /* collect */
  1055. int res = obj_getInt(self, "res");
  1056. /* assert */
  1057. EXPECT_EQ(res, 84);
  1058. /* deinit */
  1059. obj_deinit(self);
  1060. EXPECT_EQ(pikaMemNow(), 0);
  1061. }
  1062. #if PIKA_BUILTIN_STRUCT_ENABLE
  1063. TEST(VM, list_add) {
  1064. char* line = "[1, 2, 3] + [4, 5, 6]";
  1065. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1066. obj_run(self, line);
  1067. /* collect */
  1068. /* assert */
  1069. EXPECT_STREQ(log_buff[0], "[1, 2, 3, 4, 5, 6]\r\n");
  1070. /* deinit */
  1071. obj_deinit(self);
  1072. EXPECT_EQ(pikaMemNow(), 0);
  1073. }
  1074. #endif
  1075. TEST(VM, science_num) {
  1076. char* line =
  1077. "a = 1.0e-3\n"
  1078. "b = 2e-5\n"
  1079. "c = -3e-5\n"
  1080. "d = 0.4e2\n";
  1081. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1082. obj_run(self, line);
  1083. /* collect */
  1084. double a = obj_getFloat(self, "a");
  1085. double b = obj_getFloat(self, "b");
  1086. double c = obj_getFloat(self, "c");
  1087. double d = obj_getFloat(self, "d");
  1088. /* assert */
  1089. #if PIKA_FLOAT_TYPE_DOUBLE
  1090. EXPECT_DOUBLE_EQ(a, 1.0e-3);
  1091. EXPECT_DOUBLE_EQ(b, 2.0e-5);
  1092. EXPECT_DOUBLE_EQ(c, -3e-5);
  1093. EXPECT_DOUBLE_EQ(d, 0.4e2);
  1094. #else
  1095. EXPECT_FLOAT_EQ(a, 1.0e-3);
  1096. EXPECT_FLOAT_EQ(b, 2.0e-5);
  1097. EXPECT_FLOAT_EQ(c, -3e-5);
  1098. EXPECT_FLOAT_EQ(d, 0.4e2);
  1099. #endif
  1100. /* deinit */
  1101. obj_deinit(self);
  1102. EXPECT_EQ(pikaMemNow(), 0);
  1103. }
  1104. TEST(VM, issue_I5OJQB) {
  1105. char* line = "s = '\\\\'";
  1106. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1107. obj_run(self, line);
  1108. /* collect */
  1109. char* s = obj_getStr(self, "s");
  1110. /* assert */
  1111. EXPECT_STREQ(s, "\\");
  1112. /* deinit */
  1113. obj_deinit(self);
  1114. EXPECT_EQ(pikaMemNow(), 0);
  1115. }
  1116. #if !PIKA_NANO_ENABLE
  1117. TEST(vm, keyword_2) {
  1118. char* line =
  1119. "def test(a, b):\n"
  1120. " print(a, b)\n"
  1121. "test(a=1, b= 2)";
  1122. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1123. obj_run(self, line);
  1124. /* collect */
  1125. /* assert */
  1126. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1127. /* deinit */
  1128. obj_deinit(self);
  1129. EXPECT_EQ(pikaMemNow(), 0);
  1130. }
  1131. TEST(vm, keyword_3) {
  1132. char* line =
  1133. "def test(**keys):\n"
  1134. " print(keys['a'], keys['b'])\n"
  1135. "test(a=1, b= 2)";
  1136. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1137. obj_run(self, line);
  1138. /* collect */
  1139. /* assert */
  1140. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1141. /* deinit */
  1142. obj_deinit(self);
  1143. EXPECT_EQ(pikaMemNow(), 0);
  1144. }
  1145. TEST(vm, keyword_4) {
  1146. char* line =
  1147. "def test(a, b, **keys):\n"
  1148. " print(keys['c'], keys['d'], a, b)\n"
  1149. "test(3, 4, c=1, d= 2)";
  1150. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1151. obj_run(self, line);
  1152. /* collect */
  1153. /* assert */
  1154. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  1155. /* deinit */
  1156. obj_deinit(self);
  1157. EXPECT_EQ(pikaMemNow(), 0);
  1158. }
  1159. TEST(vm, class_keyword) {
  1160. char* line =
  1161. "class Test:\n"
  1162. " def __init__(self, a, b = 1, c =2):\n"
  1163. " print(a, b, c)\n"
  1164. "t = Test(0, b = 3)\n";
  1165. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1166. obj_run(self, line);
  1167. /* collect */
  1168. /* assert */
  1169. EXPECT_STREQ(log_buff[0], "0 3 2\r\n");
  1170. /* deinit */
  1171. obj_deinit(self);
  1172. EXPECT_EQ(pikaMemNow(), 0);
  1173. }
  1174. TEST(vm, class_keyword_mqtt) {
  1175. char* line =
  1176. "class MQTT:\n"
  1177. " def __init__(self,\n"
  1178. " ip: str,\n"
  1179. " port=1883,\n"
  1180. " clinetID='mac',\n"
  1181. " username='',\n"
  1182. " password='',\n"
  1183. " version='3.1.1',\n"
  1184. " ca='',\n"
  1185. " keepalive=60):\n"
  1186. " print('ip:', ip)\n"
  1187. " print('port:', port)\n"
  1188. "c = MQTT('test')\n";
  1189. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1190. obj_run(self, line);
  1191. /* collect */
  1192. /* assert */
  1193. EXPECT_STREQ(log_buff[1], "ip: test\r\n");
  1194. EXPECT_STREQ(log_buff[0], "port: 1883\r\n");
  1195. /* deinit */
  1196. obj_deinit(self);
  1197. EXPECT_EQ(pikaMemNow(), 0);
  1198. }
  1199. TEST(vm, vars_keyward) {
  1200. /* init */
  1201. g_PikaMemInfo.heapUsedMax = 0;
  1202. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1203. extern unsigned char pikaModules_py_a[];
  1204. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1205. /* run */
  1206. __platform_printf("BEGIN\r\n");
  1207. pikaVM_runSingleFile(pikaMain, "../../examples/BuiltIn/function.py");
  1208. /* collect */
  1209. /* assert */
  1210. EXPECT_STREQ(log_buff[1], "1 2 3 4 5 10\r\n");
  1211. EXPECT_STREQ(log_buff[0], "1 2 3 4 5 12\r\n");
  1212. /* deinit */
  1213. obj_deinit(pikaMain);
  1214. EXPECT_EQ(pikaMemNow(), 0);
  1215. }
  1216. TEST(vm, cb_1) {
  1217. /* init */
  1218. g_PikaMemInfo.heapUsedMax = 0;
  1219. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1220. extern unsigned char pikaModules_py_a[];
  1221. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1222. /* run */
  1223. __platform_printf("BEGIN\r\n");
  1224. pikaVM_runSingleFile(pikaMain, "../../examples/Callback/test1.py");
  1225. /* collect */
  1226. /* assert */
  1227. EXPECT_STREQ(log_buff[3], "__init__\r\n");
  1228. EXPECT_STREQ(log_buff[1], "a\r\n");
  1229. EXPECT_STREQ(log_buff[0], "b\r\n");
  1230. /* deinit */
  1231. obj_deinit(pikaMain);
  1232. EXPECT_EQ(pikaMemNow(), 0);
  1233. }
  1234. TEST(vm, cb_2) {
  1235. /* init */
  1236. g_PikaMemInfo.heapUsedMax = 0;
  1237. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1238. extern unsigned char pikaModules_py_a[];
  1239. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1240. /* run */
  1241. __platform_printf("BEGIN\r\n");
  1242. pikaVM_runSingleFile(pikaMain, "../../examples/Callback/test2.py");
  1243. /* collect */
  1244. /* assert */
  1245. EXPECT_STREQ(log_buff[7], "__init__\r\n");
  1246. EXPECT_STREQ(log_buff[5], "a\r\n");
  1247. EXPECT_STREQ(log_buff[4], "a\r\n");
  1248. EXPECT_STREQ(log_buff[3], "b\r\n");
  1249. EXPECT_STREQ(log_buff[2], "b\r\n");
  1250. EXPECT_STREQ(log_buff[1], "ppp\r\n");
  1251. EXPECT_STREQ(log_buff[0], "ppp\r\n");
  1252. /* deinit */
  1253. obj_deinit(pikaMain);
  1254. EXPECT_EQ(pikaMemNow(), 0);
  1255. }
  1256. TEST(vm, cb_3) {
  1257. /* init */
  1258. g_PikaMemInfo.heapUsedMax = 0;
  1259. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1260. extern unsigned char pikaModules_py_a[];
  1261. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1262. /* run */
  1263. __platform_printf("BEGIN\r\n");
  1264. pikaVM_runSingleFile(pikaMain, "test/python/callback/test3.py");
  1265. /* collect */
  1266. /* deinit */
  1267. obj_deinit(pikaMain);
  1268. EXPECT_EQ(pikaMemNow(), 0);
  1269. }
  1270. #endif
  1271. #if !PIKA_NANO_ENABLE
  1272. TEST(vm, default_no_input) {
  1273. /* init */
  1274. g_PikaMemInfo.heapUsedMax = 0;
  1275. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1276. extern unsigned char pikaModules_py_a[];
  1277. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1278. /* run */
  1279. __platform_printf("BEGIN\r\n");
  1280. obj_run(pikaMain,
  1281. "def test(a = 1):\n"
  1282. " print(a)\n"
  1283. "test()");
  1284. /* collect */
  1285. /* assert */
  1286. EXPECT_STREQ(log_buff[0], "1\r\n");
  1287. /* deinit */
  1288. obj_deinit(pikaMain);
  1289. EXPECT_EQ(pikaMemNow(), 0);
  1290. }
  1291. TEST(vm, default_1) {
  1292. /* init */
  1293. g_PikaMemInfo.heapUsedMax = 0;
  1294. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1295. extern unsigned char pikaModules_py_a[];
  1296. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1297. /* run */
  1298. __platform_printf("BEGIN\r\n");
  1299. obj_run(pikaMain,
  1300. "def test(a = 1):\n"
  1301. " print(a)\n"
  1302. "test(a = 2)");
  1303. /* collect */
  1304. /* assert */
  1305. EXPECT_STREQ(log_buff[0], "2\r\n");
  1306. /* deinit */
  1307. obj_deinit(pikaMain);
  1308. EXPECT_EQ(pikaMemNow(), 0);
  1309. }
  1310. TEST(vm, default_2) {
  1311. /* init */
  1312. g_PikaMemInfo.heapUsedMax = 0;
  1313. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1314. extern unsigned char pikaModules_py_a[];
  1315. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1316. /* run */
  1317. __platform_printf("BEGIN\r\n");
  1318. obj_run(pikaMain,
  1319. "def test(a, b = 1):\n"
  1320. " print(a, b)\n"
  1321. "test(1, b = 2)");
  1322. /* collect */
  1323. /* assert */
  1324. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1325. /* deinit */
  1326. obj_deinit(pikaMain);
  1327. EXPECT_EQ(pikaMemNow(), 0);
  1328. }
  1329. TEST(vm, default_3) {
  1330. /* init */
  1331. g_PikaMemInfo.heapUsedMax = 0;
  1332. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1333. extern unsigned char pikaModules_py_a[];
  1334. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1335. /* run */
  1336. __platform_printf("BEGIN\r\n");
  1337. obj_run(pikaMain,
  1338. "def myprint(end='\\n', *var):\n"
  1339. " for v in var:\n"
  1340. " print(v, end = end)\n"
  1341. "myprint(1, 2)\n"
  1342. "myprint(1, 2, end=';\\n')\n");
  1343. /* collect */
  1344. /* assert */
  1345. EXPECT_STREQ(log_buff[3], "1\n");
  1346. EXPECT_STREQ(log_buff[2], "2\n");
  1347. EXPECT_STREQ(log_buff[1], "1;\n");
  1348. EXPECT_STREQ(log_buff[0], "2;\n");
  1349. /* deinit */
  1350. obj_deinit(pikaMain);
  1351. EXPECT_EQ(pikaMemNow(), 0);
  1352. }
  1353. TEST(vm, default_4) {
  1354. /* init */
  1355. g_PikaMemInfo.heapUsedMax = 0;
  1356. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1357. extern unsigned char pikaModules_py_a[];
  1358. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1359. /* run */
  1360. __platform_printf("BEGIN\r\n");
  1361. obj_run(pikaMain,
  1362. "def test(end=5, *var):\n"
  1363. " res = end\n"
  1364. " print('end', end)\n"
  1365. " for v in var:\n"
  1366. " res += v\n"
  1367. " return res\n"
  1368. "test(1, 2, 3)\n"
  1369. "test(1, 2, 3, end=6)\n");
  1370. /* collect */
  1371. /* assert */
  1372. EXPECT_STREQ(log_buff[3], "end 5\r\n");
  1373. EXPECT_STREQ(log_buff[2], "11\r\n");
  1374. EXPECT_STREQ(log_buff[1], "end 6\r\n");
  1375. EXPECT_STREQ(log_buff[0], "12\r\n");
  1376. /* deinit */
  1377. obj_deinit(pikaMain);
  1378. EXPECT_EQ(pikaMemNow(), 0);
  1379. }
  1380. TEST(vm, default_no_kw) {
  1381. /* init */
  1382. g_PikaMemInfo.heapUsedMax = 0;
  1383. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1384. extern unsigned char pikaModules_py_a[];
  1385. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1386. /* run */
  1387. __platform_printf("BEGIN\r\n");
  1388. obj_run(pikaMain,
  1389. "def test(a, b=3):\n"
  1390. " print(a ,b)\n"
  1391. "test(1, 2)\n"
  1392. "test(1, b=2)\n"
  1393. "test(1)\n");
  1394. /* collect */
  1395. /* assert */
  1396. EXPECT_STREQ(log_buff[2], "1 2\r\n");
  1397. EXPECT_STREQ(log_buff[1], "1 2\r\n");
  1398. EXPECT_STREQ(log_buff[0], "1 3\r\n");
  1399. /* deinit */
  1400. obj_deinit(pikaMain);
  1401. EXPECT_EQ(pikaMemNow(), 0);
  1402. }
  1403. #endif
  1404. TEST(vm, none) {
  1405. /* init */
  1406. g_PikaMemInfo.heapUsedMax = 0;
  1407. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1408. extern unsigned char pikaModules_py_a[];
  1409. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1410. /* run */
  1411. __platform_printf("BEGIN\r\n");
  1412. obj_run(pikaMain, "GTestTask.test(1, None)");
  1413. /* collect */
  1414. /* assert */
  1415. EXPECT_STREQ(log_buff[0], "GTestTask_test: 1, 0\r\n");
  1416. /* deinit */
  1417. obj_deinit(pikaMain);
  1418. EXPECT_EQ(pikaMemNow(), 0);
  1419. }
  1420. #if !PIKA_NANO_ENABLE
  1421. TEST(vm, super_) {
  1422. /* init */
  1423. g_PikaMemInfo.heapUsedMax = 0;
  1424. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1425. extern unsigned char pikaModules_py_a[];
  1426. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1427. /* run */
  1428. __platform_printf("BEGIN\r\n");
  1429. obj_run(pikaMain,
  1430. "class test:\n"
  1431. " def __init__(self):\n"
  1432. " print('in test init')\n"
  1433. "class test2(test):\n"
  1434. " def __init__(self):\n"
  1435. " super().__init__()\n"
  1436. " print('in test2 init')\n"
  1437. "t = test2()");
  1438. /* collect */
  1439. /* assert */
  1440. EXPECT_STREQ(log_buff[0], "in test2 init\r\n");
  1441. EXPECT_STREQ(log_buff[1], "in test init\r\n");
  1442. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1443. /* deinit */
  1444. obj_deinit(pikaMain);
  1445. EXPECT_EQ(pikaMemNow(), 0);
  1446. }
  1447. TEST(vm, super_val) {
  1448. /* init */
  1449. g_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. "class test:\n"
  1457. " def __init__(self):\n"
  1458. " self.a = 1\n"
  1459. " self.b = 2\n"
  1460. "class test2(test):\n"
  1461. " def __init__(self):\n"
  1462. " super().__init__()\n"
  1463. " self.c = 3\n"
  1464. " self.d = 4\n"
  1465. "t = test2()\n"
  1466. "print(t.a, t.b, t.c, t.d)\n");
  1467. /* collect */
  1468. /* assert */
  1469. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1470. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  1471. /* deinit */
  1472. obj_deinit(pikaMain);
  1473. EXPECT_EQ(pikaMemNow(), 0);
  1474. }
  1475. TEST(vm, super_val_) {
  1476. /* init */
  1477. g_PikaMemInfo.heapUsedMax = 0;
  1478. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1479. extern unsigned char pikaModules_py_a[];
  1480. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1481. /* run */
  1482. __platform_printf("BEGIN\r\n");
  1483. obj_run(pikaMain,
  1484. "class test:\n"
  1485. " def test(self, a):\n"
  1486. " self.a = a\n"
  1487. "class test2(test):\n"
  1488. " def test(self, a):\n"
  1489. " super().test(str(a))\n"
  1490. "t1 = test()\n"
  1491. "t2 = test2()\n"
  1492. "t1.test(1)\n"
  1493. "t2.test(1)\n");
  1494. /* collect */
  1495. int t1_a = obj_getInt(pikaMain, "t1.a");
  1496. char* t2_a = obj_getStr(pikaMain, "t2.a");
  1497. /* assert */
  1498. EXPECT_EQ(t1_a, 1);
  1499. EXPECT_STREQ(t2_a, "1");
  1500. /* deinit */
  1501. obj_deinit(pikaMain);
  1502. EXPECT_EQ(pikaMemNow(), 0);
  1503. }
  1504. #endif
  1505. #if !PIKA_NANO_ENABLE
  1506. TEST(vm, multi_return) {
  1507. /* init */
  1508. g_PikaMemInfo.heapUsedMax = 0;
  1509. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1510. extern unsigned char pikaModules_py_a[];
  1511. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1512. /* run */
  1513. __platform_printf("BEGIN\r\n");
  1514. obj_run(pikaMain, "a,b = (1,2)");
  1515. /* collect */
  1516. int a = obj_getInt(pikaMain, "a");
  1517. int b = obj_getInt(pikaMain, "b");
  1518. /* assert */
  1519. EXPECT_EQ(a, 1);
  1520. EXPECT_EQ(b, 2);
  1521. /* deinit */
  1522. obj_deinit(pikaMain);
  1523. EXPECT_EQ(pikaMemNow(), 0);
  1524. }
  1525. #endif
  1526. #if !PIKA_NANO_ENABLE
  1527. TEST(vm, multi_return_fn) {
  1528. /* init */
  1529. g_PikaMemInfo.heapUsedMax = 0;
  1530. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1531. extern unsigned char pikaModules_py_a[];
  1532. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1533. /* run */
  1534. __platform_printf("BEGIN\r\n");
  1535. obj_run(pikaMain,
  1536. "def test():\n"
  1537. " return 1,2\n"
  1538. "a,b = test()");
  1539. /* collect */
  1540. int a = obj_getInt(pikaMain, "a");
  1541. int b = obj_getInt(pikaMain, "b");
  1542. /* assert */
  1543. EXPECT_EQ(a, 1);
  1544. EXPECT_EQ(b, 2);
  1545. /* deinit */
  1546. obj_deinit(pikaMain);
  1547. EXPECT_EQ(pikaMemNow(), 0);
  1548. }
  1549. #endif
  1550. TEST(vm, range_1) {
  1551. /* init */
  1552. g_PikaMemInfo.heapUsedMax = 0;
  1553. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1554. extern unsigned char pikaModules_py_a[];
  1555. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1556. /* run */
  1557. __platform_printf("BEGIN\r\n");
  1558. obj_run(pikaMain,
  1559. "sum = 0\n"
  1560. "for i in range(10):\n"
  1561. " sum += i\n");
  1562. /* collect */
  1563. int sum = obj_getInt(pikaMain, "sum");
  1564. /* assert */
  1565. EXPECT_EQ(sum, 45);
  1566. /* deinit */
  1567. obj_deinit(pikaMain);
  1568. EXPECT_EQ(pikaMemNow(), 0);
  1569. }
  1570. TEST(vm, rang_3) {
  1571. /* init */
  1572. g_PikaMemInfo.heapUsedMax = 0;
  1573. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1574. extern unsigned char pikaModules_py_a[];
  1575. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1576. /* run */
  1577. __platform_printf("BEGIN\r\n");
  1578. obj_run(pikaMain,
  1579. "sum = 0\n"
  1580. "for i in range(-5, 20, 2):\n"
  1581. " sum += i\n");
  1582. /* collect */
  1583. int sum = obj_getInt(pikaMain, "sum");
  1584. /* assert */
  1585. EXPECT_EQ(sum, 91);
  1586. /* deinit */
  1587. obj_deinit(pikaMain);
  1588. EXPECT_EQ(pikaMemNow(), 0);
  1589. }
  1590. TEST(vm, test64) {
  1591. /* init */
  1592. g_PikaMemInfo.heapUsedMax = 0;
  1593. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1594. extern unsigned char pikaModules_py_a[];
  1595. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1596. /* run */
  1597. __platform_printf("BEGIN\r\n");
  1598. obj_run(pikaMain,
  1599. "import GTestTask\n"
  1600. "res = GTestTask.test64(4294967295, 20)\n");
  1601. /* collect */
  1602. int64_t res = obj_getInt(pikaMain, "res");
  1603. /* assert */
  1604. EXPECT_EQ(res, 4294967295 * 20);
  1605. /* deinit */
  1606. obj_deinit(pikaMain);
  1607. EXPECT_EQ(pikaMemNow(), 0);
  1608. }
  1609. #if !PIKA_NANO_ENABLE
  1610. TEST(vm, exit) {
  1611. /* init */
  1612. g_PikaMemInfo.heapUsedMax = 0;
  1613. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1614. extern unsigned char pikaModules_py_a[];
  1615. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1616. /* run */
  1617. __platform_printf("BEGIN\r\n");
  1618. obj_run(pikaMain,
  1619. "i = 0\n"
  1620. "while True:\n"
  1621. " i += 1\n"
  1622. " print(i)\n"
  1623. " if i == 10:\n"
  1624. " exit()\n");
  1625. /* collect */
  1626. int i = obj_getInt(pikaMain, "i");
  1627. /* assert */
  1628. EXPECT_EQ(i, 10);
  1629. /* deinit */
  1630. obj_deinit(pikaMain);
  1631. EXPECT_EQ(pikaMemNow(), 0);
  1632. }
  1633. TEST(vm, exit_fn) {
  1634. /* init */
  1635. g_PikaMemInfo.heapUsedMax = 0;
  1636. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1637. extern unsigned char pikaModules_py_a[];
  1638. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1639. /* run */
  1640. __platform_printf("BEGIN\r\n");
  1641. obj_run(pikaMain,
  1642. "i = 0\n"
  1643. "PikaStdLib.MemChecker.now()\n"
  1644. "def test():\n"
  1645. " global i\n"
  1646. " while True:\n"
  1647. " i += 1\n"
  1648. " print(i)\n"
  1649. " if i == 10:\n"
  1650. " exit()\n"
  1651. "while True:\n"
  1652. " test()\n");
  1653. obj_run(pikaMain, "PikaStdLib.MemChecker.now()\n");
  1654. /* collect */
  1655. int i = obj_getInt(pikaMain, "i");
  1656. /* assert */
  1657. EXPECT_EQ(i, 10);
  1658. /* deinit */
  1659. obj_deinit(pikaMain);
  1660. EXPECT_EQ(pikaMemNow(), 0);
  1661. }
  1662. #endif
  1663. #if PIKA_INSTRUCT_HOOK_ENABLE
  1664. typedef void (*hook_func)(void);
  1665. extern volatile int g_hook_cnt;
  1666. extern volatile hook_func g_hook_func;
  1667. extern "C" {
  1668. volatile int g_hook_cnt_triggle;
  1669. void hook_func_exit_issue_1(void) {
  1670. if (g_hook_cnt >= g_hook_cnt_triggle) {
  1671. pks_vm_exit();
  1672. }
  1673. }
  1674. void __gtest_hook_default_(void) {
  1675. return;
  1676. }
  1677. }
  1678. void _vm_exit_fn_issue_1_item(int hook_cnt) {
  1679. g_hook_cnt_triggle = hook_cnt;
  1680. /* init */
  1681. g_PikaMemInfo.heapUsedMax = 0;
  1682. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1683. extern unsigned char pikaModules_py_a[];
  1684. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1685. /* run */
  1686. __platform_printf("BEGIN\r\n");
  1687. obj_run(pikaMain,
  1688. "PikaStdLib.MemChecker.now()\n"
  1689. "def task1():\n"
  1690. " print('task1 hello')\n"
  1691. "\n"
  1692. "def task2():\n"
  1693. " print('task2 hello')\n"
  1694. "\n"
  1695. "def fake_runtask():\n"
  1696. " i = 0\n"
  1697. " while True:\n"
  1698. " i += 1\n"
  1699. " print(i)\n"
  1700. " task1()\n"
  1701. " task2()\n"
  1702. "fake_runtask()\n"
  1703. "fake_runtask()\n"
  1704. "\n");
  1705. obj_run(pikaMain, "PikaStdLib.MemChecker.now()\n");
  1706. /* collect */
  1707. /* assert */
  1708. obj_deinit(pikaMain);
  1709. EXPECT_EQ(pikaMemNow(), 0);
  1710. }
  1711. TEST(vm, exit_fn_issue_1) {
  1712. g_hook_func = hook_func_exit_issue_1;
  1713. for (int i = 0; i < 50; i++) {
  1714. g_hook_cnt = 0;
  1715. _vm_exit_fn_issue_1_item(i);
  1716. }
  1717. g_hook_func = __gtest_hook_default_;
  1718. }
  1719. #endif
  1720. TEST(vm, pass_) {
  1721. /* init */
  1722. g_PikaMemInfo.heapUsedMax = 0;
  1723. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1724. extern unsigned char pikaModules_py_a[];
  1725. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1726. /* run */
  1727. __platform_printf("BEGIN\r\n");
  1728. obj_run(pikaMain,
  1729. "pass\n"
  1730. "def testpass():\n"
  1731. " pass\n"
  1732. "for i in range(10):\n"
  1733. " pass\n"
  1734. "testpass()\n"
  1735. "print('after pass')\n");
  1736. /* collect */
  1737. /* assert */
  1738. EXPECT_STREQ(log_buff[0], "after pass\r\n");
  1739. /* deinit */
  1740. obj_deinit(pikaMain);
  1741. EXPECT_EQ(pikaMemNow(), 0);
  1742. }
  1743. TEST(vm, test64_hex) {
  1744. /* init */
  1745. g_PikaMemInfo.heapUsedMax = 0;
  1746. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1747. extern unsigned char pikaModules_py_a[];
  1748. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1749. /* run */
  1750. __platform_printf("BEGIN\r\n");
  1751. obj_run(pikaMain,
  1752. "import GTestTask\n"
  1753. "res = GTestTask.test64(0xffffffff, 20)\n");
  1754. /* collect */
  1755. int64_t res = obj_getInt(pikaMain, "res");
  1756. /* assert */
  1757. EXPECT_EQ(res, 4294967295 * 20);
  1758. /* deinit */
  1759. obj_deinit(pikaMain);
  1760. EXPECT_EQ(pikaMemNow(), 0);
  1761. }
  1762. TEST(vm, test64_hex_print) {
  1763. /* init */
  1764. g_PikaMemInfo.heapUsedMax = 0;
  1765. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1766. extern unsigned char pikaModules_py_a[];
  1767. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1768. /* run */
  1769. __platform_printf("BEGIN\r\n");
  1770. obj_run(pikaMain,
  1771. "0xffffffff\n"
  1772. "print(0xffffffff)\n");
  1773. /* collect */
  1774. /* assert */
  1775. EXPECT_STREQ(log_buff[0], "4294967295\r\n");
  1776. EXPECT_STREQ(log_buff[1], "4294967295\r\n");
  1777. /* deinit */
  1778. obj_deinit(pikaMain);
  1779. EXPECT_EQ(pikaMemNow(), 0);
  1780. }
  1781. #if !PIKA_NANO_ENABLE
  1782. TEST(vm, call_dict_err) {
  1783. /* init */
  1784. g_PikaMemInfo.heapUsedMax = 0;
  1785. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1786. extern unsigned char pikaModules_py_a[];
  1787. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1788. /* run */
  1789. __platform_printf("BEGIN\r\n");
  1790. obj_run(pikaMain,
  1791. "a = {'a': 1}\n"
  1792. "a('a')\n");
  1793. /* collect */
  1794. /* assert */
  1795. EXPECT_STREQ(log_buff[4], "TypeError: 'a' object is not callable\r\n");
  1796. /* deinit */
  1797. obj_deinit(pikaMain);
  1798. EXPECT_EQ(pikaMemNow(), 0);
  1799. }
  1800. TEST(vm, getattribute) {
  1801. /* init */
  1802. g_PikaMemInfo.heapUsedMax = 0;
  1803. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1804. extern unsigned char pikaModules_py_a[];
  1805. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1806. /* run */
  1807. __platform_printf("BEGIN\r\n");
  1808. obj_run(pikaMain,
  1809. "class test:\n"
  1810. " def __getattribute__(self, name):\n"
  1811. " return name\n"
  1812. "t = test()\n"
  1813. "t.a\n");
  1814. /* collect */
  1815. /* assert */
  1816. EXPECT_STREQ(log_buff[0], "'a'\r\n");
  1817. /* deinit */
  1818. obj_deinit(pikaMain);
  1819. EXPECT_EQ(pikaMemNow(), 0);
  1820. }
  1821. TEST(vm, getattr) {
  1822. /* init */
  1823. g_PikaMemInfo.heapUsedMax = 0;
  1824. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1825. extern unsigned char pikaModules_py_a[];
  1826. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1827. /* run */
  1828. __platform_printf("BEGIN\r\n");
  1829. obj_run(pikaMain,
  1830. "class test:\n"
  1831. " a = 1\n"
  1832. " def __getattr__(self, name):\n"
  1833. " return name\n"
  1834. "t = test()\n"
  1835. "t.a\n"
  1836. "t.b\n");
  1837. /* collect */
  1838. /* assert */
  1839. EXPECT_STREQ(log_buff[0], "'b'\r\n");
  1840. EXPECT_STREQ(log_buff[1], "1\r\n");
  1841. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1842. /* deinit */
  1843. obj_deinit(pikaMain);
  1844. EXPECT_EQ(pikaMemNow(), 0);
  1845. }
  1846. TEST(vm, setattr) {
  1847. /* init */
  1848. g_PikaMemInfo.heapUsedMax = 0;
  1849. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1850. extern unsigned char pikaModules_py_a[];
  1851. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1852. /* run */
  1853. __platform_printf("BEGIN\r\n");
  1854. obj_run(pikaMain,
  1855. "class test:\n"
  1856. " a = 1\n"
  1857. " def __setattr__(self, name, val):\n"
  1858. " print((name, val))\n"
  1859. "t = test()\n"
  1860. "t.a = 1\n"
  1861. "t.b = 'test'\n");
  1862. /* collect */
  1863. /* assert */
  1864. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1865. EXPECT_STREQ(log_buff[1], "('a', 1)\r\n");
  1866. EXPECT_STREQ(log_buff[0], "('b', 'test')\r\n");
  1867. /* deinit */
  1868. obj_deinit(pikaMain);
  1869. EXPECT_EQ(pikaMemNow(), 0);
  1870. }
  1871. TEST(vm, c_module_get_set_attr) {
  1872. /* init */
  1873. g_PikaMemInfo.heapUsedMax = 0;
  1874. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1875. extern unsigned char pikaModules_py_a[];
  1876. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1877. /* run */
  1878. __platform_printf("BEGIN\r\n");
  1879. obj_run(pikaMain,
  1880. "t = GTestTask.ProxyTest()\n"
  1881. "t.a\n"
  1882. "t.b\n"
  1883. "t.a = 'test1'\n"
  1884. "t.b = 'test2'\n");
  1885. /* collect */
  1886. /* assert */
  1887. EXPECT_STREQ(log_buff[3], "'a'\r\n");
  1888. EXPECT_STREQ(log_buff[2], "'b'\r\n");
  1889. EXPECT_STREQ(log_buff[1], "GTestTask_ProxyTest___setattr__: a, test1\r\n");
  1890. EXPECT_STREQ(log_buff[0], "GTestTask_ProxyTest___setattr__: b, test2\r\n");
  1891. /* deinit */
  1892. obj_deinit(pikaMain);
  1893. EXPECT_EQ(pikaMemNow(), 0);
  1894. }
  1895. TEST(vm, class_attr_ref) {
  1896. /* init */
  1897. g_PikaMemInfo.heapUsedMax = 0;
  1898. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1899. extern unsigned char pikaModules_py_a[];
  1900. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1901. /* run */
  1902. __platform_printf("BEGIN\r\n");
  1903. obj_run(pikaMain,
  1904. "class test:\n"
  1905. " a = 1\n"
  1906. " b = a\n"
  1907. "t = test()\n");
  1908. /* collect */
  1909. /* assert */
  1910. /* deinit */
  1911. obj_deinit(pikaMain);
  1912. EXPECT_EQ(pikaMemNow(), 0);
  1913. }
  1914. TEST(vm, getattr_native) {
  1915. /* init */
  1916. g_PikaMemInfo.heapUsedMax = 0;
  1917. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1918. extern unsigned char pikaModules_py_a[];
  1919. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1920. /* run */
  1921. __platform_printf("BEGIN\r\n");
  1922. obj_run(pikaMain,
  1923. "mem = PikaStdLib.MemChecker()\n"
  1924. "if hasattr(mem, 'max'):\n"
  1925. " max = getattr(mem, 'max')\n"
  1926. " max()\n");
  1927. /* collect */
  1928. /* assert */
  1929. /* deinit */
  1930. obj_deinit(pikaMain);
  1931. EXPECT_EQ(pikaMemNow(), 0);
  1932. }
  1933. TEST(vm, issue_dict_update) {
  1934. /* init */
  1935. g_PikaMemInfo.heapUsedMax = 0;
  1936. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1937. extern unsigned char pikaModules_py_a[];
  1938. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1939. /* run */
  1940. __platform_printf("BEGIN\r\n");
  1941. pikaVM_runSingleFile(pikaMain, "test/python/issue/issue_dict_update.py");
  1942. /* collect */
  1943. /* assert */
  1944. EXPECT_STREQ(obj_getStr(pikaMain, "inner"), "widget5");
  1945. /* deinit */
  1946. obj_deinit(pikaMain);
  1947. EXPECT_EQ(pikaMemNow(), 0);
  1948. }
  1949. TEST(vm, issue_big_dict_update) {
  1950. /* init */
  1951. g_PikaMemInfo.heapUsedMax = 0;
  1952. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1953. extern unsigned char pikaModules_py_a[];
  1954. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1955. /* run */
  1956. __platform_printf("BEGIN\r\n");
  1957. pikaVM_runSingleFile(pikaMain,
  1958. "test/python/issue/issue_big_dict_update.py");
  1959. /* collect */
  1960. /* assert */
  1961. /* deinit */
  1962. obj_deinit(pikaMain);
  1963. EXPECT_EQ(pikaMemNow(), 0);
  1964. }
  1965. #endif
  1966. TEST(vm, i_pp) {
  1967. /* init */
  1968. g_PikaMemInfo.heapUsedMax = 0;
  1969. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1970. extern unsigned char pikaModules_py_a[];
  1971. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1972. /* run */
  1973. __platform_printf("BEGIN\r\n");
  1974. /* clang-format off */
  1975. PIKA_PYTHON(
  1976. i = 0
  1977. while i < 10000:
  1978. i += 1
  1979. )
  1980. /* clang-format on */
  1981. const uint8_t bytes[] = {
  1982. 0x30, 0x00, 0x00, 0x00, /* instruct array size */
  1983. 0x00, 0x85, 0x01, 0x00, 0x00, 0x04, 0x03, 0x00, 0x10, 0x81, 0x03, 0x00,
  1984. 0x10, 0x05, 0x05, 0x00, 0x00, 0x08, 0x0b, 0x00, 0x00, 0x07, 0x0d, 0x00,
  1985. 0x11, 0x81, 0x03, 0x00, 0x21, 0x05, 0x0f, 0x00, 0x11, 0x02, 0x00, 0x00,
  1986. 0x01, 0x08, 0x11, 0x00, 0x01, 0x04, 0x03, 0x00, 0x00, 0x86, 0x13, 0x00,
  1987. /* instruct array */
  1988. 0x16, 0x00, 0x00, 0x00, /* const pool size */
  1989. 0x00, 0x30, 0x00, 0x69, 0x00, 0x31, 0x30, 0x30, 0x30, 0x30, 0x00, 0x3c,
  1990. 0x00, 0x32, 0x00, 0x31, 0x00, 0x2b, 0x00, 0x2d, 0x31,
  1991. 0x00, /* const pool */
  1992. };
  1993. pikaVM_runByteCode(pikaMain, (uint8_t*)bytes);
  1994. EXPECT_EQ(obj_getInt(pikaMain, "i"), 10000);
  1995. /* deinit */
  1996. obj_deinit(pikaMain);
  1997. EXPECT_EQ(pikaMemNow(), 0);
  1998. }
  1999. TEST(vm, benchmark) {
  2000. /* init */
  2001. g_PikaMemInfo.heapUsedMax = 0;
  2002. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2003. extern unsigned char pikaModules_py_a[];
  2004. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2005. /* run */
  2006. __platform_printf("BEGIN\r\n");
  2007. /* clang-format off */
  2008. PIKA_PYTHON(
  2009. num = 0
  2010. i = 2
  2011. while i < 1000:
  2012. is_prime = 1
  2013. j = 2
  2014. while j < i:
  2015. if i%j==0 :
  2016. is_prime = 0
  2017. break
  2018. j += 1
  2019. if is_prime:
  2020. num = num + i
  2021. i += 1
  2022. )
  2023. /* clang-format on */
  2024. const uint8_t bytes[] = {
  2025. 0xac, 0x00, 0x00, 0x00, /* instruct array size */
  2026. 0x00, 0x85, 0x01, 0x00, 0x00, 0x04, 0x03, 0x00, 0x00, 0x85, 0x07, 0x00,
  2027. 0x00, 0x04, 0x09, 0x00, 0x10, 0x81, 0x09, 0x00, 0x10, 0x05, 0x0b, 0x00,
  2028. 0x00, 0x08, 0x10, 0x00, 0x00, 0x07, 0x07, 0x00, 0x01, 0x85, 0x12, 0x00,
  2029. 0x01, 0x04, 0x14, 0x00, 0x01, 0x85, 0x07, 0x00, 0x01, 0x04, 0x1d, 0x00,
  2030. 0x11, 0x81, 0x1d, 0x00, 0x11, 0x01, 0x09, 0x00, 0x01, 0x08, 0x10, 0x00,
  2031. 0x01, 0x07, 0x07, 0x00, 0x22, 0x81, 0x09, 0x00, 0x22, 0x01, 0x1d, 0x00,
  2032. 0x12, 0x08, 0x1f, 0x00, 0x12, 0x05, 0x01, 0x00, 0x02, 0x08, 0x21, 0x00,
  2033. 0x02, 0x07, 0x12, 0x00, 0x03, 0x85, 0x01, 0x00, 0x03, 0x04, 0x14, 0x00,
  2034. 0x03, 0x8e, 0x00, 0x00, 0x12, 0x81, 0x1d, 0x00, 0x22, 0x05, 0x12, 0x00,
  2035. 0x12, 0x02, 0x00, 0x00, 0x02, 0x08, 0x24, 0x00, 0x02, 0x04, 0x1d, 0x00,
  2036. 0x01, 0x86, 0x26, 0x00, 0x01, 0x81, 0x14, 0x00, 0x01, 0x07, 0x12, 0x00,
  2037. 0x12, 0x81, 0x03, 0x00, 0x12, 0x01, 0x09, 0x00, 0x02, 0x08, 0x24, 0x00,
  2038. 0x02, 0x04, 0x03, 0x00, 0x11, 0x81, 0x09, 0x00, 0x21, 0x05, 0x12, 0x00,
  2039. 0x11, 0x02, 0x00, 0x00, 0x01, 0x08, 0x24, 0x00, 0x01, 0x04, 0x09, 0x00,
  2040. 0x00, 0x86, 0x26, 0x00, /* instruct array */
  2041. 0x29, 0x00, 0x00, 0x00, /* const pool size */
  2042. 0x00, 0x30, 0x00, 0x6e, 0x75, 0x6d, 0x00, 0x32, 0x00, 0x69, 0x00, 0x31,
  2043. 0x30, 0x30, 0x30, 0x00, 0x3c, 0x00, 0x31, 0x00, 0x69, 0x73, 0x5f, 0x70,
  2044. 0x72, 0x69, 0x6d, 0x65, 0x00, 0x6a, 0x00, 0x25, 0x00, 0x3d, 0x3d, 0x00,
  2045. 0x2b, 0x00, 0x2d, 0x31, 0x00, /* const pool */
  2046. };
  2047. pikaVM_runByteCode(pikaMain, (uint8_t*)bytes);
  2048. /* deinit */
  2049. obj_deinit(pikaMain);
  2050. EXPECT_EQ(pikaMemNow(), 0);
  2051. }
  2052. TEST(VM, print_None) {
  2053. char* line = "print(None)\n";
  2054. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  2055. obj_run(self, line);
  2056. /* collect */
  2057. /* assert */
  2058. EXPECT_STREQ(log_buff[0], "None\r\n");
  2059. /* deinit */
  2060. obj_deinit(self);
  2061. EXPECT_EQ(pikaMemNow(), 0);
  2062. }
  2063. TEST(VM, bc_fn) {
  2064. PikaObj* self = newRootObj("root", New_PikaMain);
  2065. obj_run(self, "print('hello world')\n");
  2066. /* clang-format off */
  2067. PIKA_PYTHON(
  2068. def test():
  2069. print('test')
  2070. )
  2071. /* clang-format on */
  2072. const uint8_t
  2073. bytes[] =
  2074. {
  2075. 0x14, 0x00, 0x00, 0x00, /* instruct array size */
  2076. 0x00, 0x89, 0x01, 0x00, 0x00, 0x06, 0x08,
  2077. 0x00, 0x11, 0x83, 0x0a, 0x00, 0x01, 0x02,
  2078. 0x0f, 0x00, 0x01, 0x8a, 0x00, 0x00, /* instruct array */
  2079. 0x15, 0x00, 0x00, 0x00, /* const pool size */
  2080. 0x00, 0x74, 0x65, 0x73, 0x74, 0x28, 0x29,
  2081. 0x00, 0x31, 0x00, 0x74, 0x65, 0x73, 0x74,
  2082. 0x00, 0x70, 0x72, 0x69, 0x6e, 0x74, 0x00, /* const pool */
  2083. };
  2084. pikaVM_runByteCode(self, (uint8_t*)bytes);
  2085. obj_run(self, "test()\n");
  2086. EXPECT_STREQ(log_buff[1], "hello world\r\n");
  2087. EXPECT_STREQ(log_buff[0], "test\r\n");
  2088. obj_deinit(self);
  2089. EXPECT_EQ(pikaMemNow(), 0);
  2090. }
  2091. #if !PIKA_NANO_ENABLE
  2092. TEST(VM, default_num_err) {
  2093. char* line =
  2094. "def test(a, b=1):\n"
  2095. " print(a, b)\n"
  2096. "test(1, 2, 3)\n";
  2097. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  2098. obj_run(self, line);
  2099. /* collect */
  2100. /* assert */
  2101. EXPECT_STREQ(log_buff[8],
  2102. "TypeError: test() takes from 1 to 2 positional arguments but "
  2103. "3 were given\r\n");
  2104. /* deinit */
  2105. obj_deinit(self);
  2106. EXPECT_EQ(pikaMemNow(), 0);
  2107. }
  2108. #endif
  2109. TEST(VM, bc_fn_file_cb0) {
  2110. PikaObj* self = newRootObj("root", New_PikaMain);
  2111. pikaVM_runByteCodeFile(self,
  2112. "package/pikascript/pikascript-api/cb_test.py.o");
  2113. obj_run(self, "test()\n");
  2114. EXPECT_STREQ(log_buff[0], "test\r\n");
  2115. obj_deinit(self);
  2116. EXPECT_EQ(pikaMemNow(), 0);
  2117. }
  2118. TEST(VM, bc_fn_file_cb2) {
  2119. PikaObj* self = newRootObj("root", New_PikaMain);
  2120. obj_run(self, "print('hello world')\n");
  2121. pikaVM_runByteCodeFile(self,
  2122. "package/pikascript/pikascript-api/cb_test.py.o");
  2123. obj_run(self, "test()\n");
  2124. EXPECT_STREQ(log_buff[1], "hello world\r\n");
  2125. EXPECT_STREQ(log_buff[0], "test\r\n");
  2126. obj_deinit(self);
  2127. EXPECT_EQ(pikaMemNow(), 0);
  2128. }
  2129. #if !PIKA_NANO_ENABLE
  2130. TEST(vm, slice_str_end) {
  2131. /* init */
  2132. g_PikaMemInfo.heapUsedMax = 0;
  2133. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2134. /* run */
  2135. __platform_printf("BEGIN\r\n");
  2136. obj_run(pikaMain, "'test'[:-1]");
  2137. /* collect */
  2138. /* assert */
  2139. EXPECT_STREQ(log_buff[0], "'tes'\r\n");
  2140. /* deinit */
  2141. obj_deinit(pikaMain);
  2142. EXPECT_EQ(pikaMemNow(), 0);
  2143. }
  2144. TEST(vm, slice_list) {
  2145. /* init */
  2146. g_PikaMemInfo.heapUsedMax = 0;
  2147. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2148. /* run */
  2149. __platform_printf("BEGIN\r\n");
  2150. obj_run(pikaMain, "[1,2,3][1:]");
  2151. /* collect */
  2152. /* assert */
  2153. EXPECT_STREQ(log_buff[0], "[2, 3]\r\n");
  2154. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2155. /* deinit */
  2156. obj_deinit(pikaMain);
  2157. EXPECT_EQ(pikaMemNow(), 0);
  2158. }
  2159. TEST(vm, slice_list_list) {
  2160. /* init */
  2161. g_PikaMemInfo.heapUsedMax = 0;
  2162. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2163. /* run */
  2164. __platform_printf("BEGIN\r\n");
  2165. obj_run(pikaMain,
  2166. "l = "
  2167. "[[\"ID\",\"TCP_URL\",\"VERSION\",\"SENSOR_SCAN_s\",\"DATA_UPLOAD_"
  2168. "s\",\"ACT_LOGIC_SCAN_s\",\"NETWORK\",\"AUTO_ACT\",\"START_TIME\"],"
  2169. "[\"ABC123\",\"iot.365sn.cn/operate/h/"
  2170. "1234\",\"V1.1\",60,1800,0.5,\"4G\",\"TRUE\",1669017826]]\n"
  2171. "ll = l[1:]\n"
  2172. "print(len(l), l)\n"
  2173. "print(len(ll),ll)\n");
  2174. /* collect */
  2175. /* assert */
  2176. EXPECT_STREQ(log_buff[0],
  2177. "9 ['ABC123', 'iot.365sn.cn/operate/h/1234', 'V1.1', 60, "
  2178. "1800, 0.500000, '4G', 'TRUE', 1669017826]\r\n");
  2179. EXPECT_STREQ(log_buff[1],
  2180. "2 [['ID', 'TCP_URL', 'VERSION', 'SENSOR_SCAN_s', "
  2181. "'DATA_UPLOAD_s', 'ACT_LOGIC_SCAN_s', 'NETWORK', 'AUTO_ACT', "
  2182. "'START_TIME'], ['ABC123', 'iot.365sn.cn/operate/h/1234', "
  2183. "'V1.1', 60, 1800, 0.500000, '4G', 'TRUE', 1669017826]]\r\n");
  2184. /* deinit */
  2185. obj_deinit(pikaMain);
  2186. EXPECT_EQ(pikaMemNow(), 0);
  2187. }
  2188. #endif
  2189. #if !PIKA_NANO_ENABLE
  2190. TEST(vm, fn_pos_kw) {
  2191. /* init */
  2192. g_PikaMemInfo.heapUsedMax = 0;
  2193. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2194. /* run */
  2195. __platform_printf("BEGIN\r\n");
  2196. obj_run(pikaMain,
  2197. "def test(a, b):\n"
  2198. " print(a, b)\n"
  2199. "test(b=1,a=2)\n");
  2200. /* collect */
  2201. /* assert */
  2202. EXPECT_STREQ(log_buff[0], "2 1\r\n");
  2203. /* deinit */
  2204. obj_deinit(pikaMain);
  2205. EXPECT_EQ(pikaMemNow(), 0);
  2206. }
  2207. TEST(vm, fn_pos_kw2) {
  2208. /* init */
  2209. g_PikaMemInfo.heapUsedMax = 0;
  2210. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2211. /* run */
  2212. __platform_printf("BEGIN\r\n");
  2213. obj_run(pikaMain,
  2214. "def test(a, b):\n"
  2215. " print(a, b)\n"
  2216. "test(1, b=2)\n"
  2217. "test(1, a=2)\n");
  2218. /* collect */
  2219. /* assert */
  2220. EXPECT_STREQ(log_buff[1], "1 2\r\n");
  2221. EXPECT_STREQ(log_buff[0], "2 1\r\n");
  2222. /* deinit */
  2223. obj_deinit(pikaMain);
  2224. EXPECT_EQ(pikaMemNow(), 0);
  2225. }
  2226. TEST(vm, fn_star) {
  2227. /* init */
  2228. g_PikaMemInfo.heapUsedMax = 0;
  2229. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2230. /* run */
  2231. __platform_printf("BEGIN\r\n");
  2232. obj_run(pikaMain,
  2233. "a = (1,2,3)\n"
  2234. "print(*a)\n");
  2235. /* collect */
  2236. /* assert */
  2237. EXPECT_STREQ(log_buff[0], "1 2 3\r\n");
  2238. /* deinit */
  2239. obj_deinit(pikaMain);
  2240. EXPECT_EQ(pikaMemNow(), 0);
  2241. }
  2242. TEST(vm, fn_star_pos) {
  2243. /* init */
  2244. g_PikaMemInfo.heapUsedMax = 0;
  2245. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2246. /* run */
  2247. __platform_printf("BEGIN\r\n");
  2248. obj_run(pikaMain,
  2249. "a = (1,2,3)\n"
  2250. "print(*a, 4)\n");
  2251. /* collect */
  2252. /* assert */
  2253. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  2254. /* deinit */
  2255. obj_deinit(pikaMain);
  2256. EXPECT_EQ(pikaMemNow(), 0);
  2257. }
  2258. TEST(vm, fn_star_pos_2) {
  2259. /* init */
  2260. g_PikaMemInfo.heapUsedMax = 0;
  2261. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2262. /* run */
  2263. __platform_printf("BEGIN\r\n");
  2264. obj_run(pikaMain,
  2265. "a = (1,2,3)\n"
  2266. "print(0, *a, 4)\n");
  2267. /* collect */
  2268. /* assert */
  2269. EXPECT_STREQ(log_buff[0], "0 1 2 3 4\r\n");
  2270. /* deinit */
  2271. obj_deinit(pikaMain);
  2272. EXPECT_EQ(pikaMemNow(), 0);
  2273. }
  2274. TEST(vm, fn_star_star) {
  2275. /* init */
  2276. g_PikaMemInfo.heapUsedMax = 0;
  2277. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2278. /* run */
  2279. __platform_printf("BEGIN\r\n");
  2280. obj_run(pikaMain,
  2281. "def test(**kw):\n"
  2282. " print(kw['a'], kw['b'], kw['c'])\n"
  2283. "a = {'a':1,'b':2,'c':3}\n"
  2284. "test(**a)\n");
  2285. /* collect */
  2286. /* assert */
  2287. EXPECT_STREQ(log_buff[0], "1 2 3\r\n");
  2288. /* deinit */
  2289. obj_deinit(pikaMain);
  2290. EXPECT_EQ(pikaMemNow(), 0);
  2291. }
  2292. TEST(vm, issue_keyword_mem_leak) {
  2293. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2294. extern unsigned char pikaModules_py_a[];
  2295. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2296. pikaVM_runSingleFile(pikaMain,
  2297. "test/python/issue/issue_keyword_mem_leak.py");
  2298. /* assert */
  2299. /* deinit */
  2300. obj_deinit(pikaMain);
  2301. EXPECT_EQ(pikaMemNow(), 0);
  2302. }
  2303. TEST(vm, issue_not_in) {
  2304. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2305. extern unsigned char pikaModules_py_a[];
  2306. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2307. obj_run(pikaMain,
  2308. "a = [1,2,3]\n"
  2309. "print(1 not in a)\n"
  2310. "print(4 not in a)\n");
  2311. /* assert */
  2312. /* deinit */
  2313. obj_deinit(pikaMain);
  2314. EXPECT_EQ(pikaMemNow(), 0);
  2315. }
  2316. TEST(vm, def_not_in) {
  2317. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2318. extern unsigned char pikaModules_py_a[];
  2319. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2320. obj_run(pikaMain,
  2321. "t = ['hmac-md5', 'other']\n"
  2322. "def test(a=\"hmac-md5\"):\n"
  2323. " print(\"input:\", a)\n"
  2324. " print(\"table:\", t)\n"
  2325. " if a not in t:\n"
  2326. " print(\"a not in t\")\n"
  2327. " else:\n"
  2328. " print(\"a in t\")\n"
  2329. "test()\n");
  2330. /* assert */
  2331. EXPECT_STREQ(log_buff[0], "a in t\r\n");
  2332. /* deinit */
  2333. obj_deinit(pikaMain);
  2334. EXPECT_EQ(pikaMemNow(), 0);
  2335. }
  2336. TEST(vm, fn_pos_kw_issue1) {
  2337. /* init */
  2338. g_PikaMemInfo.heapUsedMax = 0;
  2339. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2340. /* run */
  2341. __platform_printf("BEGIN\r\n");
  2342. obj_run(pikaMain,
  2343. "def new(a=1, b=2):\n"
  2344. " print(a, b)\n"
  2345. "new(3, b = 2)\n");
  2346. /* collect */
  2347. /* assert */
  2348. EXPECT_STREQ(log_buff[0], "3 2\r\n");
  2349. /* deinit */
  2350. obj_deinit(pikaMain);
  2351. EXPECT_EQ(pikaMemNow(), 0);
  2352. }
  2353. TEST(vm, fn_pos_kw_issue2) {
  2354. /* init */
  2355. g_PikaMemInfo.heapUsedMax = 0;
  2356. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2357. /* run */
  2358. __platform_printf("BEGIN\r\n");
  2359. obj_run(pikaMain,
  2360. "def new(key:any, msg=None, digestmod=\"md5\"):\n"
  2361. " print(key, msg, digestmod)\n"
  2362. "new('123','456',digestmod=\"md5\")\n");
  2363. /* collect */
  2364. /* assert */
  2365. EXPECT_STREQ(log_buff[0], "123 456 md5\r\n");
  2366. /* deinit */
  2367. obj_deinit(pikaMain);
  2368. EXPECT_EQ(pikaMemNow(), 0);
  2369. }
  2370. TEST(vm, num_issue_lakj) {
  2371. /* init */
  2372. g_PikaMemInfo.heapUsedMax = 0;
  2373. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2374. /* run */
  2375. __platform_printf("BEGIN\r\n");
  2376. obj_run(pikaMain, "a = 1/2*3\n");
  2377. /* collect */
  2378. /* assert */
  2379. EXPECT_DOUBLE_EQ(obj_getFloat(pikaMain, "a"), 1.5);
  2380. /* deinit */
  2381. obj_deinit(pikaMain);
  2382. EXPECT_EQ(pikaMemNow(), 0);
  2383. }
  2384. TEST(vm, dir_issue) {
  2385. /* init */
  2386. g_PikaMemInfo.heapUsedMax = 0;
  2387. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2388. /* run */
  2389. __platform_printf("BEGIN\r\n");
  2390. obj_run(pikaMain,
  2391. "class test:\n"
  2392. " pass\n"
  2393. "dir(test)\n");
  2394. /* collect */
  2395. /* assert */
  2396. /* deinit */
  2397. obj_deinit(pikaMain);
  2398. EXPECT_EQ(pikaMemNow(), 0);
  2399. }
  2400. TEST(vm, dir_issue1lk) {
  2401. /* init */
  2402. g_PikaMemInfo.heapUsedMax = 0;
  2403. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2404. extern unsigned char pikaModules_py_a[];
  2405. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2406. /* run */
  2407. __platform_printf("BEGIN\r\n");
  2408. obj_run(pikaMain,
  2409. "import requests\n"
  2410. "dir(requests)\n");
  2411. /* collect */
  2412. /* assert */
  2413. /* deinit */
  2414. obj_deinit(pikaMain);
  2415. EXPECT_EQ(pikaMemNow(), 0);
  2416. }
  2417. TEST(vm, type_int) {
  2418. /* init */
  2419. g_PikaMemInfo.heapUsedMax = 0;
  2420. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2421. extern unsigned char pikaModules_py_a[];
  2422. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2423. /* run */
  2424. __platform_printf("BEGIN\r\n");
  2425. obj_run(pikaMain,
  2426. "i = type(1)\n"
  2427. "ii = i(10)\n");
  2428. /* collect */
  2429. EXPECT_EQ(obj_getInt(pikaMain, "ii"), 10);
  2430. /* assert */
  2431. /* deinit */
  2432. obj_deinit(pikaMain);
  2433. EXPECT_EQ(pikaMemNow(), 0);
  2434. }
  2435. TEST(vm, method_int) {
  2436. /* init */
  2437. g_PikaMemInfo.heapUsedMax = 0;
  2438. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2439. extern unsigned char pikaModules_py_a[];
  2440. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2441. /* run */
  2442. __platform_printf("BEGIN\r\n");
  2443. obj_run(pikaMain,
  2444. "i = int\n"
  2445. "ii = i(10)\n");
  2446. /* collect */
  2447. /* assert */
  2448. EXPECT_EQ(obj_getInt(pikaMain, "ii"), 10);
  2449. /* deinit */
  2450. obj_deinit(pikaMain);
  2451. EXPECT_EQ(pikaMemNow(), 0);
  2452. }
  2453. TEST(vm, fn_method_int) {
  2454. /* init */
  2455. g_PikaMemInfo.heapUsedMax = 0;
  2456. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2457. extern unsigned char pikaModules_py_a[];
  2458. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2459. /* run */
  2460. __platform_printf("BEGIN\r\n");
  2461. obj_run(pikaMain,
  2462. "def test():\n"
  2463. " i = int\n"
  2464. " ii = i(10)\n"
  2465. " return ii\n"
  2466. "ii = test()\n");
  2467. /* collect */
  2468. /* assert */
  2469. EXPECT_EQ(obj_getInt(pikaMain, "ii"), 10);
  2470. /* deinit */
  2471. obj_deinit(pikaMain);
  2472. EXPECT_EQ(pikaMemNow(), 0);
  2473. }
  2474. TEST(vm, kw_no_empty) {
  2475. /* init */
  2476. g_PikaMemInfo.heapUsedMax = 0;
  2477. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2478. extern unsigned char pikaModules_py_a[];
  2479. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2480. /* run */
  2481. __platform_printf("BEGIN\r\n");
  2482. obj_run(pikaMain,
  2483. "def test(a=None, b=None, c=None):\n"
  2484. " print(a, b, c)\n"
  2485. "test(1, 2)\n");
  2486. /* collect */
  2487. /* assert */
  2488. EXPECT_STREQ(log_buff[0], "1 2 None\r\n");
  2489. /* deinit */
  2490. obj_deinit(pikaMain);
  2491. EXPECT_EQ(pikaMemNow(), 0);
  2492. }
  2493. TEST(vm, tuple_void) {
  2494. /* init */
  2495. g_PikaMemInfo.heapUsedMax = 0;
  2496. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2497. extern unsigned char pikaModules_py_a[];
  2498. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2499. /* run */
  2500. __platform_printf("BEGIN\r\n");
  2501. obj_run(pikaMain, "()\n");
  2502. /* collect */
  2503. /* assert */
  2504. EXPECT_STREQ(log_buff[0], "()\r\n");
  2505. /* deinit */
  2506. obj_deinit(pikaMain);
  2507. EXPECT_EQ(pikaMemNow(), 0);
  2508. }
  2509. TEST(vm, run_file) {
  2510. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2511. pikaVM_runFile(pikaMain, "package/pikascript/main.py");
  2512. obj_deinit(pikaMain);
  2513. EXPECT_EQ(pikaMemNow(), 0);
  2514. }
  2515. TEST(vm, bool_) {
  2516. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2517. obj_run(pikaMain, "print(True)\n");
  2518. obj_deinit(pikaMain);
  2519. EXPECT_STREQ(log_buff[0], "True\r\n");
  2520. EXPECT_EQ(pikaMemNow(), 0);
  2521. }
  2522. TEST(vm, method_cb) {
  2523. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2524. obj_run(pikaMain,
  2525. "class Test:\n"
  2526. " _cb = None\n"
  2527. " def callback(self):\n"
  2528. " print('test')\n"
  2529. " def addcb(self, cb):\n"
  2530. " self._cb = cb\n"
  2531. " def runcb(self):\n"
  2532. " self._cb()\n"
  2533. " def __init__(self):\n"
  2534. " self.addcb(self.callback)\n"
  2535. "t = Test()\n"
  2536. "t.runcb()");
  2537. obj_deinit(pikaMain);
  2538. EXPECT_STREQ(log_buff[0], "test\r\n");
  2539. EXPECT_EQ(pikaMemNow(), 0);
  2540. }
  2541. TEST(vm, class_getattr) {
  2542. /* init */
  2543. g_PikaMemInfo.heapUsedMax = 0;
  2544. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2545. extern unsigned char pikaModules_py_a[];
  2546. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2547. /* run */
  2548. __platform_printf("BEGIN\r\n");
  2549. pikaVM_runSingleFile(pikaMain, "test/python/callback/class_getattr.py");
  2550. /* collect */
  2551. /* assert */
  2552. EXPECT_STREQ(log_buff[0], "2\r\n");
  2553. EXPECT_STREQ(log_buff[1], "1\r\n");
  2554. /* deinit */
  2555. obj_deinit(pikaMain);
  2556. EXPECT_EQ(pikaMemNow(), 0);
  2557. }
  2558. TEST(vm, type_fullfealure) {
  2559. /* init */
  2560. g_PikaMemInfo.heapUsedMax = 0;
  2561. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2562. extern unsigned char pikaModules_py_a[];
  2563. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2564. /* run */
  2565. __platform_printf("BEGIN\r\n");
  2566. pikaVM_runSingleFile(pikaMain, "test/python/builtin/type.py");
  2567. /* collect */
  2568. /* assert */
  2569. EXPECT_STREQ(log_buff[0], "PASS\r\n");
  2570. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2571. /* deinit */
  2572. obj_deinit(pikaMain);
  2573. EXPECT_EQ(pikaMemNow(), 0);
  2574. }
  2575. TEST(vm, dir_print_arg) {
  2576. /* init */
  2577. g_PikaMemInfo.heapUsedMax = 0;
  2578. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2579. extern unsigned char pikaModules_py_a[];
  2580. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2581. /* run */
  2582. __platform_printf("BEGIN\r\n");
  2583. obj_run(pikaMain, "print(dir)\r\n");
  2584. /* collect */
  2585. /* assert */
  2586. EXPECT_STREQ(log_buff[0], "<built-in function dir>\r\n");
  2587. /* deinit */
  2588. obj_deinit(pikaMain);
  2589. EXPECT_EQ(pikaMemNow(), 0);
  2590. }
  2591. TEST(vm, fn_pos_vars) {
  2592. /* init */
  2593. g_PikaMemInfo.heapUsedMax = 0;
  2594. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2595. extern unsigned char pikaModules_py_a[];
  2596. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2597. /* run */
  2598. __platform_printf("BEGIN\r\n");
  2599. pikaVM_runSingleFile(pikaMain, "test/python/builtin/fn_pos_vars.py");
  2600. /* collect */
  2601. /* assert */
  2602. EXPECT_STREQ(
  2603. log_buff[4],
  2604. "TypeError: log() takes 2 positional argument but 1 were given\r\n");
  2605. /* deinit */
  2606. obj_deinit(pikaMain);
  2607. EXPECT_EQ(pikaMemNow(), 0);
  2608. }
  2609. TEST(vm, for_return) {
  2610. /* init */
  2611. g_PikaMemInfo.heapUsedMax = 0;
  2612. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2613. extern unsigned char pikaModules_py_a[];
  2614. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2615. /* run */
  2616. __platform_printf("BEGIN\r\n");
  2617. pikaVM_runSingleFile(pikaMain, "test/python/builtin/for_return.py");
  2618. /* collect */
  2619. /* assert */
  2620. /* deinit */
  2621. obj_deinit(pikaMain);
  2622. EXPECT_EQ(pikaMemNow(), 0);
  2623. }
  2624. TEST(vm, slice_issue) {
  2625. /* init */
  2626. g_PikaMemInfo.heapUsedMax = 0;
  2627. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2628. extern unsigned char pikaModules_py_a[];
  2629. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2630. /* run */
  2631. __platform_printf("BEGIN\r\n");
  2632. pikaVM_runSingleFile(pikaMain, "test/python/builtin/slice.py");
  2633. /* collect */
  2634. /* assert */
  2635. EXPECT_STREQ(log_buff[0], "PASS\r\n");
  2636. /* deinit */
  2637. obj_deinit(pikaMain);
  2638. EXPECT_EQ(pikaMemNow(), 0);
  2639. }
  2640. TEST_RUN_LINES(vm, dot_issue, ".")
  2641. TEST_RUN_LINES(vm, char_issue1, "~")
  2642. TEST_RUN_LINES(vm, char_issue2, "/")
  2643. TEST_RUN_LINES(vm, char_issue3, "%")
  2644. TEST_RUN_LINES(vm, char_issue4, "=")
  2645. TEST_RUN_SINGLE_FILE(vm,
  2646. issue_star_dict,
  2647. "test/python/issue/issue_star_dict.py")
  2648. TEST_RUN_SINGLE_FILE_PASS(vm, proxy2, "test/python/proxy/proxy2.py")
  2649. #if 0 //! need fix
  2650. TEST_RUN_LINES_PASS(vm,
  2651. none_none,
  2652. "assert (None is not None) == False\n"
  2653. "print('PASS')\n")
  2654. #endif
  2655. TEST_RUN_LINES(vm, minus_none, "10 - None")
  2656. TEST_RUN_LINES_PASS(vm,
  2657. abs,
  2658. "assert abs(-1) == 1\n"
  2659. "assert abs(1) == 1\n"
  2660. "assert abs(-1.0) == 1.0\n"
  2661. "assert abs(0.1) == 0.1\n"
  2662. "print('PASS')\n")
  2663. TEST_RUN_LINES(vm, abs_none, "abs(None)")
  2664. TEST_RUN_LINES(vm, abs_str, "abs('test')")
  2665. TEST_RUN_SINGLE_FILE_PASS(datastruct,
  2666. circlequeue,
  2667. "test/python/datastruct/circlequeue.py")
  2668. TEST_RUN_LINES_EXCEPT_OUTPUT(vm, single_tuple, "(1,)", "(1,)\r\n")
  2669. TEST_RUN_LINES_EXCEPT_OUTPUT(vm, single_tuple_str, "('test',)", "('test',)\r\n")
  2670. TEST_RUN_SINGLE_FILE_PASS(vm, is_not, "test/python/builtin/is_not.py")
  2671. TEST_RUN_LINES(vm,
  2672. var_global,
  2673. "import PikaStdLib\n"
  2674. "mem = PikaStdLib.MemChecker()\n"
  2675. "mem.clear\n")
  2676. TEST_RUN_LINES(vm,
  2677. var_global_run,
  2678. "import PikaStdLib\n"
  2679. "mem = PikaStdLib.MemChecker()\n"
  2680. "mem.clear()\n")
  2681. TEST_RUN_LINES(vm,
  2682. var_global_module,
  2683. "import configparser\n"
  2684. "configparser.clear\n")
  2685. TEST_RUN_LINES(vm, import_void, "import \n")
  2686. #endif
  2687. TEST_END