VM-test.cpp 77 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653
  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[6], "__init__\r\n");
  1263. EXPECT_STREQ(log_buff[5], "a\r\n");
  1264. EXPECT_STREQ(log_buff[4], "a\r\n");
  1265. EXPECT_STREQ(log_buff[3], "b\r\n");
  1266. EXPECT_STREQ(log_buff[2], "b\r\n");
  1267. EXPECT_STREQ(log_buff[1], "ppp\r\n");
  1268. EXPECT_STREQ(log_buff[0], "ppp\r\n");
  1269. /* deinit */
  1270. obj_deinit(pikaMain);
  1271. EXPECT_EQ(pikaMemNow(), 0);
  1272. }
  1273. TEST(vm, cb_3) {
  1274. /* init */
  1275. pikaMemInfo.heapUsedMax = 0;
  1276. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1277. extern unsigned char pikaModules_py_a[];
  1278. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1279. /* run */
  1280. __platform_printf("BEGIN\r\n");
  1281. pikaVM_runSingleFile(pikaMain, "test/python/callback/test3.py");
  1282. /* collect */
  1283. /* deinit */
  1284. obj_deinit(pikaMain);
  1285. EXPECT_EQ(pikaMemNow(), 0);
  1286. }
  1287. #endif
  1288. #if !PIKA_NANO_ENABLE
  1289. TEST(vm, default_no_input) {
  1290. /* init */
  1291. pikaMemInfo.heapUsedMax = 0;
  1292. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1293. extern unsigned char pikaModules_py_a[];
  1294. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1295. /* run */
  1296. __platform_printf("BEGIN\r\n");
  1297. obj_run(pikaMain,
  1298. "def test(a = 1):\n"
  1299. " print(a)\n"
  1300. "test()");
  1301. /* collect */
  1302. /* assert */
  1303. EXPECT_STREQ(log_buff[0], "1\r\n");
  1304. /* deinit */
  1305. obj_deinit(pikaMain);
  1306. EXPECT_EQ(pikaMemNow(), 0);
  1307. }
  1308. TEST(vm, default_1) {
  1309. /* init */
  1310. pikaMemInfo.heapUsedMax = 0;
  1311. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1312. extern unsigned char pikaModules_py_a[];
  1313. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1314. /* run */
  1315. __platform_printf("BEGIN\r\n");
  1316. obj_run(pikaMain,
  1317. "def test(a = 1):\n"
  1318. " print(a)\n"
  1319. "test(a = 2)");
  1320. /* collect */
  1321. /* assert */
  1322. EXPECT_STREQ(log_buff[0], "2\r\n");
  1323. /* deinit */
  1324. obj_deinit(pikaMain);
  1325. EXPECT_EQ(pikaMemNow(), 0);
  1326. }
  1327. TEST(vm, default_2) {
  1328. /* init */
  1329. pikaMemInfo.heapUsedMax = 0;
  1330. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1331. extern unsigned char pikaModules_py_a[];
  1332. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1333. /* run */
  1334. __platform_printf("BEGIN\r\n");
  1335. obj_run(pikaMain,
  1336. "def test(a, b = 1):\n"
  1337. " print(a, b)\n"
  1338. "test(1, b = 2)");
  1339. /* collect */
  1340. /* assert */
  1341. EXPECT_STREQ(log_buff[0], "1 2\r\n");
  1342. /* deinit */
  1343. obj_deinit(pikaMain);
  1344. EXPECT_EQ(pikaMemNow(), 0);
  1345. }
  1346. TEST(vm, default_3) {
  1347. /* init */
  1348. pikaMemInfo.heapUsedMax = 0;
  1349. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1350. extern unsigned char pikaModules_py_a[];
  1351. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1352. /* run */
  1353. __platform_printf("BEGIN\r\n");
  1354. obj_run(pikaMain,
  1355. "def myprint(end='\\n', *var):\n"
  1356. " for v in var:\n"
  1357. " print(v, end = end)\n"
  1358. "myprint(1, 2)\n"
  1359. "myprint(1, 2, end=';\\n')\n");
  1360. /* collect */
  1361. /* assert */
  1362. EXPECT_STREQ(log_buff[3], "1\n");
  1363. EXPECT_STREQ(log_buff[2], "2\n");
  1364. EXPECT_STREQ(log_buff[1], "1;\n");
  1365. EXPECT_STREQ(log_buff[0], "2;\n");
  1366. /* deinit */
  1367. obj_deinit(pikaMain);
  1368. EXPECT_EQ(pikaMemNow(), 0);
  1369. }
  1370. TEST(vm, default_4) {
  1371. /* init */
  1372. pikaMemInfo.heapUsedMax = 0;
  1373. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1374. extern unsigned char pikaModules_py_a[];
  1375. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1376. /* run */
  1377. __platform_printf("BEGIN\r\n");
  1378. obj_run(pikaMain,
  1379. "def test(end=5, *var):\n"
  1380. " res = end\n"
  1381. " print('end', end)\n"
  1382. " for v in var:\n"
  1383. " res += v\n"
  1384. " return res\n"
  1385. "test(1, 2, 3)\n"
  1386. "test(1, 2, 3, end=6)\n");
  1387. /* collect */
  1388. /* assert */
  1389. EXPECT_STREQ(log_buff[3], "end 5\r\n");
  1390. EXPECT_STREQ(log_buff[2], "11\r\n");
  1391. EXPECT_STREQ(log_buff[1], "end 6\r\n");
  1392. EXPECT_STREQ(log_buff[0], "12\r\n");
  1393. /* deinit */
  1394. obj_deinit(pikaMain);
  1395. EXPECT_EQ(pikaMemNow(), 0);
  1396. }
  1397. TEST(vm, default_no_kw) {
  1398. /* init */
  1399. pikaMemInfo.heapUsedMax = 0;
  1400. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1401. extern unsigned char pikaModules_py_a[];
  1402. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1403. /* run */
  1404. __platform_printf("BEGIN\r\n");
  1405. obj_run(pikaMain,
  1406. "def test(a, b=3):\n"
  1407. " print(a ,b)\n"
  1408. "test(1, 2)\n"
  1409. "test(1, b=2)\n"
  1410. "test(1)\n");
  1411. /* collect */
  1412. /* assert */
  1413. EXPECT_STREQ(log_buff[2], "1 2\r\n");
  1414. EXPECT_STREQ(log_buff[1], "1 2\r\n");
  1415. EXPECT_STREQ(log_buff[0], "1 3\r\n");
  1416. /* deinit */
  1417. obj_deinit(pikaMain);
  1418. EXPECT_EQ(pikaMemNow(), 0);
  1419. }
  1420. #endif
  1421. TEST(vm, none) {
  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, "GTestTask.test(1, None)");
  1430. /* collect */
  1431. /* assert */
  1432. EXPECT_STREQ(log_buff[0], "GTestTask_test: 1, 0\r\n");
  1433. /* deinit */
  1434. obj_deinit(pikaMain);
  1435. EXPECT_EQ(pikaMemNow(), 0);
  1436. }
  1437. #if !PIKA_NANO_ENABLE
  1438. TEST(vm, super_) {
  1439. /* init */
  1440. pikaMemInfo.heapUsedMax = 0;
  1441. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1442. extern unsigned char pikaModules_py_a[];
  1443. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1444. /* run */
  1445. __platform_printf("BEGIN\r\n");
  1446. obj_run(pikaMain,
  1447. "class test:\n"
  1448. " def __init__(self):\n"
  1449. " print('in test init')\n"
  1450. "class test2(test):\n"
  1451. " def __init__(self):\n"
  1452. " super().__init__()\n"
  1453. " print('in test2 init')\n"
  1454. "t = test2()");
  1455. /* collect */
  1456. /* assert */
  1457. EXPECT_STREQ(log_buff[0], "in test2 init\r\n");
  1458. EXPECT_STREQ(log_buff[1], "in test init\r\n");
  1459. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1460. /* deinit */
  1461. obj_deinit(pikaMain);
  1462. EXPECT_EQ(pikaMemNow(), 0);
  1463. }
  1464. TEST(vm, super_val) {
  1465. /* init */
  1466. pikaMemInfo.heapUsedMax = 0;
  1467. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1468. extern unsigned char pikaModules_py_a[];
  1469. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1470. /* run */
  1471. __platform_printf("BEGIN\r\n");
  1472. obj_run(pikaMain,
  1473. "class test:\n"
  1474. " def __init__(self):\n"
  1475. " self.a = 1\n"
  1476. " self.b = 2\n"
  1477. "class test2(test):\n"
  1478. " def __init__(self):\n"
  1479. " super().__init__()\n"
  1480. " self.c = 3\n"
  1481. " self.d = 4\n"
  1482. "t = test2()\n"
  1483. "print(t.a, t.b, t.c, t.d)\n");
  1484. /* collect */
  1485. /* assert */
  1486. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1487. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  1488. /* deinit */
  1489. obj_deinit(pikaMain);
  1490. EXPECT_EQ(pikaMemNow(), 0);
  1491. }
  1492. TEST(vm, super_val_) {
  1493. /* init */
  1494. pikaMemInfo.heapUsedMax = 0;
  1495. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1496. extern unsigned char pikaModules_py_a[];
  1497. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1498. /* run */
  1499. __platform_printf("BEGIN\r\n");
  1500. obj_run(pikaMain,
  1501. "class test:\n"
  1502. " def test(self, a):\n"
  1503. " self.a = a\n"
  1504. "class test2(test):\n"
  1505. " def test(self, a):\n"
  1506. " super().test(str(a))\n"
  1507. "t1 = test()\n"
  1508. "t2 = test2()\n"
  1509. "t1.test(1)\n"
  1510. "t2.test(1)\n");
  1511. /* collect */
  1512. int t1_a = obj_getInt(pikaMain, "t1.a");
  1513. char* t2_a = obj_getStr(pikaMain, "t2.a");
  1514. /* assert */
  1515. EXPECT_EQ(t1_a, 1);
  1516. EXPECT_STREQ(t2_a, "1");
  1517. /* deinit */
  1518. obj_deinit(pikaMain);
  1519. EXPECT_EQ(pikaMemNow(), 0);
  1520. }
  1521. #endif
  1522. #if !PIKA_NANO_ENABLE
  1523. TEST(vm, multi_return) {
  1524. /* init */
  1525. pikaMemInfo.heapUsedMax = 0;
  1526. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1527. extern unsigned char pikaModules_py_a[];
  1528. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1529. /* run */
  1530. __platform_printf("BEGIN\r\n");
  1531. obj_run(pikaMain, "a,b = (1,2)");
  1532. /* collect */
  1533. int a = obj_getInt(pikaMain, "a");
  1534. int b = obj_getInt(pikaMain, "b");
  1535. /* assert */
  1536. EXPECT_EQ(a, 1);
  1537. EXPECT_EQ(b, 2);
  1538. /* deinit */
  1539. obj_deinit(pikaMain);
  1540. EXPECT_EQ(pikaMemNow(), 0);
  1541. }
  1542. #endif
  1543. #if !PIKA_NANO_ENABLE
  1544. TEST(vm, multi_return_fn) {
  1545. /* init */
  1546. pikaMemInfo.heapUsedMax = 0;
  1547. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1548. extern unsigned char pikaModules_py_a[];
  1549. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1550. /* run */
  1551. __platform_printf("BEGIN\r\n");
  1552. obj_run(pikaMain,
  1553. "def test():\n"
  1554. " return 1,2\n"
  1555. "a,b = test()");
  1556. /* collect */
  1557. int a = obj_getInt(pikaMain, "a");
  1558. int b = obj_getInt(pikaMain, "b");
  1559. /* assert */
  1560. EXPECT_EQ(a, 1);
  1561. EXPECT_EQ(b, 2);
  1562. /* deinit */
  1563. obj_deinit(pikaMain);
  1564. EXPECT_EQ(pikaMemNow(), 0);
  1565. }
  1566. #endif
  1567. TEST(vm, range_1) {
  1568. /* init */
  1569. pikaMemInfo.heapUsedMax = 0;
  1570. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1571. extern unsigned char pikaModules_py_a[];
  1572. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1573. /* run */
  1574. __platform_printf("BEGIN\r\n");
  1575. obj_run(pikaMain,
  1576. "sum = 0\n"
  1577. "for i in range(10):\n"
  1578. " sum += i\n");
  1579. /* collect */
  1580. int sum = obj_getInt(pikaMain, "sum");
  1581. /* assert */
  1582. EXPECT_EQ(sum, 45);
  1583. /* deinit */
  1584. obj_deinit(pikaMain);
  1585. EXPECT_EQ(pikaMemNow(), 0);
  1586. }
  1587. TEST(vm, rang_3) {
  1588. /* init */
  1589. pikaMemInfo.heapUsedMax = 0;
  1590. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1591. extern unsigned char pikaModules_py_a[];
  1592. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1593. /* run */
  1594. __platform_printf("BEGIN\r\n");
  1595. obj_run(pikaMain,
  1596. "sum = 0\n"
  1597. "for i in range(-5, 20, 2):\n"
  1598. " sum += i\n");
  1599. /* collect */
  1600. int sum = obj_getInt(pikaMain, "sum");
  1601. /* assert */
  1602. EXPECT_EQ(sum, 91);
  1603. /* deinit */
  1604. obj_deinit(pikaMain);
  1605. EXPECT_EQ(pikaMemNow(), 0);
  1606. }
  1607. TEST(vm, test64) {
  1608. /* init */
  1609. pikaMemInfo.heapUsedMax = 0;
  1610. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1611. extern unsigned char pikaModules_py_a[];
  1612. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1613. /* run */
  1614. __platform_printf("BEGIN\r\n");
  1615. obj_run(pikaMain,
  1616. "import GTestTask\n"
  1617. "res = GTestTask.test64(4294967295, 20)\n");
  1618. /* collect */
  1619. int64_t res = obj_getInt(pikaMain, "res");
  1620. /* assert */
  1621. EXPECT_EQ(res, 4294967295 * 20);
  1622. /* deinit */
  1623. obj_deinit(pikaMain);
  1624. EXPECT_EQ(pikaMemNow(), 0);
  1625. }
  1626. #if !PIKA_NANO_ENABLE
  1627. TEST(vm, exit) {
  1628. /* init */
  1629. pikaMemInfo.heapUsedMax = 0;
  1630. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1631. extern unsigned char pikaModules_py_a[];
  1632. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1633. /* run */
  1634. __platform_printf("BEGIN\r\n");
  1635. obj_run(pikaMain,
  1636. "i = 0\n"
  1637. "while True:\n"
  1638. " i += 1\n"
  1639. " print(i)\n"
  1640. " if i == 10:\n"
  1641. " exit()\n");
  1642. /* collect */
  1643. int i = obj_getInt(pikaMain, "i");
  1644. /* assert */
  1645. EXPECT_EQ(i, 10);
  1646. /* deinit */
  1647. obj_deinit(pikaMain);
  1648. EXPECT_EQ(pikaMemNow(), 0);
  1649. }
  1650. TEST(vm, exit_fn) {
  1651. /* init */
  1652. pikaMemInfo.heapUsedMax = 0;
  1653. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1654. extern unsigned char pikaModules_py_a[];
  1655. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1656. /* run */
  1657. __platform_printf("BEGIN\r\n");
  1658. obj_run(pikaMain,
  1659. "i = 0\n"
  1660. "PikaStdLib.MemChecker.now()\n"
  1661. "def test():\n"
  1662. " global i\n"
  1663. " while True:\n"
  1664. " i += 1\n"
  1665. " print(i)\n"
  1666. " if i == 10:\n"
  1667. " exit()\n"
  1668. "while True:\n"
  1669. " test()\n");
  1670. obj_run(pikaMain, "PikaStdLib.MemChecker.now()\n");
  1671. /* collect */
  1672. int i = obj_getInt(pikaMain, "i");
  1673. /* assert */
  1674. EXPECT_EQ(i, 10);
  1675. /* deinit */
  1676. obj_deinit(pikaMain);
  1677. EXPECT_EQ(pikaMemNow(), 0);
  1678. }
  1679. #endif
  1680. #if PIKA_INSTRUCT_HOOK_ENABLE
  1681. typedef void (*hook_func)(void);
  1682. extern volatile int g_hook_cnt;
  1683. extern volatile hook_func g_hook_func;
  1684. extern "C" {
  1685. volatile int g_hook_cnt_triggle;
  1686. void hook_func_exit_issue_1(void) {
  1687. if (g_hook_cnt >= g_hook_cnt_triggle) {
  1688. pks_vm_exit();
  1689. }
  1690. }
  1691. void __gtest_hook_default_(void) {
  1692. return;
  1693. }
  1694. }
  1695. void _vm_exit_fn_issue_1_item(int hook_cnt) {
  1696. g_hook_cnt_triggle = hook_cnt;
  1697. /* init */
  1698. pikaMemInfo.heapUsedMax = 0;
  1699. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1700. extern unsigned char pikaModules_py_a[];
  1701. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1702. /* run */
  1703. __platform_printf("BEGIN\r\n");
  1704. obj_run(pikaMain,
  1705. "PikaStdLib.MemChecker.now()\n"
  1706. "def task1():\n"
  1707. " print('task1 hello')\n"
  1708. "\n"
  1709. "def task2():\n"
  1710. " print('task2 hello')\n"
  1711. "\n"
  1712. "def fake_runtask():\n"
  1713. " i = 0\n"
  1714. " while True:\n"
  1715. " i += 1\n"
  1716. " print(i)\n"
  1717. " task1()\n"
  1718. " task2()\n"
  1719. "fake_runtask()\n"
  1720. "fake_runtask()\n"
  1721. "\n");
  1722. obj_run(pikaMain, "PikaStdLib.MemChecker.now()\n");
  1723. /* collect */
  1724. /* assert */
  1725. obj_deinit(pikaMain);
  1726. EXPECT_EQ(pikaMemNow(), 0);
  1727. }
  1728. TEST(vm, exit_fn_issue_1) {
  1729. g_hook_func = hook_func_exit_issue_1;
  1730. for (int i = 0; i < 50; i++) {
  1731. g_hook_cnt = 0;
  1732. _vm_exit_fn_issue_1_item(i);
  1733. }
  1734. g_hook_func = __gtest_hook_default_;
  1735. }
  1736. #endif
  1737. TEST(vm, pass_) {
  1738. /* init */
  1739. pikaMemInfo.heapUsedMax = 0;
  1740. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1741. extern unsigned char pikaModules_py_a[];
  1742. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1743. /* run */
  1744. __platform_printf("BEGIN\r\n");
  1745. obj_run(pikaMain,
  1746. "pass\n"
  1747. "def testpass():\n"
  1748. " pass\n"
  1749. "for i in range(10):\n"
  1750. " pass\n"
  1751. "testpass()\n"
  1752. "print('after pass')\n");
  1753. /* collect */
  1754. /* assert */
  1755. EXPECT_STREQ(log_buff[0], "after pass\r\n");
  1756. /* deinit */
  1757. obj_deinit(pikaMain);
  1758. EXPECT_EQ(pikaMemNow(), 0);
  1759. }
  1760. TEST(vm, test64_hex) {
  1761. /* init */
  1762. pikaMemInfo.heapUsedMax = 0;
  1763. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1764. extern unsigned char pikaModules_py_a[];
  1765. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1766. /* run */
  1767. __platform_printf("BEGIN\r\n");
  1768. obj_run(pikaMain,
  1769. "import GTestTask\n"
  1770. "res = GTestTask.test64(0xffffffff, 20)\n");
  1771. /* collect */
  1772. int64_t res = obj_getInt(pikaMain, "res");
  1773. /* assert */
  1774. EXPECT_EQ(res, 4294967295 * 20);
  1775. /* deinit */
  1776. obj_deinit(pikaMain);
  1777. EXPECT_EQ(pikaMemNow(), 0);
  1778. }
  1779. TEST(vm, test64_hex_print) {
  1780. /* init */
  1781. pikaMemInfo.heapUsedMax = 0;
  1782. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1783. extern unsigned char pikaModules_py_a[];
  1784. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1785. /* run */
  1786. __platform_printf("BEGIN\r\n");
  1787. obj_run(pikaMain,
  1788. "0xffffffff\n"
  1789. "print(0xffffffff)\n");
  1790. /* collect */
  1791. /* assert */
  1792. EXPECT_STREQ(log_buff[0], "4294967295\r\n");
  1793. EXPECT_STREQ(log_buff[1], "4294967295\r\n");
  1794. /* deinit */
  1795. obj_deinit(pikaMain);
  1796. EXPECT_EQ(pikaMemNow(), 0);
  1797. }
  1798. #if !PIKA_NANO_ENABLE
  1799. TEST(vm, call_dict_err) {
  1800. /* init */
  1801. pikaMemInfo.heapUsedMax = 0;
  1802. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1803. extern unsigned char pikaModules_py_a[];
  1804. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1805. /* run */
  1806. __platform_printf("BEGIN\r\n");
  1807. obj_run(pikaMain,
  1808. "a = {'a': 1}\n"
  1809. "a('a')\n");
  1810. /* collect */
  1811. /* assert */
  1812. EXPECT_STREQ(log_buff[4], "TypeError: 'a' object is not callable\r\n");
  1813. /* deinit */
  1814. obj_deinit(pikaMain);
  1815. EXPECT_EQ(pikaMemNow(), 0);
  1816. }
  1817. TEST(vm, getattribute) {
  1818. /* init */
  1819. pikaMemInfo.heapUsedMax = 0;
  1820. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1821. extern unsigned char pikaModules_py_a[];
  1822. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1823. /* run */
  1824. __platform_printf("BEGIN\r\n");
  1825. obj_run(pikaMain,
  1826. "class test:\n"
  1827. " def __getattribute__(self, name):\n"
  1828. " return name\n"
  1829. "t = test()\n"
  1830. "t.a\n");
  1831. /* collect */
  1832. /* assert */
  1833. EXPECT_STREQ(log_buff[0], "'a'\r\n");
  1834. /* deinit */
  1835. obj_deinit(pikaMain);
  1836. EXPECT_EQ(pikaMemNow(), 0);
  1837. }
  1838. TEST(vm, getattr) {
  1839. /* init */
  1840. pikaMemInfo.heapUsedMax = 0;
  1841. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1842. extern unsigned char pikaModules_py_a[];
  1843. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1844. /* run */
  1845. __platform_printf("BEGIN\r\n");
  1846. obj_run(pikaMain,
  1847. "class test:\n"
  1848. " a = 1\n"
  1849. " def __getattr__(self, name):\n"
  1850. " return name\n"
  1851. "t = test()\n"
  1852. "t.a\n"
  1853. "t.b\n");
  1854. /* collect */
  1855. /* assert */
  1856. EXPECT_STREQ(log_buff[0], "'b'\r\n");
  1857. EXPECT_STREQ(log_buff[1], "1\r\n");
  1858. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1859. /* deinit */
  1860. obj_deinit(pikaMain);
  1861. EXPECT_EQ(pikaMemNow(), 0);
  1862. }
  1863. TEST(vm, setattr) {
  1864. /* init */
  1865. pikaMemInfo.heapUsedMax = 0;
  1866. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1867. extern unsigned char pikaModules_py_a[];
  1868. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1869. /* run */
  1870. __platform_printf("BEGIN\r\n");
  1871. obj_run(pikaMain,
  1872. "class test:\n"
  1873. " a = 1\n"
  1874. " def __setattr__(self, name, val):\n"
  1875. " print((name, val))\n"
  1876. "t = test()\n"
  1877. "t.a = 1\n"
  1878. "t.b = 'test'\n");
  1879. /* collect */
  1880. /* assert */
  1881. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1882. EXPECT_STREQ(log_buff[1], "('a', 1)\r\n");
  1883. EXPECT_STREQ(log_buff[0], "('b', 'test')\r\n");
  1884. /* deinit */
  1885. obj_deinit(pikaMain);
  1886. EXPECT_EQ(pikaMemNow(), 0);
  1887. }
  1888. TEST(vm, c_module_get_set_attr) {
  1889. /* init */
  1890. pikaMemInfo.heapUsedMax = 0;
  1891. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1892. extern unsigned char pikaModules_py_a[];
  1893. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1894. /* run */
  1895. __platform_printf("BEGIN\r\n");
  1896. obj_run(pikaMain,
  1897. "t = GTestTask.ProxyTest()\n"
  1898. "t.a\n"
  1899. "t.b\n"
  1900. "t.a = 'test1'\n"
  1901. "t.b = 'test2'\n");
  1902. /* collect */
  1903. /* assert */
  1904. EXPECT_STREQ(log_buff[3], "'a'\r\n");
  1905. EXPECT_STREQ(log_buff[2], "'b'\r\n");
  1906. EXPECT_STREQ(log_buff[1], "GTestTask_ProxyTest___setattr__: a, test1\r\n");
  1907. EXPECT_STREQ(log_buff[0], "GTestTask_ProxyTest___setattr__: b, test2\r\n");
  1908. /* deinit */
  1909. obj_deinit(pikaMain);
  1910. EXPECT_EQ(pikaMemNow(), 0);
  1911. }
  1912. TEST(vm, class_attr_ref) {
  1913. /* init */
  1914. pikaMemInfo.heapUsedMax = 0;
  1915. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1916. extern unsigned char pikaModules_py_a[];
  1917. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1918. /* run */
  1919. __platform_printf("BEGIN\r\n");
  1920. obj_run(pikaMain,
  1921. "class test:\n"
  1922. " a = 1\n"
  1923. " b = a\n"
  1924. "t = test()\n");
  1925. /* collect */
  1926. /* assert */
  1927. /* deinit */
  1928. obj_deinit(pikaMain);
  1929. EXPECT_EQ(pikaMemNow(), 0);
  1930. }
  1931. TEST(vm, getattr_native) {
  1932. /* init */
  1933. pikaMemInfo.heapUsedMax = 0;
  1934. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1935. extern unsigned char pikaModules_py_a[];
  1936. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1937. /* run */
  1938. __platform_printf("BEGIN\r\n");
  1939. obj_run(pikaMain,
  1940. "mem = PikaStdLib.MemChecker()\n"
  1941. "if hasattr(mem, 'max'):\n"
  1942. " max = getattr(mem, 'max')\n"
  1943. " max()\n");
  1944. /* collect */
  1945. /* assert */
  1946. /* deinit */
  1947. obj_deinit(pikaMain);
  1948. EXPECT_EQ(pikaMemNow(), 0);
  1949. }
  1950. TEST(vm, issue_dict_update) {
  1951. /* init */
  1952. pikaMemInfo.heapUsedMax = 0;
  1953. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1954. extern unsigned char pikaModules_py_a[];
  1955. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1956. /* run */
  1957. __platform_printf("BEGIN\r\n");
  1958. pikaVM_runSingleFile(pikaMain, "test/python/issue/issue_dict_update.py");
  1959. /* collect */
  1960. /* assert */
  1961. EXPECT_STREQ(obj_getStr(pikaMain, "inner"), "widget5");
  1962. /* deinit */
  1963. obj_deinit(pikaMain);
  1964. EXPECT_EQ(pikaMemNow(), 0);
  1965. }
  1966. TEST(vm, issue_big_dict_update) {
  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. pikaVM_runSingleFile(pikaMain,
  1975. "test/python/issue/issue_big_dict_update.py");
  1976. /* collect */
  1977. /* assert */
  1978. /* deinit */
  1979. obj_deinit(pikaMain);
  1980. EXPECT_EQ(pikaMemNow(), 0);
  1981. }
  1982. #endif
  1983. TEST(vm, i_pp) {
  1984. /* init */
  1985. pikaMemInfo.heapUsedMax = 0;
  1986. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1987. extern unsigned char pikaModules_py_a[];
  1988. obj_linkLibrary(pikaMain, pikaModules_py_a);
  1989. /* run */
  1990. __platform_printf("BEGIN\r\n");
  1991. /* clang-format off */
  1992. PIKA_PYTHON(
  1993. i = 0
  1994. while i < 10000:
  1995. i += 1
  1996. )
  1997. /* clang-format on */
  1998. const uint8_t bytes[] = {
  1999. 0x30, 0x00, 0x00, 0x00, /* instruct array size */
  2000. 0x00, 0x85, 0x01, 0x00, 0x00, 0x04, 0x03, 0x00, 0x10, 0x81, 0x03, 0x00,
  2001. 0x10, 0x05, 0x05, 0x00, 0x00, 0x08, 0x0b, 0x00, 0x00, 0x07, 0x0d, 0x00,
  2002. 0x11, 0x81, 0x03, 0x00, 0x21, 0x05, 0x0f, 0x00, 0x11, 0x02, 0x00, 0x00,
  2003. 0x01, 0x08, 0x11, 0x00, 0x01, 0x04, 0x03, 0x00, 0x00, 0x86, 0x13, 0x00,
  2004. /* instruct array */
  2005. 0x16, 0x00, 0x00, 0x00, /* const pool size */
  2006. 0x00, 0x30, 0x00, 0x69, 0x00, 0x31, 0x30, 0x30, 0x30, 0x30, 0x00, 0x3c,
  2007. 0x00, 0x32, 0x00, 0x31, 0x00, 0x2b, 0x00, 0x2d, 0x31,
  2008. 0x00, /* const pool */
  2009. };
  2010. pikaVM_runByteCode(pikaMain, (uint8_t*)bytes);
  2011. EXPECT_EQ(obj_getInt(pikaMain, "i"), 10000);
  2012. /* deinit */
  2013. obj_deinit(pikaMain);
  2014. EXPECT_EQ(pikaMemNow(), 0);
  2015. }
  2016. TEST(vm, benchmark) {
  2017. /* init */
  2018. pikaMemInfo.heapUsedMax = 0;
  2019. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2020. extern unsigned char pikaModules_py_a[];
  2021. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2022. /* run */
  2023. __platform_printf("BEGIN\r\n");
  2024. /* clang-format off */
  2025. PIKA_PYTHON(
  2026. num = 0
  2027. i = 2
  2028. while i < 1000:
  2029. is_prime = 1
  2030. j = 2
  2031. while j < i:
  2032. if i%j==0 :
  2033. is_prime = 0
  2034. break
  2035. j += 1
  2036. if is_prime:
  2037. num = num + i
  2038. i += 1
  2039. )
  2040. /* clang-format on */
  2041. const uint8_t bytes[] = {
  2042. 0xac, 0x00, 0x00, 0x00, /* instruct array size */
  2043. 0x00, 0x85, 0x01, 0x00, 0x00, 0x04, 0x03, 0x00, 0x00, 0x85, 0x07, 0x00,
  2044. 0x00, 0x04, 0x09, 0x00, 0x10, 0x81, 0x09, 0x00, 0x10, 0x05, 0x0b, 0x00,
  2045. 0x00, 0x08, 0x10, 0x00, 0x00, 0x07, 0x07, 0x00, 0x01, 0x85, 0x12, 0x00,
  2046. 0x01, 0x04, 0x14, 0x00, 0x01, 0x85, 0x07, 0x00, 0x01, 0x04, 0x1d, 0x00,
  2047. 0x11, 0x81, 0x1d, 0x00, 0x11, 0x01, 0x09, 0x00, 0x01, 0x08, 0x10, 0x00,
  2048. 0x01, 0x07, 0x07, 0x00, 0x22, 0x81, 0x09, 0x00, 0x22, 0x01, 0x1d, 0x00,
  2049. 0x12, 0x08, 0x1f, 0x00, 0x12, 0x05, 0x01, 0x00, 0x02, 0x08, 0x21, 0x00,
  2050. 0x02, 0x07, 0x12, 0x00, 0x03, 0x85, 0x01, 0x00, 0x03, 0x04, 0x14, 0x00,
  2051. 0x03, 0x8e, 0x00, 0x00, 0x12, 0x81, 0x1d, 0x00, 0x22, 0x05, 0x12, 0x00,
  2052. 0x12, 0x02, 0x00, 0x00, 0x02, 0x08, 0x24, 0x00, 0x02, 0x04, 0x1d, 0x00,
  2053. 0x01, 0x86, 0x26, 0x00, 0x01, 0x81, 0x14, 0x00, 0x01, 0x07, 0x12, 0x00,
  2054. 0x12, 0x81, 0x03, 0x00, 0x12, 0x01, 0x09, 0x00, 0x02, 0x08, 0x24, 0x00,
  2055. 0x02, 0x04, 0x03, 0x00, 0x11, 0x81, 0x09, 0x00, 0x21, 0x05, 0x12, 0x00,
  2056. 0x11, 0x02, 0x00, 0x00, 0x01, 0x08, 0x24, 0x00, 0x01, 0x04, 0x09, 0x00,
  2057. 0x00, 0x86, 0x26, 0x00, /* instruct array */
  2058. 0x29, 0x00, 0x00, 0x00, /* const pool size */
  2059. 0x00, 0x30, 0x00, 0x6e, 0x75, 0x6d, 0x00, 0x32, 0x00, 0x69, 0x00, 0x31,
  2060. 0x30, 0x30, 0x30, 0x00, 0x3c, 0x00, 0x31, 0x00, 0x69, 0x73, 0x5f, 0x70,
  2061. 0x72, 0x69, 0x6d, 0x65, 0x00, 0x6a, 0x00, 0x25, 0x00, 0x3d, 0x3d, 0x00,
  2062. 0x2b, 0x00, 0x2d, 0x31, 0x00, /* const pool */
  2063. };
  2064. pikaVM_runByteCode(pikaMain, (uint8_t*)bytes);
  2065. /* deinit */
  2066. obj_deinit(pikaMain);
  2067. EXPECT_EQ(pikaMemNow(), 0);
  2068. }
  2069. TEST(VM, print_None) {
  2070. char* line = "print(None)\n";
  2071. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  2072. obj_run(self, line);
  2073. /* collect */
  2074. /* assert */
  2075. EXPECT_STREQ(log_buff[0], "None\r\n");
  2076. /* deinit */
  2077. obj_deinit(self);
  2078. EXPECT_EQ(pikaMemNow(), 0);
  2079. }
  2080. TEST(VM, bc_fn) {
  2081. PikaObj* self = newRootObj("root", New_PikaMain);
  2082. obj_run(self, "print('hello world')\n");
  2083. /* clang-format off */
  2084. PIKA_PYTHON(
  2085. def test():
  2086. print('test')
  2087. )
  2088. /* clang-format on */
  2089. const uint8_t
  2090. bytes[] =
  2091. {
  2092. 0x14, 0x00, 0x00, 0x00, /* instruct array size */
  2093. 0x00, 0x89, 0x01, 0x00, 0x00, 0x06, 0x08,
  2094. 0x00, 0x11, 0x83, 0x0a, 0x00, 0x01, 0x02,
  2095. 0x0f, 0x00, 0x01, 0x8a, 0x00, 0x00, /* instruct array */
  2096. 0x15, 0x00, 0x00, 0x00, /* const pool size */
  2097. 0x00, 0x74, 0x65, 0x73, 0x74, 0x28, 0x29,
  2098. 0x00, 0x31, 0x00, 0x74, 0x65, 0x73, 0x74,
  2099. 0x00, 0x70, 0x72, 0x69, 0x6e, 0x74, 0x00, /* const pool */
  2100. };
  2101. pikaVM_runByteCode(self, (uint8_t*)bytes);
  2102. obj_run(self, "test()\n");
  2103. EXPECT_STREQ(log_buff[1], "hello world\r\n");
  2104. EXPECT_STREQ(log_buff[0], "test\r\n");
  2105. obj_deinit(self);
  2106. EXPECT_EQ(pikaMemNow(), 0);
  2107. }
  2108. #if !PIKA_NANO_ENABLE
  2109. TEST(VM, default_num_err) {
  2110. char* line =
  2111. "def test(a, b=1):\n"
  2112. " print(a, b)\n"
  2113. "test(1, 2, 3)\n";
  2114. PikaObj* self = newRootObj("root", New_PikaStdLib_SysObj);
  2115. obj_run(self, line);
  2116. /* collect */
  2117. /* assert */
  2118. EXPECT_STREQ(log_buff[8],
  2119. "TypeError: test() takes from 1 to 2 positional arguments but "
  2120. "3 were given\r\n");
  2121. /* deinit */
  2122. obj_deinit(self);
  2123. EXPECT_EQ(pikaMemNow(), 0);
  2124. }
  2125. #endif
  2126. TEST(VM, bc_fn_file_cb0) {
  2127. PikaObj* self = newRootObj("root", New_PikaMain);
  2128. pikaVM_runByteCodeFile(self,
  2129. "package/pikascript/pikascript-api/cb_test.py.o");
  2130. obj_run(self, "test()\n");
  2131. EXPECT_STREQ(log_buff[0], "test\r\n");
  2132. obj_deinit(self);
  2133. EXPECT_EQ(pikaMemNow(), 0);
  2134. }
  2135. TEST(VM, bc_fn_file_cb2) {
  2136. PikaObj* self = newRootObj("root", New_PikaMain);
  2137. obj_run(self, "print('hello world')\n");
  2138. pikaVM_runByteCodeFile(self,
  2139. "package/pikascript/pikascript-api/cb_test.py.o");
  2140. obj_run(self, "test()\n");
  2141. EXPECT_STREQ(log_buff[1], "hello world\r\n");
  2142. EXPECT_STREQ(log_buff[0], "test\r\n");
  2143. obj_deinit(self);
  2144. EXPECT_EQ(pikaMemNow(), 0);
  2145. }
  2146. #if !PIKA_NANO_ENABLE
  2147. TEST(vm, slice_str_end) {
  2148. /* init */
  2149. pikaMemInfo.heapUsedMax = 0;
  2150. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2151. /* run */
  2152. __platform_printf("BEGIN\r\n");
  2153. obj_run(pikaMain, "'test'[:-1]");
  2154. /* collect */
  2155. /* assert */
  2156. EXPECT_STREQ(log_buff[0], "'tes'\r\n");
  2157. /* deinit */
  2158. obj_deinit(pikaMain);
  2159. EXPECT_EQ(pikaMemNow(), 0);
  2160. }
  2161. #endif
  2162. #if !PIKA_NANO_ENABLE
  2163. TEST(vm, fn_pos_kw) {
  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(b=1,a=2)\n");
  2173. /* collect */
  2174. /* assert */
  2175. EXPECT_STREQ(log_buff[0], "2 1\r\n");
  2176. /* deinit */
  2177. obj_deinit(pikaMain);
  2178. EXPECT_EQ(pikaMemNow(), 0);
  2179. }
  2180. TEST(vm, fn_pos_kw2) {
  2181. /* init */
  2182. pikaMemInfo.heapUsedMax = 0;
  2183. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2184. /* run */
  2185. __platform_printf("BEGIN\r\n");
  2186. obj_run(pikaMain,
  2187. "def test(a, b):\n"
  2188. " print(a, b)\n"
  2189. "test(1, b=2)\n"
  2190. "test(1, a=2)\n");
  2191. /* collect */
  2192. /* assert */
  2193. EXPECT_STREQ(log_buff[1], "1 2\r\n");
  2194. EXPECT_STREQ(log_buff[0], "2 1\r\n");
  2195. /* deinit */
  2196. obj_deinit(pikaMain);
  2197. EXPECT_EQ(pikaMemNow(), 0);
  2198. }
  2199. TEST(vm, fn_star) {
  2200. /* init */
  2201. pikaMemInfo.heapUsedMax = 0;
  2202. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2203. /* run */
  2204. __platform_printf("BEGIN\r\n");
  2205. obj_run(pikaMain,
  2206. "a = (1,2,3)\n"
  2207. "print(*a)\n");
  2208. /* collect */
  2209. /* assert */
  2210. EXPECT_STREQ(log_buff[0], "1 2 3\r\n");
  2211. /* deinit */
  2212. obj_deinit(pikaMain);
  2213. EXPECT_EQ(pikaMemNow(), 0);
  2214. }
  2215. TEST(vm, fn_star_pos) {
  2216. /* init */
  2217. pikaMemInfo.heapUsedMax = 0;
  2218. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2219. /* run */
  2220. __platform_printf("BEGIN\r\n");
  2221. obj_run(pikaMain,
  2222. "a = (1,2,3)\n"
  2223. "print(*a, 4)\n");
  2224. /* collect */
  2225. /* assert */
  2226. EXPECT_STREQ(log_buff[0], "1 2 3 4\r\n");
  2227. /* deinit */
  2228. obj_deinit(pikaMain);
  2229. EXPECT_EQ(pikaMemNow(), 0);
  2230. }
  2231. TEST(vm, fn_star_pos_2) {
  2232. /* init */
  2233. pikaMemInfo.heapUsedMax = 0;
  2234. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2235. /* run */
  2236. __platform_printf("BEGIN\r\n");
  2237. obj_run(pikaMain,
  2238. "a = (1,2,3)\n"
  2239. "print(0, *a, 4)\n");
  2240. /* collect */
  2241. /* assert */
  2242. EXPECT_STREQ(log_buff[0], "0 1 2 3 4\r\n");
  2243. /* deinit */
  2244. obj_deinit(pikaMain);
  2245. EXPECT_EQ(pikaMemNow(), 0);
  2246. }
  2247. TEST(vm, fn_star_star) {
  2248. /* init */
  2249. pikaMemInfo.heapUsedMax = 0;
  2250. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2251. /* run */
  2252. __platform_printf("BEGIN\r\n");
  2253. obj_run(pikaMain,
  2254. "def test(**kw):\n"
  2255. " print(kw['a'], kw['b'], kw['c'])\n"
  2256. "a = {'a':1,'b':2,'c':3}\n"
  2257. "test(**a)\n");
  2258. /* collect */
  2259. /* assert */
  2260. EXPECT_STREQ(log_buff[0], "1 2 3\r\n");
  2261. /* deinit */
  2262. obj_deinit(pikaMain);
  2263. EXPECT_EQ(pikaMemNow(), 0);
  2264. }
  2265. TEST(vm, issue_keyword_mem_leak) {
  2266. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2267. extern unsigned char pikaModules_py_a[];
  2268. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2269. pikaVM_runSingleFile(pikaMain,
  2270. "test/python/issue/issue_keyword_mem_leak.py");
  2271. /* assert */
  2272. /* deinit */
  2273. obj_deinit(pikaMain);
  2274. EXPECT_EQ(pikaMemNow(), 0);
  2275. }
  2276. TEST(vm, issue_not_in) {
  2277. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2278. extern unsigned char pikaModules_py_a[];
  2279. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2280. obj_run(pikaMain,
  2281. "a = [1,2,3]\n"
  2282. "print(1 not in a)\n"
  2283. "print(4 not in a)\n");
  2284. /* assert */
  2285. /* deinit */
  2286. obj_deinit(pikaMain);
  2287. EXPECT_EQ(pikaMemNow(), 0);
  2288. }
  2289. TEST(vm, def_not_in) {
  2290. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2291. extern unsigned char pikaModules_py_a[];
  2292. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2293. obj_run(pikaMain,
  2294. "t = ['hmac-md5', 'other']\n"
  2295. "def test(a=\"hmac-md5\"):\n"
  2296. " print(\"input:\", a)\n"
  2297. " print(\"table:\", t)\n"
  2298. " if a not in t:\n"
  2299. " print(\"a not in t\")\n"
  2300. " else:\n"
  2301. " print(\"a in t\")\n"
  2302. "test()\n");
  2303. /* assert */
  2304. EXPECT_STREQ(log_buff[0], "a in t\r\n");
  2305. /* deinit */
  2306. obj_deinit(pikaMain);
  2307. EXPECT_EQ(pikaMemNow(), 0);
  2308. }
  2309. TEST(vm, fn_pos_kw_issue1) {
  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(a=1, b=2):\n"
  2317. " print(a, b)\n"
  2318. "new(3, b = 2)\n");
  2319. /* collect */
  2320. /* assert */
  2321. EXPECT_STREQ(log_buff[0], "3 2\r\n");
  2322. /* deinit */
  2323. obj_deinit(pikaMain);
  2324. EXPECT_EQ(pikaMemNow(), 0);
  2325. }
  2326. TEST(vm, fn_pos_kw_issue2) {
  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,
  2333. "def new(key:any, msg=None, digestmod=\"md5\"):\n"
  2334. " print(key, msg, digestmod)\n"
  2335. "new('123','456',digestmod=\"md5\")\n");
  2336. /* collect */
  2337. /* assert */
  2338. EXPECT_STREQ(log_buff[0], "123 456 md5\r\n");
  2339. /* deinit */
  2340. obj_deinit(pikaMain);
  2341. EXPECT_EQ(pikaMemNow(), 0);
  2342. }
  2343. TEST(vm, num_issue_lakj) {
  2344. /* init */
  2345. pikaMemInfo.heapUsedMax = 0;
  2346. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2347. /* run */
  2348. __platform_printf("BEGIN\r\n");
  2349. obj_run(pikaMain, "a = 1/2*3\n");
  2350. /* collect */
  2351. /* assert */
  2352. EXPECT_DOUBLE_EQ(obj_getFloat(pikaMain, "a"), 1.5);
  2353. /* deinit */
  2354. obj_deinit(pikaMain);
  2355. EXPECT_EQ(pikaMemNow(), 0);
  2356. }
  2357. TEST(vm, dir_issue) {
  2358. /* init */
  2359. pikaMemInfo.heapUsedMax = 0;
  2360. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2361. /* run */
  2362. __platform_printf("BEGIN\r\n");
  2363. obj_run(pikaMain,
  2364. "class test:\n"
  2365. " pass\n"
  2366. "dir(test)\n");
  2367. /* collect */
  2368. /* assert */
  2369. /* deinit */
  2370. obj_deinit(pikaMain);
  2371. EXPECT_EQ(pikaMemNow(), 0);
  2372. }
  2373. TEST(vm, dir_issue1lk) {
  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. "import requests\n"
  2383. "dir(requests)\n");
  2384. /* collect */
  2385. /* assert */
  2386. /* deinit */
  2387. obj_deinit(pikaMain);
  2388. EXPECT_EQ(pikaMemNow(), 0);
  2389. }
  2390. TEST(vm, type_int) {
  2391. /* init */
  2392. pikaMemInfo.heapUsedMax = 0;
  2393. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2394. extern unsigned char pikaModules_py_a[];
  2395. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2396. /* run */
  2397. __platform_printf("BEGIN\r\n");
  2398. obj_run(pikaMain,
  2399. "i = type(1)\n"
  2400. "ii = i(10)\n");
  2401. /* collect */
  2402. EXPECT_EQ(obj_getInt(pikaMain, "ii"), 10);
  2403. /* assert */
  2404. /* deinit */
  2405. obj_deinit(pikaMain);
  2406. EXPECT_EQ(pikaMemNow(), 0);
  2407. }
  2408. TEST(vm, method_int) {
  2409. /* init */
  2410. pikaMemInfo.heapUsedMax = 0;
  2411. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2412. extern unsigned char pikaModules_py_a[];
  2413. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2414. /* run */
  2415. __platform_printf("BEGIN\r\n");
  2416. obj_run(pikaMain,
  2417. "i = int\n"
  2418. "ii = i(10)\n");
  2419. /* collect */
  2420. /* assert */
  2421. EXPECT_EQ(obj_getInt(pikaMain, "ii"), 10);
  2422. /* deinit */
  2423. obj_deinit(pikaMain);
  2424. EXPECT_EQ(pikaMemNow(), 0);
  2425. }
  2426. TEST(vm, fn_method_int) {
  2427. /* init */
  2428. pikaMemInfo.heapUsedMax = 0;
  2429. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2430. extern unsigned char pikaModules_py_a[];
  2431. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2432. /* run */
  2433. __platform_printf("BEGIN\r\n");
  2434. obj_run(pikaMain,
  2435. "def test():\n"
  2436. " i = int\n"
  2437. " ii = i(10)\n"
  2438. " return ii\n"
  2439. "ii = test()\n");
  2440. /* collect */
  2441. /* assert */
  2442. EXPECT_EQ(obj_getInt(pikaMain, "ii"), 10);
  2443. /* deinit */
  2444. obj_deinit(pikaMain);
  2445. EXPECT_EQ(pikaMemNow(), 0);
  2446. }
  2447. TEST(vm, kw_no_empty) {
  2448. /* init */
  2449. pikaMemInfo.heapUsedMax = 0;
  2450. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2451. extern unsigned char pikaModules_py_a[];
  2452. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2453. /* run */
  2454. __platform_printf("BEGIN\r\n");
  2455. obj_run(pikaMain,
  2456. "def test(a=None, b=None, c=None):\n"
  2457. " print(a, b, c)\n"
  2458. "test(1, 2)\n");
  2459. /* collect */
  2460. /* assert */
  2461. EXPECT_STREQ(log_buff[0], "1 2 None\r\n");
  2462. /* deinit */
  2463. obj_deinit(pikaMain);
  2464. EXPECT_EQ(pikaMemNow(), 0);
  2465. }
  2466. TEST(vm, tuple_void) {
  2467. /* init */
  2468. pikaMemInfo.heapUsedMax = 0;
  2469. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2470. extern unsigned char pikaModules_py_a[];
  2471. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2472. /* run */
  2473. __platform_printf("BEGIN\r\n");
  2474. obj_run(pikaMain, "()\n");
  2475. /* collect */
  2476. /* assert */
  2477. EXPECT_STREQ(log_buff[0], "()\r\n");
  2478. /* deinit */
  2479. obj_deinit(pikaMain);
  2480. EXPECT_EQ(pikaMemNow(), 0);
  2481. }
  2482. TEST(vm, run_file) {
  2483. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2484. pikaVM_runFile(pikaMain, "package/pikascript/main.py");
  2485. obj_deinit(pikaMain);
  2486. EXPECT_EQ(pikaMemNow(), 0);
  2487. }
  2488. #endif
  2489. TEST_END