VM-test.cpp 84 KB

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