VM-test.cpp 76 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611
  1. #include "test_common.h"
  2. TEST_START
  3. extern "C" {
  4. extern char log_buff[LOG_BUFF_MAX][LOG_SIZE];
  5. /* test head */
  6. VMParameters* pikaVM_runAsmWithPars(PikaObj* self,
  7. VMParameters* locals,
  8. VMParameters* globals,
  9. char* pikaAsm);
  10. char* Parser_LineToAsm(Args* buffs, char* line, Stack* blockStack);
  11. char* Parser_linesToAsm(Args* outBuffs, char* multiLine);
  12. int32_t __clearInvokeStackes(VMParameters* locals);
  13. int32_t pikaVM_runAsmLine(PikaObj* self,
  14. VMParameters* locals,
  15. VMParameters* globals,
  16. char* pikaAsm,
  17. int32_t lineAddr);
  18. }
  19. TEST(VM, num1) {
  20. char* line = "1";
  21. Args* buffs = New_strBuff();
  22. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  23. printf("%s", pikaAsm);
  24. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  25. __platform_printf("BEGIN\r\n");
  26. pikaVM_runAsm(self, pikaAsm);
  27. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  28. EXPECT_STREQ(log_buff[0], "1\r\n");
  29. obj_deinit(self);
  30. args_deinit(buffs);
  31. EXPECT_EQ(pikaMemNow(), 0);
  32. }
  33. TEST(VM, a_1) {
  34. char* line = "a = 1";
  35. Args* buffs = New_strBuff();
  36. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  37. printf("%s", pikaAsm);
  38. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  39. pikaVM_runAsm(self, pikaAsm);
  40. ASSERT_EQ(args_getInt(self->list, "a"), 1);
  41. obj_deinit(self);
  42. args_deinit(buffs);
  43. EXPECT_EQ(pikaMemNow(), 0);
  44. }
  45. TEST(VM, a_1d1) {
  46. char* line = "a = 1.1";
  47. Args* buffs = New_strBuff();
  48. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  49. printf("%s", pikaAsm);
  50. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  51. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  52. ASSERT_FLOAT_EQ(args_getFloat(globals->list, "a"), 1.1);
  53. obj_deinit(self);
  54. // obj_deinit(globals);
  55. args_deinit(buffs);
  56. EXPECT_EQ(pikaMemNow(), 0);
  57. }
  58. TEST(VM, str_xy) {
  59. char* line = "a = 'xy'";
  60. Args* buffs = New_strBuff();
  61. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  62. printf("%s", pikaAsm);
  63. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  64. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  65. ASSERT_STREQ(args_getStr(globals->list, "a"), "xy");
  66. obj_deinit(self);
  67. // obj_deinit(globals);
  68. args_deinit(buffs);
  69. EXPECT_EQ(pikaMemNow(), 0);
  70. }
  71. TEST(VM, str_xy_space) {
  72. char* line = "a = 'xy '";
  73. Args* buffs = New_strBuff();
  74. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  75. printf("%s", pikaAsm);
  76. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  77. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  78. ASSERT_STREQ(args_getStr(globals->list, "a"), "xy ");
  79. obj_deinit(self);
  80. args_deinit(buffs);
  81. // obj_deinit(globals);
  82. EXPECT_EQ(pikaMemNow(), 0);
  83. }
  84. TEST(VM, ref_a_b) {
  85. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  86. Args* buffs = New_strBuff();
  87. pikaVM_runAsm(self, Parser_LineToAsm(buffs, "a = 'xy '", NULL));
  88. args_deinit(buffs);
  89. obj_deinit(self);
  90. // obj_deinit(globals);
  91. EXPECT_EQ(pikaMemNow(), 0);
  92. }
  93. TEST(VM, Run_add) {
  94. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  95. Args* buffs = New_strBuff();
  96. char* pikaAsm = Parser_LineToAsm(buffs, "a = plusInt(1,2)", NULL);
  97. __platform_printf("%s", pikaAsm);
  98. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  99. args_deinit(buffs);
  100. int a = args_getInt(globals->list, "a");
  101. ASSERT_EQ(a, 3);
  102. obj_deinit(self);
  103. // obj_deinit(globals);
  104. EXPECT_EQ(pikaMemNow(), 0);
  105. }
  106. TEST(VM, Run_add_multy) {
  107. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  108. Args* buffs = New_strBuff();
  109. pikaVM_runAsm(self, Parser_LineToAsm(buffs, "b = 2", NULL));
  110. args_deinit(buffs);
  111. obj_deinit(self);
  112. // obj_deinit(globals);
  113. EXPECT_EQ(pikaMemNow(), 0);
  114. }
  115. TEST(VM, Run_add_1_2_3) {
  116. PikaObj* self = newRootObj("root", New_PikaMath_Operator);
  117. Args* buffs = New_strBuff();
  118. char* pikaAsm =
  119. Parser_LineToAsm(buffs, "a = plusInt(1, plusInt(2,3) )", NULL);
  120. __platform_printf("%s", pikaAsm);
  121. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  122. args_deinit(buffs);
  123. int a = args_getInt(globals->list, "a");
  124. ASSERT_EQ(a, 6);
  125. obj_deinit(self);
  126. // obj_deinit(globals);
  127. EXPECT_EQ(pikaMemNow(), 0);
  128. }
  129. extern PikaMemInfo pikaMemInfo;
  130. TEST(VM, WHILE) {
  131. pikaMemInfo.heapUsedMax = 0;
  132. Args* buffs = New_strBuff();
  133. char* lines =(char *)
  134. "a = 1\n"
  135. "b = 0\n"
  136. "while a:\n"
  137. " b = 1\n"
  138. " a = 0\n"
  139. "\n";
  140. printf("%s", lines);
  141. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  142. printf("%s", pikaAsm);
  143. pikaMemInfo.heapUsedMax = 0;
  144. PikaObj* self = newRootObj("root", New_TinyObj);
  145. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  146. EXPECT_EQ(args_getInt(globals->list, "a"), 0);
  147. EXPECT_EQ(args_getInt(globals->list, "b"), 1);
  148. // obj_deinit(globals);
  149. args_deinit(buffs);
  150. obj_deinit(self);
  151. EXPECT_EQ(pikaMemNow(), 0);
  152. }
  153. TEST(VM, a_1_1) {
  154. char* line = "a = 1 + 1";
  155. Args* buffs = New_strBuff();
  156. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  157. printf("%s", pikaAsm);
  158. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  159. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  160. int res = args_getInt(globals->list, "a");
  161. obj_deinit(self);
  162. args_deinit(buffs);
  163. ASSERT_EQ(res, 2);
  164. EXPECT_EQ(pikaMemNow(), 0);
  165. }
  166. TEST(VM, a_1_1d1) {
  167. char* line = "a = 1 + 1.1";
  168. Args* buffs = New_strBuff();
  169. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  170. printf("%s", pikaAsm);
  171. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  172. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  173. float res = args_getFloat(globals->list, "a");
  174. obj_deinit(self);
  175. // obj_deinit(globals);
  176. args_deinit(buffs);
  177. ASSERT_FLOAT_EQ(res, 2.1);
  178. EXPECT_EQ(pikaMemNow(), 0);
  179. }
  180. TEST(VM, a_jjcc) {
  181. char* line = "a = (1 + 1.1) * 3 - 2 /4.0";
  182. printf("%s\r\n", line);
  183. Args* buffs = New_strBuff();
  184. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  185. printf("%s", pikaAsm);
  186. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  187. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  188. float res = args_getFloat(globals->list, "a");
  189. obj_deinit(self);
  190. args_deinit(buffs);
  191. // obj_deinit(globals);
  192. ASSERT_FLOAT_EQ(res, 5.8);
  193. EXPECT_EQ(pikaMemNow(), 0);
  194. }
  195. TEST(VM, while_a_1to10) {
  196. char* line =
  197. "a = 0\n"
  198. "while a < 10:\n"
  199. " a += 1\n"
  200. " print(a)\n"
  201. "\n";
  202. Args* buffs = New_strBuff();
  203. char* pikaAsm = Parser_linesToAsm(buffs, line);
  204. printf("%s", pikaAsm);
  205. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  206. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  207. int res = args_getInt(globals->list, "a");
  208. obj_deinit(self);
  209. args_deinit(buffs);
  210. // obj_deinit(globals);
  211. ASSERT_FLOAT_EQ(res, 10);
  212. EXPECT_EQ(pikaMemNow(), 0);
  213. }
  214. TEST(VM, mem_x) {
  215. char* line =
  216. "mem = PikaStdLib.MemChecker()\n"
  217. "mem.x = 1\n"
  218. "print(mem.x)\n"
  219. "mem.x = 2\n"
  220. "print(mem.x)\n"
  221. "\n";
  222. Args* buffs = New_strBuff();
  223. char* pikaAsm = Parser_linesToAsm(buffs, line);
  224. printf("%s", pikaAsm);
  225. PikaObj* self = newRootObj("", New_PikaMain);
  226. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  227. int res = obj_getInt(globals, "mem.x");
  228. EXPECT_EQ(2, res);
  229. obj_deinit(self);
  230. // obj_deinit(globals);
  231. args_deinit(buffs);
  232. EXPECT_EQ(pikaMemNow(), 0);
  233. }
  234. TEST(VM, RET_instruct) {
  235. char* pikaAsm =
  236. "B1\n"
  237. "0 NUM 13\n"
  238. "0 RET \n"
  239. "0 NUM 2\n"
  240. "0 RET \n"
  241. "B0\n";
  242. PikaObj* self = New_TinyObj(NULL);
  243. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  244. Arg* return_arg = args_getArg(globals->list, "@rt");
  245. int num = arg_getInt(return_arg);
  246. EXPECT_EQ(num, 13);
  247. obj_deinit(self);
  248. // obj_deinit(globals);
  249. EXPECT_EQ(pikaMemNow(), 0);
  250. }
  251. TEST(VM, RUN_DEF) {
  252. char* pikaAsm =
  253. "B0\n"
  254. "0 DEF test()\n"
  255. "0 JMP 1\n"
  256. "B1\n"
  257. "0 NUM 1\n"
  258. "0 RET \n"
  259. "B0\n"
  260. "0 RUN test\n"
  261. "0 OUT a\n";
  262. PikaObj* self = New_TinyObj(NULL);
  263. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  264. int num = obj_getInt(globals, "a");
  265. EXPECT_EQ(num, 1);
  266. obj_deinit(self);
  267. // obj_deinit(globals);
  268. EXPECT_EQ(pikaMemNow(), 0);
  269. }
  270. TEST(VM, RUN_global) {
  271. char* pikaAsm =
  272. "B0\n"
  273. "0 NUM 1\n"
  274. "0 OUT a\n"
  275. "B0\n"
  276. "0 DEF test()\n"
  277. "0 JMP 1\n"
  278. "B1\n"
  279. "0 REF a\n"
  280. "0 RET \n"
  281. "B0\n"
  282. "0 RUN test\n"
  283. "0 OUT b\n";
  284. PikaObj* self = New_TinyObj(NULL);
  285. pikaVM_runAsm(self, pikaAsm);
  286. int a = obj_getInt(self, "a");
  287. int b = obj_getInt(self, "b");
  288. EXPECT_EQ(a, 1);
  289. EXPECT_EQ(b, 1);
  290. obj_deinit(self);
  291. // obj_deinit(globals);
  292. EXPECT_EQ(pikaMemNow(), 0);
  293. }
  294. TEST(VM, RUN_local_b) {
  295. char* pikaAsm =
  296. "B0\n"
  297. "0 NUM 1\n"
  298. "0 OUT a\n"
  299. "B0\n"
  300. "0 DEF test()\n"
  301. "0 JMP 1\n"
  302. "B1\n"
  303. "0 REF a\n"
  304. "0 OUT b\n"
  305. "1 REF b\n"
  306. "1 REF a\n"
  307. "0 OPT +\n"
  308. "0 RET \n"
  309. "B0\n"
  310. "0 RUN test\n"
  311. "0 OUT c\n";
  312. PikaObj* self = newRootObj("", New_BaseObj);
  313. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  314. int a = obj_getInt(globals, "a");
  315. int b = obj_getInt(globals, "b");
  316. int c = obj_getInt(globals, "c");
  317. EXPECT_EQ(a, 1);
  318. /* b is local, should not be exist in globals */
  319. EXPECT_EQ(b, -999999999);
  320. EXPECT_EQ(c, 2);
  321. obj_deinit(self);
  322. // obj_deinit(globals);
  323. EXPECT_EQ(pikaMemNow(), 0);
  324. }
  325. TEST(VM, RUN_DEF_add) {
  326. char* pikaAsm =
  327. "B0\n"
  328. "0 DEF add(a,b)\n"
  329. "0 JMP 1\n"
  330. "B1\n"
  331. "1 REF b\n"
  332. "1 REF a\n"
  333. "0 OPT +\n"
  334. "0 RET \n"
  335. "B0\n"
  336. "1 NUM 1\n"
  337. "1 NUM 2\n"
  338. "0 RUN add\n"
  339. "0 OUT c\n";
  340. PikaObj* self = newRootObj("", New_BaseObj);
  341. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  342. int a = obj_getInt(globals, "a");
  343. int b = obj_getInt(globals, "b");
  344. int c = obj_getInt(globals, "c");
  345. /* a is local, should not be exist in globals */
  346. EXPECT_EQ(a, -999999999);
  347. /* b is local, should not be exist in globals */
  348. EXPECT_EQ(b, -999999999);
  349. EXPECT_EQ(c, 3);
  350. obj_deinit(self);
  351. // obj_deinit(globals);
  352. EXPECT_EQ(pikaMemNow(), 0);
  353. }
  354. TEST(VM, run_def_add) {
  355. char* line =
  356. "def add(a, b):\n"
  357. " return a + b\n"
  358. "c = add(1,3)\n"
  359. "\n";
  360. Args* buffs = New_strBuff();
  361. char* pikaAsm = Parser_linesToAsm(buffs, line);
  362. printf("%s", pikaAsm);
  363. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  364. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  365. int c = args_getInt(globals->list, "c");
  366. obj_deinit(self);
  367. args_deinit(buffs);
  368. // obj_deinit(globals);
  369. ASSERT_FLOAT_EQ(c, 4);
  370. EXPECT_EQ(pikaMemNow(), 0);
  371. }
  372. TEST(VM, equ) {
  373. char* line =
  374. "if -1 == -1:\n"
  375. " a = 1\n"
  376. "\n";
  377. Args* buffs = New_strBuff();
  378. char* pikaAsm = Parser_linesToAsm(buffs, line);
  379. printf("%s", pikaAsm);
  380. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  381. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  382. int c = args_getInt(globals->list, "a");
  383. obj_deinit(self);
  384. args_deinit(buffs);
  385. // obj_deinit(globals);
  386. ASSERT_FLOAT_EQ(c, 1);
  387. EXPECT_EQ(pikaMemNow(), 0);
  388. }
  389. TEST(VM, if_elif) {
  390. char* line =
  391. "a = 2\n"
  392. "b = 0\n"
  393. "if a > 1:\n"
  394. " b = 1\n"
  395. "elif a > 0:\n"
  396. " b = 2\n"
  397. "\n";
  398. Args* buffs = New_strBuff();
  399. char* pikaAsm = Parser_linesToAsm(buffs, line);
  400. printf("%s", pikaAsm);
  401. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  402. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  403. int b = args_getInt(globals->list, "b");
  404. obj_deinit(self);
  405. args_deinit(buffs);
  406. ASSERT_FLOAT_EQ(b, 1);
  407. EXPECT_EQ(pikaMemNow(), 0);
  408. }
  409. TEST(VM, if_else) {
  410. char* line =
  411. "a = 0\n"
  412. "b = 0\n"
  413. "if a > 1:\n"
  414. " b = 1\n"
  415. "else:\n"
  416. " b = 2\n"
  417. "\n";
  418. Args* buffs = New_strBuff();
  419. char* pikaAsm = Parser_linesToAsm(buffs, line);
  420. printf("%s", pikaAsm);
  421. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  422. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  423. int b = args_getInt(globals->list, "b");
  424. obj_deinit(self);
  425. args_deinit(buffs);
  426. // obj_deinit(globals);
  427. ASSERT_FLOAT_EQ(b, 2);
  428. EXPECT_EQ(pikaMemNow(), 0);
  429. }
  430. TEST(VM, if_else_) {
  431. char* line =
  432. "a = 2\n"
  433. "b = 0\n"
  434. "if a > 1:\n"
  435. " b = 1\n"
  436. "else:\n"
  437. " b = 2\n"
  438. "\n";
  439. Args* buffs = New_strBuff();
  440. char* pikaAsm = Parser_linesToAsm(buffs, line);
  441. printf("%s", pikaAsm);
  442. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  443. VMParameters* globals = pikaVM_runAsm(self, pikaAsm);
  444. int b = args_getInt(globals->list, "b");
  445. obj_deinit(self);
  446. args_deinit(buffs);
  447. // obj_deinit(globals);
  448. ASSERT_FLOAT_EQ(b, 1);
  449. EXPECT_EQ(pikaMemNow(), 0);
  450. }
  451. TEST(VM, EST) {
  452. char* pikaAsm =
  453. "B0\n"
  454. "0 EST a\n"
  455. "0 OUT b\n";
  456. PikaObj* self = newRootObj("", New_BaseObj);
  457. pikaVM_runAsm(self, pikaAsm);
  458. int a = obj_getInt(self, "a");
  459. int b = obj_getInt(self, "b");
  460. /* a is local, should not be exist in globals */
  461. EXPECT_EQ(a, -999999999);
  462. /* b is local, should not be exist in globals */
  463. EXPECT_EQ(b, 0);
  464. obj_deinit(self);
  465. // obj_deinit(globals);
  466. EXPECT_EQ(pikaMemNow(), 0);
  467. }
  468. TEST(VM, RUN_AS) {
  469. char* pikaAsm =
  470. "B0\n"
  471. "0 RUN TinyObj\n"
  472. "0 OUT as\n"
  473. "B0\n"
  474. "0 RAS as\n"
  475. "B0\n"
  476. "0 NUM 1\n"
  477. "0 OUT x\n"
  478. "B0\n"
  479. "0 RAS $origin\n"
  480. "B0\n"
  481. "0 NUM 2\n"
  482. "0 OUT x\n";
  483. PikaObj* self = newRootObj("", New_PikaMain);
  484. pikaVM_runAsm(self, pikaAsm);
  485. PikaObj* as = obj_getObj(self, "as");
  486. int x_as_ = obj_getInt(as, "x");
  487. int x_as = obj_getInt(self, "as.x");
  488. int x_origin = obj_getInt(self, "x");
  489. EXPECT_EQ(x_as_, 1);
  490. EXPECT_EQ(x_as, 1);
  491. EXPECT_EQ(x_origin, 2);
  492. obj_deinit(self);
  493. // obj_deinit(globals);
  494. EXPECT_EQ(pikaMemNow(), 0);
  495. }
  496. TEST(VM, RUN_NEW) {
  497. char* pikaAsm =
  498. "B0\n"
  499. "0 RUN TinyObj\n"
  500. "0 OUT newObj\n"
  501. "0 NUM 1\n"
  502. "0 OUT newObj.x\n"
  503. "B0\n"
  504. "0 NEW newObj\n"
  505. "0 OUT outObj\n";
  506. PikaObj* self = newRootObj("", New_PikaMain);
  507. pikaVM_runAsm(self, pikaAsm);
  508. Arg* newObj = obj_getArg(self, "newObj");
  509. Arg* outObj = obj_getArg(self, "outObj");
  510. void* newObj_ptr = arg_getPtr(newObj);
  511. void* outObj_ptr = arg_getPtr(outObj);
  512. EXPECT_EQ(newObj_ptr, outObj_ptr);
  513. int x = obj_getInt(self, "outObj.x");
  514. ArgType newObj_type = arg_getType(newObj);
  515. ArgType outObj_type = arg_getType(outObj);
  516. EXPECT_EQ(x, 1);
  517. EXPECT_EQ(newObj_type, ARG_TYPE_OBJECT);
  518. EXPECT_EQ(outObj_type, ARG_TYPE_OBJECT);
  519. obj_deinit(self);
  520. // obj_deinit(globals);
  521. EXPECT_EQ(pikaMemNow(), 0);
  522. }
  523. TEST(VM, RUN_DEF_NEW) {
  524. char* pikaAsm =
  525. "B0\n"
  526. "0 DEF testnew()\n"
  527. "0 JMP 1\n"
  528. "B1\n"
  529. "0 RUN TinyObj\n"
  530. "0 OUT newObj\n"
  531. "B1\n"
  532. "0 NUM 1\n"
  533. "0 OUT newObj.x\n"
  534. "B1\n"
  535. "0 NEW newObj\n"
  536. "0 RET \n"
  537. "B1\n"
  538. "0 RET \n"
  539. "B0\n"
  540. "0 RUN testnew\n"
  541. "0 OUT outobj\n";
  542. PikaObj* self = newRootObj("", New_PikaMain);
  543. pikaVM_runAsm(self, pikaAsm);
  544. /* assert */
  545. PikaObj* outobj = obj_getObj(self, "outobj");
  546. int x = obj_getInt(outobj, "x");
  547. EXPECT_EQ(x, 1);
  548. /* deinit */
  549. obj_deinit(self);
  550. // obj_deinit(globals);
  551. EXPECT_EQ(pikaMemNow(), 0);
  552. }
  553. TEST(VM, class_x_1) {
  554. char* line =
  555. "class Test(TinyObj):\n"
  556. " x = 1\n"
  557. "test = Test()\n"
  558. "print('test.x: ' + str(test.x))\n"
  559. "\n";
  560. Args* buffs = New_strBuff();
  561. char* pikaAsm = Parser_linesToAsm(buffs, line);
  562. printf("%s", pikaAsm);
  563. PikaObj* self = newRootObj("", New_PikaMain);
  564. pikaVM_runAsm(self, pikaAsm);
  565. PikaObj* test = obj_getObj(self, "test");
  566. Arg* test_arg = obj_getArg(self, "test");
  567. ArgType test_arg_type = arg_getType(test_arg);
  568. EXPECT_EQ(test_arg_type, ARG_TYPE_OBJECT);
  569. int x = obj_getInt(test, "x");
  570. EXPECT_EQ(x, 1);
  571. obj_deinit(self);
  572. args_deinit(buffs);
  573. EXPECT_EQ(pikaMemNow(), 0);
  574. }
  575. TEST(VM, nag_a) {
  576. char* line =
  577. "a = 1\n"
  578. "print(-a)\n"
  579. "b = 0.5\n"
  580. "print(-b)\n";
  581. Args* buffs = New_strBuff();
  582. char* pikaAsm = Parser_linesToAsm(buffs, line);
  583. printf("%s", pikaAsm);
  584. PikaObj* self = newRootObj("", New_PikaMain);
  585. __platform_printf("BEGIN\r\n");
  586. pikaVM_runAsm(self, pikaAsm);
  587. /* assert */
  588. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  589. EXPECT_STREQ(log_buff[1], "-1\r\n");
  590. EXPECT_STREQ(log_buff[0], "-0.500000\r\n");
  591. /* deinit */
  592. obj_deinit(self);
  593. args_deinit(buffs);
  594. /* check mem */
  595. EXPECT_EQ(pikaMemNow(), 0);
  596. }
  597. TEST(InstructUnit, base) {
  598. __platform_printf("BEGIN\r\n");
  599. InstructUnit ins_unit;
  600. instructUnit_init(&ins_unit);
  601. instructUnit_setBlockDeepth(&ins_unit, 2);
  602. instructUnit_setIsNewLine(&ins_unit, 1);
  603. instructUnit_setInvokeDeepth(&ins_unit, 3);
  604. instructUnit_setInstruct(&ins_unit, (Instruct)4);
  605. instructUnit_setConstPoolIndex(&ins_unit, 12);
  606. EXPECT_EQ(instructUnit_getBlockDeepth(&ins_unit), 2);
  607. EXPECT_EQ(instructUnit_getIsNewLine(&ins_unit), 1);
  608. EXPECT_EQ(instructUnit_getInvokeDeepth(&ins_unit), 3);
  609. EXPECT_EQ(instructUnit_getInstruct(&ins_unit), 4);
  610. EXPECT_EQ(instructUnit_getConstPoolIndex(&ins_unit), 12);
  611. instructUnit_print(&ins_unit);
  612. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  613. EXPECT_STREQ(log_buff[1], "B2\r\n");
  614. EXPECT_STREQ(log_buff[0], "3 OUT #12\r\n");
  615. EXPECT_EQ(pikaMemNow(), 0);
  616. }
  617. TEST(ConstPool, get) {
  618. __platform_printf("BEGIN\r\n");
  619. ConstPool cp;
  620. constPool_init(&cp);
  621. char* next_content;
  622. uint16_t next_offset;
  623. next_offset = constPool_getLastOffset(&cp);
  624. constPool_append(&cp, "hello");
  625. next_content = constPool_getByOffset(&cp, next_offset);
  626. EXPECT_STREQ(next_content, "hello");
  627. next_offset = constPool_getLastOffset(&cp);
  628. constPool_append(&cp, "world");
  629. next_content = constPool_getByOffset(&cp, next_offset);
  630. EXPECT_STREQ(next_content, "world");
  631. char* first = constPool_getNow(&cp);
  632. char* second = constPool_getNext(&cp);
  633. char* third = constPool_getNext(&cp);
  634. char* forth = constPool_getNext(&cp);
  635. EXPECT_STREQ(first, "");
  636. EXPECT_STREQ(second, "hello");
  637. EXPECT_STREQ(third, "world");
  638. EXPECT_EQ((uintptr_t)forth, (uintptr_t)NULL);
  639. EXPECT_STREQ(constPool_getByIndex(&cp, 0), "");
  640. EXPECT_STREQ(constPool_getByIndex(&cp, 1), "hello");
  641. EXPECT_STREQ(constPool_getByIndex(&cp, 2), "world");
  642. EXPECT_EQ((uintptr_t)constPool_getByIndex(&cp, 3), (uintptr_t)NULL);
  643. constPool_print(&cp);
  644. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  645. EXPECT_STREQ(log_buff[1], "1: hello\r\n");
  646. EXPECT_STREQ(log_buff[0], "7: world\r\n");
  647. constPool_deinit(&cp);
  648. EXPECT_EQ(pikaMemNow(), 0);
  649. }
  650. TEST(InstructArray, set) {
  651. __platform_printf("BEGIN\r\n");
  652. InstructUnit ins_unit;
  653. instructUnit_init(&ins_unit);
  654. instructUnit_setBlockDeepth(&ins_unit, 2);
  655. instructUnit_setIsNewLine(&ins_unit, 1);
  656. instructUnit_setInvokeDeepth(&ins_unit, 3);
  657. instructUnit_setInstruct(&ins_unit, (Instruct)4);
  658. instructUnit_setConstPoolIndex(&ins_unit, 12);
  659. InstructArray ins_array;
  660. instructArray_init(&ins_array);
  661. instructArray_append(&ins_array, &ins_unit);
  662. instructArray_print(&ins_array);
  663. instructArray_deinit(&ins_array);
  664. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  665. EXPECT_STREQ(log_buff[1], "B2\r\n");
  666. EXPECT_STREQ(log_buff[0], "3 OUT #12\r\n");
  667. EXPECT_EQ(pikaMemNow(), 0);
  668. }
  669. TEST(VM, bytecode_jjcc) {
  670. char* line = "a = (1 + 1.1) * 3 - 2 /4.0";
  671. Args* buffs = New_strBuff();
  672. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  673. printf("%s", pikaAsm);
  674. ByteCodeFrame byte_frame;
  675. byteCodeFrame_init(&byte_frame);
  676. byteCodeFrame_appendFromAsm(&byte_frame, pikaAsm);
  677. byteCodeFrame_print(&byte_frame);
  678. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  679. // pikaVM_runAsm(self, pikaAsm);
  680. pikaVM_runByteCodeFrame(self, &byte_frame);
  681. float res = obj_getFloat(self, "a");
  682. obj_deinit(self);
  683. args_deinit(buffs);
  684. byteCodeFrame_deinit(&byte_frame);
  685. ASSERT_FLOAT_EQ(res, 5.8);
  686. EXPECT_EQ(pikaMemNow(), 0);
  687. }
  688. TEST(VM, WHILE_byte) {
  689. pikaMemInfo.heapUsedMax = 0;
  690. Args* buffs = New_strBuff();
  691. char* lines =(char *)
  692. "a = 1\n"
  693. "b = 0\n"
  694. "while a:\n"
  695. " b = 1\n"
  696. " a = 0\n"
  697. "\n";
  698. printf("%s", lines);
  699. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  700. printf("%s", pikaAsm);
  701. pikaMemInfo.heapUsedMax = 0;
  702. PikaObj* self = newRootObj("root", New_TinyObj);
  703. pikaVM_run(self, lines);
  704. EXPECT_EQ(obj_getInt(self, "a"), 0);
  705. EXPECT_EQ(obj_getInt(self, "b"), 1);
  706. args_deinit(buffs);
  707. obj_deinit(self);
  708. EXPECT_EQ(pikaMemNow(), 0);
  709. }
  710. TEST(VM, for_break_byte) {
  711. pikaMemInfo.heapUsedMax = 0;
  712. Args* buffs = New_strBuff();
  713. char* lines =(char *)
  714. "a = 0\n"
  715. "for i in range(0, 10):\n"
  716. " if i == 5:\n"
  717. " break\n"
  718. " a = a + i\n"
  719. "\n";
  720. printf("%s", lines);
  721. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  722. printf("%s", pikaAsm);
  723. pikaMemInfo.heapUsedMax = 0;
  724. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  725. pikaVM_run(self, lines);
  726. /* assert */
  727. int a = obj_getInt(self, "a");
  728. EXPECT_EQ(a, 10);
  729. /* deinit */
  730. args_deinit(buffs);
  731. obj_deinit(self);
  732. EXPECT_EQ(pikaMemNow(), 0);
  733. }
  734. TEST(VM, load_static_bytes) {
  735. const uint8_t bytes[] = {
  736. 0x08, 0x02, 0x00, 0x00, /* instruct array size */
  737. 0x00, 0x82, 0x01, 0x00, 0x00, 0x04, 0x0d, 0x00, 0x00, 0x83, 0x11, 0x00,
  738. 0x00, 0x04, 0x16, 0x00, 0x00, 0x85, 0x1b, 0x00, 0x00, 0x04, 0x1d, 0x00,
  739. 0x20, 0x85, 0x1b, 0x00, 0x20, 0x01, 0x0d, 0x00, 0x10, 0x02, 0x28, 0x00,
  740. 0x00, 0x02, 0x2e, 0x00, 0x00, 0x04, 0x33, 0x00, 0x00, 0x01, 0x37, 0x00,
  741. 0x00, 0x01, 0x3b, 0x00, 0x00, 0x01, 0x3f, 0x00, 0x00, 0x04, 0x43, 0x00,
  742. 0x00, 0x04, 0x4a, 0x00, 0x00, 0x04, 0x51, 0x00, 0x00, 0x82, 0x58, 0x00,
  743. 0x00, 0x04, 0x65, 0x00, 0x00, 0x0d, 0x65, 0x00, 0x00, 0x07, 0x67, 0x00,
  744. 0x11, 0x81, 0x0d, 0x00, 0x11, 0x05, 0x1b, 0x00, 0x01, 0x08, 0x69, 0x00,
  745. 0x01, 0x07, 0x6c, 0x00, 0x02, 0x8e, 0x6c, 0x00, 0x11, 0x81, 0x1d, 0x00,
  746. 0x11, 0x05, 0x1b, 0x00, 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00,
  747. 0x12, 0x81, 0x6e, 0x00, 0x12, 0x01, 0x65, 0x00, 0x02, 0x02, 0x76, 0x00,
  748. 0x02, 0x04, 0x16, 0x00, 0x02, 0x85, 0x6c, 0x00, 0x02, 0x04, 0x1d, 0x00,
  749. 0x01, 0x8b, 0x6c, 0x00, 0x11, 0x01, 0x1d, 0x00, 0x11, 0x05, 0x6c, 0x00,
  750. 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00, 0x12, 0x81, 0x16, 0x00,
  751. 0x12, 0x03, 0x7e, 0x00, 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00,
  752. 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00,
  753. 0x03, 0x04, 0x85, 0x00, 0x03, 0x82, 0x85, 0x00, 0x03, 0x85, 0x1b, 0x00,
  754. 0x03, 0x04, 0x1d, 0x00, 0x02, 0x8b, 0x6c, 0x00, 0x12, 0x01, 0x16, 0x00,
  755. 0x12, 0x03, 0x8a, 0x00, 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00,
  756. 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00,
  757. 0x03, 0x04, 0x8a, 0x00, 0x03, 0x85, 0x67, 0x00, 0x03, 0x04, 0x1d, 0x00,
  758. 0x02, 0x8b, 0x6c, 0x00, 0x12, 0x01, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00,
  759. 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x6e, 0x00,
  760. 0x13, 0x01, 0x65, 0x00, 0x03, 0x02, 0x76, 0x00, 0x03, 0x04, 0x8f, 0x00,
  761. 0x03, 0x85, 0x67, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x01, 0x8b, 0x6c, 0x00,
  762. 0x11, 0x01, 0x1d, 0x00, 0x11, 0x05, 0x67, 0x00, 0x01, 0x08, 0x69, 0x00,
  763. 0x01, 0x07, 0x6c, 0x00, 0x12, 0x81, 0x16, 0x00, 0x12, 0x03, 0x8a, 0x00,
  764. 0x02, 0x08, 0x69, 0x00, 0x02, 0x07, 0x6c, 0x00, 0x03, 0x82, 0x8a, 0x00,
  765. 0x03, 0x07, 0x6c, 0x00, 0x14, 0x81, 0x6e, 0x00, 0x14, 0x01, 0x65, 0x00,
  766. 0x04, 0x02, 0x76, 0x00, 0x04, 0x04, 0x85, 0x00, 0x04, 0x82, 0x85, 0x00,
  767. 0x03, 0x85, 0x1b, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x02, 0x8b, 0x6c, 0x00,
  768. 0x12, 0x01, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00, 0x02, 0x08, 0x69, 0x00,
  769. 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x6e, 0x00, 0x13, 0x01, 0x65, 0x00,
  770. 0x03, 0x02, 0x76, 0x00, 0x03, 0x04, 0x85, 0x00, 0x03, 0x85, 0x99, 0x00,
  771. 0x03, 0x04, 0x1d, 0x00, 0x01, 0x8b, 0x6c, 0x00, 0x11, 0x01, 0x1d, 0x00,
  772. 0x11, 0x05, 0x99, 0x00, 0x01, 0x08, 0x69, 0x00, 0x01, 0x07, 0x6c, 0x00,
  773. 0x12, 0x81, 0x16, 0x00, 0x12, 0x03, 0x8f, 0x00, 0x02, 0x08, 0x69, 0x00,
  774. 0x02, 0x07, 0x6c, 0x00, 0x13, 0x81, 0x9b, 0x00, 0x23, 0x01, 0x6e, 0x00,
  775. 0x23, 0x01, 0x65, 0x00, 0x13, 0x02, 0x76, 0x00, 0x03, 0x08, 0xa2, 0x00,
  776. 0x03, 0x07, 0x6c, 0x00, 0x04, 0x82, 0x85, 0x00, 0x14, 0x81, 0x6e, 0x00,
  777. 0x14, 0x01, 0x65, 0x00, 0x24, 0x01, 0x9b, 0x00, 0x24, 0x01, 0x8f, 0x00,
  778. 0x14, 0x08, 0xa4, 0x00, 0x14, 0x03, 0x6e, 0x00, 0x04, 0x02, 0xa6, 0x00,
  779. 0x03, 0x85, 0x1b, 0x00, 0x03, 0x04, 0x1d, 0x00, 0x00, 0x86, 0xae, 0x00,
  780. 0x00, 0x8c, 0x33, 0x00, /* instruct array */
  781. 0xb1, 0x00, 0x00, 0x00, /* const pool size */
  782. 0x00, 0x5f, 0x5f, 0x63, 0x61, 0x6c, 0x6c, 0x73, 0x2e, 0x6c, 0x65, 0x6e,
  783. 0x00, 0x6c, 0x65, 0x6e, 0x00, 0x6e, 0x6f, 0x6e, 0x65, 0x00, 0x6d, 0x6f,
  784. 0x64, 0x65, 0x00, 0x30, 0x00, 0x69, 0x6e, 0x66, 0x6f, 0x5f, 0x69, 0x6e,
  785. 0x64, 0x65, 0x78, 0x00, 0x72, 0x61, 0x6e, 0x67, 0x65, 0x00, 0x69, 0x74,
  786. 0x65, 0x72, 0x00, 0x5f, 0x6c, 0x30, 0x00, 0x5f, 0x72, 0x31, 0x00, 0x5f,
  787. 0x72, 0x32, 0x00, 0x5f, 0x72, 0x33, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x61,
  788. 0x31, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x61, 0x32, 0x00, 0x5f, 0x6c, 0x30,
  789. 0x2e, 0x61, 0x33, 0x00, 0x5f, 0x6c, 0x30, 0x2e, 0x5f, 0x5f, 0x6e, 0x65,
  790. 0x78, 0x74, 0x5f, 0x5f, 0x00, 0x69, 0x00, 0x32, 0x00, 0x3d, 0x3d, 0x00,
  791. 0x31, 0x00, 0x5f, 0x5f, 0x63, 0x61, 0x6c, 0x6c, 0x73, 0x00, 0x5f, 0x5f,
  792. 0x67, 0x65, 0x74, 0x5f, 0x5f, 0x00, 0x61, 0x6c, 0x77, 0x61, 0x79, 0x73,
  793. 0x00, 0x74, 0x6f, 0x64, 0x6f, 0x00, 0x77, 0x68, 0x65, 0x6e, 0x00, 0x70,
  794. 0x65, 0x72, 0x69, 0x6f, 0x64, 0x5f, 0x6d, 0x73, 0x00, 0x33, 0x00, 0x5f,
  795. 0x5f, 0x74, 0x69, 0x63, 0x6b, 0x00, 0x3e, 0x00, 0x2b, 0x00, 0x5f, 0x5f,
  796. 0x73, 0x65, 0x74, 0x5f, 0x5f, 0x00, 0x2d, 0x31, 0x00, /* const pool */
  797. };
  798. ByteCodeFrame bytecode_frame = {.const_pool =
  799. {
  800. .arg_buff = NULL,
  801. .content_offset_now = 0,
  802. .size = 0,
  803. .content_start = NULL,
  804. .output_redirect_fun = NULL,
  805. .output_f = NULL,
  806. },
  807. .instruct_array = {
  808. .arg_buff = NULL,
  809. .content_offset_now = 0,
  810. .size = 0,
  811. .content_start = NULL,
  812. .output_redirect_fun = NULL,
  813. .output_f = NULL,
  814. }};
  815. byteCodeFrame_loadByteCode(&bytecode_frame, (uint8_t*)bytes);
  816. byteCodeFrame_print(&bytecode_frame);
  817. EXPECT_EQ(instructArray_getSize(&(bytecode_frame.instruct_array)), 520);
  818. EXPECT_EQ(bytecode_frame.const_pool.size, 177);
  819. byteCodeFrame_deinit(&bytecode_frame);
  820. EXPECT_EQ(pikaMemNow(), 0);
  821. }
  822. TEST(VM, multi_jian) {
  823. pikaMemInfo.heapUsedMax = 0;
  824. Args* buffs = New_strBuff();
  825. char* lines = "a = (3-4) - 4\n";
  826. printf("%s", lines);
  827. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  828. printf("%s", pikaAsm);
  829. pikaMemInfo.heapUsedMax = 0;
  830. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  831. pikaVM_run(self, lines);
  832. /* assert */
  833. int a = obj_getInt(self, "a");
  834. EXPECT_EQ(a, -5);
  835. /* deinit */
  836. args_deinit(buffs);
  837. obj_deinit(self);
  838. EXPECT_EQ(pikaMemNow(), 0);
  839. }
  840. TEST(VM, _3_p_3) {
  841. char* line = "-3+3";
  842. Args* buffs = New_strBuff();
  843. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  844. printf("%s", pikaAsm);
  845. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  846. __platform_printf("BEGIN\r\n");
  847. pikaVM_runAsm(self, pikaAsm);
  848. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  849. EXPECT_STREQ(log_buff[0], "0\r\n");
  850. obj_deinit(self);
  851. args_deinit(buffs);
  852. EXPECT_EQ(pikaMemNow(), 0);
  853. }
  854. TEST(VM, _3_3) {
  855. char* line = "-3-3";
  856. Args* buffs = New_strBuff();
  857. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  858. printf("%s", pikaAsm);
  859. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  860. __platform_printf("BEGIN\r\n");
  861. pikaVM_runAsm(self, pikaAsm);
  862. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  863. EXPECT_STREQ(log_buff[0], "-6\r\n");
  864. obj_deinit(self);
  865. args_deinit(buffs);
  866. EXPECT_EQ(pikaMemNow(), 0);
  867. }
  868. TEST(VM, a_3) {
  869. char* line = "a-3";
  870. Args* buffs = New_strBuff();
  871. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  872. printf("%s", pikaAsm);
  873. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  874. __platform_printf("BEGIN\r\n");
  875. pikaVM_runAsm(self, pikaAsm);
  876. EXPECT_STREQ(log_buff[3], "BEGIN\r\n");
  877. EXPECT_STREQ(log_buff[2], "NameError: name 'a' is not defined\r\n");
  878. obj_deinit(self);
  879. args_deinit(buffs);
  880. EXPECT_EQ(pikaMemNow(), 0);
  881. }
  882. TEST(VM, hex_str) {
  883. char* line = "a = '\\x33\\x35'";
  884. Args* buffs = New_strBuff();
  885. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  886. printf("%s", pikaAsm);
  887. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  888. __platform_printf("BEGIN\r\n");
  889. pikaVM_runAsm(self, pikaAsm);
  890. /* collect */
  891. char* a = obj_getStr(self, "a");
  892. /* assert */
  893. EXPECT_STREQ(a, "35");
  894. /* deinit */
  895. obj_deinit(self);
  896. args_deinit(buffs);
  897. EXPECT_EQ(pikaMemNow(), 0);
  898. }
  899. TEST(VM, hex_bytes) {
  900. char* line =
  901. "a = b'\\x03\\x05'\n"
  902. "a\n";
  903. Args* buffs = New_strBuff();
  904. char* pikaAsm = Parser_linesToAsm(buffs, line);
  905. printf("%s", pikaAsm);
  906. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  907. __platform_printf("BEGIN\r\n");
  908. pikaVM_runAsm(self, pikaAsm);
  909. /* collect */
  910. /* assert */
  911. EXPECT_STREQ(log_buff[0], "b'\\x03\\x05'\r\n");
  912. /* deinit */
  913. obj_deinit(self);
  914. args_deinit(buffs);
  915. EXPECT_EQ(pikaMemNow(), 0);
  916. }
  917. TEST(VM, bytes_equ) {
  918. char* line =
  919. "a = b'\\x03\\x05'\n"
  920. "b = b'\\x03\\x05'\n"
  921. "if a == b:\n"
  922. " c = 1\n"
  923. "b = b'\\x02\\x05'\n"
  924. "if a != b:\n"
  925. " d = 1\n";
  926. Args* buffs = New_strBuff();
  927. char* pikaAsm = Parser_linesToAsm(buffs, line);
  928. printf("%s", pikaAsm);
  929. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  930. __platform_printf("BEGIN\r\n");
  931. pikaVM_runAsm(self, pikaAsm);
  932. /* collect */
  933. /* assert */
  934. int c = obj_getInt(self, "c");
  935. int d = obj_getInt(self, "d");
  936. EXPECT_EQ(c, 1);
  937. EXPECT_EQ(d, 1);
  938. /* deinit */
  939. obj_deinit(self);
  940. args_deinit(buffs);
  941. EXPECT_EQ(pikaMemNow(), 0);
  942. }
  943. #if PIKA_BUILTIN_STRUCT_ENABLE
  944. TEST(VM, tuple_literal) {
  945. char* line = "(1,2,3)\n";
  946. Args* buffs = New_strBuff();
  947. char* pikaAsm = Parser_linesToAsm(buffs, line);
  948. printf("%s", pikaAsm);
  949. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  950. __platform_printf("BEGIN\r\n");
  951. pikaVM_runAsm(self, pikaAsm);
  952. /* collect */
  953. /* assert */
  954. EXPECT_STREQ(log_buff[0], "(1, 2, 3)\r\n");
  955. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  956. /* deinit */
  957. obj_deinit(self);
  958. args_deinit(buffs);
  959. EXPECT_EQ(pikaMemNow(), 0);
  960. }
  961. #endif
  962. TEST(VM, dvd_opt) {
  963. char* line = "a = 10%4\n";
  964. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  965. obj_run(self, line);
  966. /* collect */
  967. int a = obj_getInt(self, "a");
  968. /* assert */
  969. EXPECT_EQ(a, 2);
  970. /* deinit */
  971. obj_deinit(self);
  972. EXPECT_EQ(pikaMemNow(), 0);
  973. }
  974. TEST(VM, del_) {
  975. char* line =
  976. "a = 1\n"
  977. "print(a)\n"
  978. "del a\n"
  979. "print(a)\n";
  980. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  981. obj_run(self, line);
  982. /* collect */
  983. /* assert */
  984. EXPECT_EQ(obj_isArgExist(self, "a"), 0);
  985. /* deinit */
  986. obj_deinit(self);
  987. EXPECT_EQ(pikaMemNow(), 0);
  988. }
  989. TEST(VM, is) {
  990. char* line =
  991. "a = 1\n"
  992. "b = a\n"
  993. "res1 = a is b\n"
  994. "res2 = 1 is 1\n"
  995. "c = TinyObj()\n"
  996. "d = c\n"
  997. "res3 = c is d\n"
  998. "res4 = c is None\n";
  999. ;
  1000. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1001. obj_run(self, line);
  1002. /* collect */
  1003. /* assert */
  1004. EXPECT_EQ(obj_getInt(self, "res1"), 1);
  1005. EXPECT_EQ(obj_getInt(self, "res2"), 1);
  1006. EXPECT_EQ(obj_getInt(self, "res3"), 1);
  1007. EXPECT_EQ(obj_getInt(self, "res4"), 0);
  1008. /* deinit */
  1009. obj_deinit(self);
  1010. EXPECT_EQ(pikaMemNow(), 0);
  1011. }
  1012. TEST(VM, in) {
  1013. char* line =
  1014. "res1 = 'a' in 'aaa'\n"
  1015. "res2 = 'aaa' in 'a'\n";
  1016. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1017. obj_run(self, line);
  1018. /* collect */
  1019. /* assert */
  1020. EXPECT_EQ(obj_getInt(self, "res1"), 1);
  1021. EXPECT_EQ(obj_getInt(self, "res2"), 0);
  1022. /* deinit */
  1023. obj_deinit(self);
  1024. EXPECT_EQ(pikaMemNow(), 0);
  1025. }
  1026. #if !PIKA_NANO_ENABLE
  1027. TEST(VM, assert) {
  1028. char* line =
  1029. "assert True\n"
  1030. "assert 1 == 1, 'testparser'\n"
  1031. "res1 = 0\n"
  1032. "try:\n"
  1033. " assert False\n"
  1034. "except:\n"
  1035. " res1 = 1\n";
  1036. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1037. obj_run(self, line);
  1038. /* collect */
  1039. int res1 = obj_getInt(self, "res1");
  1040. /* assert */
  1041. EXPECT_EQ(res1, 1);
  1042. /* deinit */
  1043. obj_deinit(self);
  1044. EXPECT_EQ(pikaMemNow(), 0);
  1045. }
  1046. #endif
  1047. TEST(VM, issue_I5LHJG) {
  1048. char* line =
  1049. "res = (a, 2, 3)\n"
  1050. "[a, 2, 3]\n"
  1051. "{'x': a, 'y': 2, 'z': 3}\n";
  1052. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1053. obj_run(self, line);
  1054. /* collect */
  1055. /* assert */
  1056. EXPECT_EQ(obj_isArgExist(self, "res"), 0);
  1057. /* deinit */
  1058. obj_deinit(self);
  1059. EXPECT_EQ(pikaMemNow(), 0);
  1060. }
  1061. TEST(VM, vars_runtime) {
  1062. char* line =
  1063. "def testvars(a, *b):\n"
  1064. " sum = 0\n"
  1065. " for i in b:\n"
  1066. " sum += i\n"
  1067. " return a * sum\n"
  1068. "res = testvars(6, 2, 3, 4, 5)\n";
  1069. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1070. obj_run(self, line);
  1071. /* collect */
  1072. int res = obj_getInt(self, "res");
  1073. /* assert */
  1074. EXPECT_EQ(res, 84);
  1075. /* deinit */
  1076. obj_deinit(self);
  1077. EXPECT_EQ(pikaMemNow(), 0);
  1078. }
  1079. #if PIKA_BUILTIN_STRUCT_ENABLE
  1080. TEST(VM, list_add) {
  1081. char* line = "[1, 2, 3] + [4, 5, 6]";
  1082. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1083. obj_run(self, line);
  1084. /* collect */
  1085. /* assert */
  1086. EXPECT_STREQ(log_buff[0], "[1, 2, 3, 4, 5, 6]\r\n");
  1087. /* deinit */
  1088. obj_deinit(self);
  1089. EXPECT_EQ(pikaMemNow(), 0);
  1090. }
  1091. #endif
  1092. TEST(VM, science_num) {
  1093. char* line =
  1094. "a = 1.0e-3\n"
  1095. "b = 2e-5\n"
  1096. "c = -3e-5\n"
  1097. "d = 0.4e2\n";
  1098. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1099. obj_run(self, line);
  1100. /* collect */
  1101. double a = obj_getFloat(self, "a");
  1102. double b = obj_getFloat(self, "b");
  1103. double c = obj_getFloat(self, "c");
  1104. double d = obj_getFloat(self, "d");
  1105. /* assert */
  1106. #if PIKA_FLOAT_TYPE_DOUBLE
  1107. EXPECT_DOUBLE_EQ(a, 1.0e-3);
  1108. EXPECT_DOUBLE_EQ(b, 2.0e-5);
  1109. EXPECT_DOUBLE_EQ(c, -3e-5);
  1110. EXPECT_DOUBLE_EQ(d, 0.4e2);
  1111. #else
  1112. EXPECT_FLOAT_EQ(a, 1.0e-3);
  1113. EXPECT_FLOAT_EQ(b, 2.0e-5);
  1114. EXPECT_FLOAT_EQ(c, -3e-5);
  1115. EXPECT_FLOAT_EQ(d, 0.4e2);
  1116. #endif
  1117. /* deinit */
  1118. obj_deinit(self);
  1119. EXPECT_EQ(pikaMemNow(), 0);
  1120. }
  1121. TEST(VM, issue_I5OJQB) {
  1122. char* line = "s = '\\\\'";
  1123. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1124. obj_run(self, line);
  1125. /* collect */
  1126. char* s = obj_getStr(self, "s");
  1127. /* assert */
  1128. EXPECT_STREQ(s, "\\");
  1129. /* deinit */
  1130. obj_deinit(self);
  1131. EXPECT_EQ(pikaMemNow(), 0);
  1132. }
  1133. #if !PIKA_NANO_ENABLE
  1134. TEST(vm, keyword_2) {
  1135. char* line =
  1136. "def test(a, b):\n"
  1137. " print(a, b)\n"
  1138. "test(a=1, b= 2)";
  1139. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1140. obj_run(self, line);
  1141. /* collect */
  1142. /* assert */
  1143. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1144. /* deinit */
  1145. obj_deinit(self);
  1146. EXPECT_EQ(pikaMemNow(), 0);
  1147. }
  1148. TEST(vm, keyword_3) {
  1149. char* line =
  1150. "def test(**keys):\n"
  1151. " print(keys['a'], keys['b'])\n"
  1152. "test(a=1, b= 2)";
  1153. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1154. obj_run(self, line);
  1155. /* collect */
  1156. /* assert */
  1157. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1158. /* deinit */
  1159. obj_deinit(self);
  1160. EXPECT_EQ(pikaMemNow(), 0);
  1161. }
  1162. TEST(vm, keyword_4) {
  1163. char* line =
  1164. "def test(a, b, **keys):\n"
  1165. " print(keys['c'], keys['d'], a, b)\n"
  1166. "test(3, 4, c=1, d= 2)";
  1167. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1168. obj_run(self, line);
  1169. /* collect */
  1170. /* assert */
  1171. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  1172. /* deinit */
  1173. obj_deinit(self);
  1174. EXPECT_EQ(pikaMemNow(), 0);
  1175. }
  1176. TEST(vm, class_keyword) {
  1177. char* line =
  1178. "class Test:\n"
  1179. " def __init__(self, a, b = 1, c =2):\n"
  1180. " print(a, b, c)\n"
  1181. "t = Test(0, b = 3)\n";
  1182. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1183. obj_run(self, line);
  1184. /* collect */
  1185. /* assert */
  1186. EXPECT_STREQ(log_buff[0], "0 3 2\r\n");
  1187. /* deinit */
  1188. obj_deinit(self);
  1189. EXPECT_EQ(pikaMemNow(), 0);
  1190. }
  1191. TEST(vm, class_keyword_mqtt) {
  1192. char* line =
  1193. "class MQTT:\n"
  1194. " def __init__(self,\n"
  1195. " ip: str,\n"
  1196. " port=1883,\n"
  1197. " clinetID='mac',\n"
  1198. " username='',\n"
  1199. " password='',\n"
  1200. " version='3.1.1',\n"
  1201. " ca='',\n"
  1202. " keepalive=60):\n"
  1203. " print('ip:', ip)\n"
  1204. " print('port:', port)\n"
  1205. "c = MQTT('test')\n";
  1206. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  1207. obj_run(self, line);
  1208. /* collect */
  1209. /* assert */
  1210. EXPECT_STREQ(log_buff[1], "ip: test\r\n");
  1211. EXPECT_STREQ(log_buff[0], "port: 1883\r\n");
  1212. /* deinit */
  1213. obj_deinit(self);
  1214. EXPECT_EQ(pikaMemNow(), 0);
  1215. }
  1216. TEST(vm, vars_keyward) {
  1217. /* init */
  1218. 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/BuiltIn/function.py");
  1225. /* collect */
  1226. /* assert */
  1227. EXPECT_STREQ(log_buff[1], "1 2 3 4 5 10\r\n");
  1228. EXPECT_STREQ(log_buff[0], "1 2 3 4 5 12\r\n");
  1229. /* deinit */
  1230. obj_deinit(pikaMain);
  1231. EXPECT_EQ(pikaMemNow(), 0);
  1232. }
  1233. TEST(vm, cb_1) {
  1234. /* init */
  1235. pikaMemInfo.heapUsedMax = 0;
  1236. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1237. extern unsigned char pikaModules_py_a[];
  1238. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1239. /* run */
  1240. __platform_printf("BEGIN\r\n");
  1241. pikaVM_runSingleFile(pikaMain, "../../examples/Callback/test1.py");
  1242. /* collect */
  1243. /* assert */
  1244. EXPECT_STREQ(log_buff[3], "__init__\r\n");
  1245. EXPECT_STREQ(log_buff[1], "a\r\n");
  1246. EXPECT_STREQ(log_buff[0], "b\r\n");
  1247. /* deinit */
  1248. obj_deinit(pikaMain);
  1249. EXPECT_EQ(pikaMemNow(), 0);
  1250. }
  1251. TEST(vm, cb_2) {
  1252. /* init */
  1253. pikaMemInfo.heapUsedMax = 0;
  1254. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1255. extern unsigned char pikaModules_py_a[];
  1256. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1257. /* run */
  1258. __platform_printf("BEGIN\r\n");
  1259. pikaVM_runSingleFile(pikaMain, "../../examples/Callback/test2.py");
  1260. /* collect */
  1261. /* assert */
  1262. EXPECT_STREQ(log_buff[4], "__init__\r\n");
  1263. EXPECT_STREQ(log_buff[2], "a\r\n");
  1264. EXPECT_STREQ(log_buff[1], "b\r\n");
  1265. EXPECT_STREQ(log_buff[0], "ppp\r\n");
  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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. #endif
  2145. #if !PIKA_NANO_ENABLE
  2146. TEST(vm, fn_pos_kw) {
  2147. /* init */
  2148. pikaMemInfo.heapUsedMax = 0;
  2149. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2150. /* run */
  2151. __platform_printf("BEGIN\r\n");
  2152. obj_run(pikaMain,
  2153. "def test(a, b):\n"
  2154. " print(a, b)\n"
  2155. "test(b=1,a=2)\n");
  2156. /* collect */
  2157. /* assert */
  2158. EXPECT_STREQ(log_buff[0], "2 1\r\n");
  2159. /* deinit */
  2160. obj_deinit(pikaMain);
  2161. EXPECT_EQ(pikaMemNow(), 0);
  2162. }
  2163. TEST(vm, fn_pos_kw2) {
  2164. /* init */
  2165. pikaMemInfo.heapUsedMax = 0;
  2166. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2167. /* run */
  2168. __platform_printf("BEGIN\r\n");
  2169. obj_run(pikaMain,
  2170. "def test(a, b):\n"
  2171. " print(a, b)\n"
  2172. "test(1, b=2)\n"
  2173. "test(1, a=2)\n");
  2174. /* collect */
  2175. /* assert */
  2176. EXPECT_STREQ(log_buff[1], "1 2\r\n");
  2177. EXPECT_STREQ(log_buff[0], "2 1\r\n");
  2178. /* deinit */
  2179. obj_deinit(pikaMain);
  2180. EXPECT_EQ(pikaMemNow(), 0);
  2181. }
  2182. TEST(vm, fn_star) {
  2183. /* init */
  2184. pikaMemInfo.heapUsedMax = 0;
  2185. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2186. /* run */
  2187. __platform_printf("BEGIN\r\n");
  2188. obj_run(pikaMain,
  2189. "a = (1,2,3)\n"
  2190. "print(*a)\n");
  2191. /* collect */
  2192. /* assert */
  2193. EXPECT_STREQ(log_buff[0], "1 2 3\r\n");
  2194. /* deinit */
  2195. obj_deinit(pikaMain);
  2196. EXPECT_EQ(pikaMemNow(), 0);
  2197. }
  2198. TEST(vm, fn_star_pos) {
  2199. /* init */
  2200. pikaMemInfo.heapUsedMax = 0;
  2201. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2202. /* run */
  2203. __platform_printf("BEGIN\r\n");
  2204. obj_run(pikaMain,
  2205. "a = (1,2,3)\n"
  2206. "print(*a, 4)\n");
  2207. /* collect */
  2208. /* assert */
  2209. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  2210. /* deinit */
  2211. obj_deinit(pikaMain);
  2212. EXPECT_EQ(pikaMemNow(), 0);
  2213. }
  2214. TEST(vm, fn_star_pos_2) {
  2215. /* init */
  2216. pikaMemInfo.heapUsedMax = 0;
  2217. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2218. /* run */
  2219. __platform_printf("BEGIN\r\n");
  2220. obj_run(pikaMain,
  2221. "a = (1,2,3)\n"
  2222. "print(0, *a, 4)\n");
  2223. /* collect */
  2224. /* assert */
  2225. EXPECT_STREQ(log_buff[0], "0 1 2 3 4\r\n");
  2226. /* deinit */
  2227. obj_deinit(pikaMain);
  2228. EXPECT_EQ(pikaMemNow(), 0);
  2229. }
  2230. TEST(vm, fn_star_star) {
  2231. /* init */
  2232. pikaMemInfo.heapUsedMax = 0;
  2233. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2234. /* run */
  2235. __platform_printf("BEGIN\r\n");
  2236. obj_run(pikaMain,
  2237. "def test(**kw):\n"
  2238. " print(kw['a'], kw['b'], kw['c'])\n"
  2239. "a = {'a':1,'b':2,'c':3}\n"
  2240. "test(**a)\n");
  2241. /* collect */
  2242. /* assert */
  2243. EXPECT_STREQ(log_buff[0], "1 2 3\r\n");
  2244. /* deinit */
  2245. obj_deinit(pikaMain);
  2246. EXPECT_EQ(pikaMemNow(), 0);
  2247. }
  2248. TEST(vm, issue_keyword_mem_leak) {
  2249. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2250. extern unsigned char pikaModules_py_a[];
  2251. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2252. pikaVM_runSingleFile(pikaMain,
  2253. "test/python/issue/issue_keyword_mem_leak.py");
  2254. /* assert */
  2255. /* deinit */
  2256. obj_deinit(pikaMain);
  2257. EXPECT_EQ(pikaMemNow(), 0);
  2258. }
  2259. TEST(vm, issue_not_in) {
  2260. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2261. extern unsigned char pikaModules_py_a[];
  2262. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2263. obj_run(pikaMain,
  2264. "a = [1,2,3]\n"
  2265. "print(1 not in a)\n"
  2266. "print(4 not in a)\n");
  2267. /* assert */
  2268. /* deinit */
  2269. obj_deinit(pikaMain);
  2270. EXPECT_EQ(pikaMemNow(), 0);
  2271. }
  2272. TEST(vm, def_not_in) {
  2273. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2274. extern unsigned char pikaModules_py_a[];
  2275. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2276. obj_run(pikaMain,
  2277. "t = ['hmac-md5', 'other']\n"
  2278. "def test(a=\"hmac-md5\"):\n"
  2279. " print(\"input:\", a)\n"
  2280. " print(\"table:\", t)\n"
  2281. " if a not in t:\n"
  2282. " print(\"a not in t\")\n"
  2283. " else:\n"
  2284. " print(\"a in t\")\n"
  2285. "test()\n");
  2286. /* assert */
  2287. EXPECT_STREQ(log_buff[0], "a in t\r\n");
  2288. /* deinit */
  2289. obj_deinit(pikaMain);
  2290. EXPECT_EQ(pikaMemNow(), 0);
  2291. }
  2292. TEST(vm, fn_pos_kw_issue1) {
  2293. /* init */
  2294. pikaMemInfo.heapUsedMax = 0;
  2295. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2296. /* run */
  2297. __platform_printf("BEGIN\r\n");
  2298. obj_run(pikaMain,
  2299. "def new(a=1, b=2):\n"
  2300. " print(a, b)\n"
  2301. "new(3, b = 2)\n");
  2302. /* collect */
  2303. /* assert */
  2304. EXPECT_STREQ(log_buff[0], "3 2\r\n");
  2305. /* deinit */
  2306. obj_deinit(pikaMain);
  2307. EXPECT_EQ(pikaMemNow(), 0);
  2308. }
  2309. TEST(vm, fn_pos_kw_issue2) {
  2310. /* init */
  2311. pikaMemInfo.heapUsedMax = 0;
  2312. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2313. /* run */
  2314. __platform_printf("BEGIN\r\n");
  2315. obj_run(pikaMain,
  2316. "def new(key:any, msg=None, digestmod=\"md5\"):\n"
  2317. " print(key, msg, digestmod)\n"
  2318. "new('123','456',digestmod=\"md5\")\n");
  2319. /* collect */
  2320. /* assert */
  2321. EXPECT_STREQ(log_buff[0], "123 456 md5\r\n");
  2322. /* deinit */
  2323. obj_deinit(pikaMain);
  2324. EXPECT_EQ(pikaMemNow(), 0);
  2325. }
  2326. TEST(vm, num_issue_lakj) {
  2327. /* init */
  2328. pikaMemInfo.heapUsedMax = 0;
  2329. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2330. /* run */
  2331. __platform_printf("BEGIN\r\n");
  2332. obj_run(pikaMain, "a = 1/2*3\n");
  2333. /* collect */
  2334. /* assert */
  2335. EXPECT_DOUBLE_EQ(obj_getFloat(pikaMain, "a"), 1.5);
  2336. /* deinit */
  2337. obj_deinit(pikaMain);
  2338. EXPECT_EQ(pikaMemNow(), 0);
  2339. }
  2340. TEST(vm, dir_issue) {
  2341. /* init */
  2342. pikaMemInfo.heapUsedMax = 0;
  2343. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2344. /* run */
  2345. __platform_printf("BEGIN\r\n");
  2346. obj_run(pikaMain,
  2347. "class test:\n"
  2348. " pass\n"
  2349. "dir(test)\n");
  2350. /* collect */
  2351. /* assert */
  2352. /* deinit */
  2353. obj_deinit(pikaMain);
  2354. EXPECT_EQ(pikaMemNow(), 0);
  2355. }
  2356. TEST(vm, dir_issue1lk) {
  2357. /* init */
  2358. pikaMemInfo.heapUsedMax = 0;
  2359. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2360. extern unsigned char pikaModules_py_a[];
  2361. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2362. /* run */
  2363. __platform_printf("BEGIN\r\n");
  2364. obj_run(pikaMain,
  2365. "import requests\n"
  2366. "dir(requests)\n");
  2367. /* collect */
  2368. /* assert */
  2369. /* deinit */
  2370. obj_deinit(pikaMain);
  2371. EXPECT_EQ(pikaMemNow(), 0);
  2372. }
  2373. TEST(vm, type_int) {
  2374. /* init */
  2375. pikaMemInfo.heapUsedMax = 0;
  2376. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2377. extern unsigned char pikaModules_py_a[];
  2378. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2379. /* run */
  2380. __platform_printf("BEGIN\r\n");
  2381. obj_run(pikaMain,
  2382. "i = type(1)\n"
  2383. "ii = i(10)\n");
  2384. /* collect */
  2385. EXPECT_EQ(obj_getInt(pikaMain, "ii"), 10);
  2386. /* assert */
  2387. /* deinit */
  2388. obj_deinit(pikaMain);
  2389. EXPECT_EQ(pikaMemNow(), 0);
  2390. }
  2391. TEST(vm, method_int) {
  2392. /* init */
  2393. pikaMemInfo.heapUsedMax = 0;
  2394. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2395. extern unsigned char pikaModules_py_a[];
  2396. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2397. /* run */
  2398. __platform_printf("BEGIN\r\n");
  2399. obj_run(pikaMain,
  2400. "i = int\n"
  2401. "ii = i(10)\n");
  2402. /* collect */
  2403. /* assert */
  2404. EXPECT_EQ(obj_getInt(pikaMain, "ii"), 10);
  2405. /* deinit */
  2406. obj_deinit(pikaMain);
  2407. EXPECT_EQ(pikaMemNow(), 0);
  2408. }
  2409. TEST(vm, fn_method_int) {
  2410. /* init */
  2411. pikaMemInfo.heapUsedMax = 0;
  2412. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2413. extern unsigned char pikaModules_py_a[];
  2414. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2415. /* run */
  2416. __platform_printf("BEGIN\r\n");
  2417. obj_run(pikaMain,
  2418. "def test():\n"
  2419. " i = int\n"
  2420. " ii = i(10)\n"
  2421. " return ii\n"
  2422. "ii = test()\n");
  2423. /* collect */
  2424. /* assert */
  2425. EXPECT_EQ(obj_getInt(pikaMain, "ii"), 10);
  2426. /* deinit */
  2427. obj_deinit(pikaMain);
  2428. EXPECT_EQ(pikaMemNow(), 0);
  2429. }
  2430. TEST(vm, kw_no_empty) {
  2431. /* init */
  2432. pikaMemInfo.heapUsedMax = 0;
  2433. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2434. extern unsigned char pikaModules_py_a[];
  2435. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2436. /* run */
  2437. __platform_printf("BEGIN\r\n");
  2438. obj_run(pikaMain,
  2439. "def test(a=None, b=None, c=None):\n"
  2440. " print(a, b, c)\n"
  2441. "test(1, 2)\n");
  2442. /* collect */
  2443. /* assert */
  2444. EXPECT_STREQ(log_buff[0], "1 2 None\r\n");
  2445. /* deinit */
  2446. obj_deinit(pikaMain);
  2447. EXPECT_EQ(pikaMemNow(), 0);
  2448. }
  2449. #endif
  2450. TEST_END