pikaMain-test.cpp 71 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648
  1. #include "test_common.h"
  2. TEST(pikaMain, init) {
  3. pikaMemInfo.heapUsedMax = 0;
  4. PikaObj* pikaMain = pikaScriptInit();
  5. obj_deinit(pikaMain);
  6. EXPECT_EQ(pikaMemNow(), 0);
  7. }
  8. TEST(pikaMain, new_mem) {
  9. pikaMemInfo.heapUsedMax = 0;
  10. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  11. VMParameters* globals = obj_runDirect(pikaMain,
  12. "mem = PikaStdLib.MemChecker()\n"
  13. "mem.max()\n");
  14. PikaObj* mem = (PikaObj*)args_getPtr(globals->list, "mem");
  15. PikaObj* memClassPtr = (PikaObj*)mem->constructor;
  16. EXPECT_EQ((void*)memClassPtr, (void*)New_PikaStdLib_MemChecker);
  17. obj_deinit(pikaMain);
  18. // obj_deinit(globals);
  19. EXPECT_EQ(pikaMemNow(), 0);
  20. }
  21. // TEST(pikaMain, unitTest) {
  22. // /* init */
  23. // pikaMemInfo.heapUsedMax = 0;
  24. // PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  25. // /* run */
  26. // __platform_printf("BEGIN\r\n");
  27. // pikaVM_runSingleFile(pikaMain, "../../examples/UnitTest/main.py");
  28. // /* collect */
  29. // /* assert */
  30. // /* deinit */
  31. // obj_deinit(pikaMain);
  32. // EXPECT_EQ(pikaMemNow(), 0);
  33. // }
  34. TEST(pikaMain, list_new) {
  35. /* init */
  36. pikaMemInfo.heapUsedMax = 0;
  37. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  38. /* run */
  39. VMParameters* globals = obj_runDirect(pikaMain,
  40. "list = PikaStdData.List()\n"
  41. "list.append(7)\n"
  42. "list.append('eee')\n"
  43. "len = list.len()\n"
  44. "a1 = list.get(0)\n"
  45. "a2 = list.get(1)\n");
  46. /* collect */
  47. int len = obj_getInt(globals, "len");
  48. int a1 = obj_getInt(globals, "a1");
  49. char* a2 = obj_getStr(globals, "a2");
  50. /* assert */
  51. EXPECT_EQ(len, 2);
  52. EXPECT_EQ(a1, 7);
  53. EXPECT_STREQ(a2, "eee");
  54. /* deinit */
  55. // obj_deinit(globals);
  56. obj_deinit(pikaMain);
  57. EXPECT_EQ(pikaMemNow(), 0);
  58. }
  59. TEST(pikaMain, int_float_convert) {
  60. /* init */
  61. pikaMemInfo.heapUsedMax = 0;
  62. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  63. /* run */
  64. VMParameters* globals = obj_runDirect(pikaMain,
  65. "a = 1\n"
  66. "a_f = float(a)\n"
  67. "b = 1.3\n"
  68. "b_i = int(b)\n");
  69. /* collect */
  70. int a = obj_getInt(globals, "a");
  71. float a_f = obj_getFloat(globals, "a_f");
  72. float b = obj_getFloat(globals, "b");
  73. int b_i = obj_getInt(globals, "b_i");
  74. /* assert */
  75. EXPECT_EQ(a, 1);
  76. EXPECT_FLOAT_EQ(a_f, 1);
  77. EXPECT_FLOAT_EQ(b, 1.3);
  78. EXPECT_FLOAT_EQ(b_i, 1);
  79. /* deinit */
  80. // obj_deinit(globals);
  81. obj_deinit(pikaMain);
  82. EXPECT_EQ(pikaMemNow(), 0);
  83. }
  84. TEST(pikaMain, type_) {
  85. /* init */
  86. pikaMemInfo.heapUsedMax = 0;
  87. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  88. /* run */
  89. obj_runDirect(pikaMain,
  90. "type(1)\n"
  91. "b = 1.4\n"
  92. "type(b)\n");
  93. /* collect */
  94. /* assert */
  95. /* deinit */
  96. // obj_deinit(globals);
  97. obj_deinit(pikaMain);
  98. EXPECT_EQ(pikaMemNow(), 0);
  99. }
  100. TEST(pikaMain, a_signed) {
  101. /* init */
  102. pikaMemInfo.heapUsedMax = 0;
  103. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  104. /* run */
  105. obj_runDirect(pikaMain, "a = -1\n");
  106. /* collect */
  107. int a = obj_getInt(pikaMain, "a");
  108. /* assert */
  109. EXPECT_EQ(-1, a);
  110. /* deinit */
  111. obj_deinit(pikaMain);
  112. EXPECT_EQ(pikaMemNow(), 0);
  113. }
  114. TEST(pikaMain, a_mm) {
  115. /* init */
  116. pikaMemInfo.heapUsedMax = 0;
  117. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  118. /* run */
  119. obj_runDirect(pikaMain, "a = -1.1 ** 3\n");
  120. /* collect */
  121. float a = obj_getFloat(pikaMain, "a");
  122. /* assert */
  123. EXPECT_FLOAT_EQ(-1.331, a);
  124. /* deinit */
  125. obj_deinit(pikaMain);
  126. EXPECT_EQ(pikaMemNow(), 0);
  127. }
  128. TEST(pikaMain, a_n1) {
  129. /* init */
  130. pikaMemInfo.heapUsedMax = 0;
  131. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  132. /* run */
  133. obj_runDirect(pikaMain, "a = -2\n");
  134. /* collect */
  135. int a = obj_getInt(pikaMain, "a");
  136. /* assert */
  137. EXPECT_EQ(-2, a);
  138. /* deinit */
  139. obj_deinit(pikaMain);
  140. EXPECT_EQ(pikaMemNow(), 0);
  141. }
  142. TEST(pikaMain, a_l4) {
  143. /* init */
  144. pikaMemInfo.heapUsedMax = 0;
  145. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  146. /* run */
  147. obj_runDirect(pikaMain, "a = 1<<4\n");
  148. /* collect */
  149. int a = obj_getInt(pikaMain, "a");
  150. /* assert */
  151. EXPECT_EQ(16, a);
  152. /* deinit */
  153. obj_deinit(pikaMain);
  154. EXPECT_EQ(pikaMemNow(), 0);
  155. }
  156. TEST(pikaMain, less_equ) {
  157. /* init */
  158. pikaMemInfo.heapUsedMax = 0;
  159. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  160. /* run */
  161. obj_runDirect(pikaMain,
  162. "a = 0\n"
  163. "while a <= 10:\n"
  164. " a += 1\n"
  165. "\n");
  166. /* collect */
  167. int a = obj_getInt(pikaMain, "a");
  168. /* assert */
  169. EXPECT_EQ(11, a);
  170. /* deinit */
  171. obj_deinit(pikaMain);
  172. EXPECT_EQ(pikaMemNow(), 0);
  173. }
  174. TEST(pikaMain, and_or_not) {
  175. /* init */
  176. pikaMemInfo.heapUsedMax = 0;
  177. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  178. /* run */
  179. obj_runDirect(pikaMain,
  180. "a = 0\n"
  181. "b = 0\n"
  182. "c = 0\n"
  183. "d = 0\n"
  184. "if (1>2) or (2>1):\n"
  185. " a = 1\n"
  186. "\n"
  187. "if (1>2) and (2>1):\n"
  188. " b = 1\n"
  189. "\n"
  190. "if not (1>2):\n"
  191. " c = 1\n"
  192. "\n"
  193. "if not (2>1):\n"
  194. " d = 1\n"
  195. "\n"
  196. "e = ~1\n");
  197. /* collect */
  198. int a = obj_getInt(pikaMain, "a");
  199. int b = obj_getInt(pikaMain, "b");
  200. int c = obj_getInt(pikaMain, "c");
  201. int d = obj_getInt(pikaMain, "d");
  202. int e = obj_getInt(pikaMain, "e");
  203. /* assert */
  204. EXPECT_EQ(1, a);
  205. EXPECT_EQ(0, b);
  206. EXPECT_EQ(1, c);
  207. EXPECT_EQ(0, d);
  208. EXPECT_EQ(e, -2);
  209. /* deinit */
  210. obj_deinit(pikaMain);
  211. EXPECT_EQ(pikaMemNow(), 0);
  212. }
  213. TEST(pikaMain, err_scop) {
  214. /* init */
  215. pikaMemInfo.heapUsedMax = 0;
  216. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  217. /* run */
  218. obj_run(pikaMain, "print('test'");
  219. /* collect */
  220. /* assert */
  221. /* deinit */
  222. obj_deinit(pikaMain);
  223. EXPECT_EQ(pikaMemNow(), 0);
  224. }
  225. TEST(pikaMain, PikaStdData) {
  226. /* init */
  227. pikaMemInfo.heapUsedMax = 0;
  228. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  229. /* run */
  230. obj_run(pikaMain,
  231. "dict = PikaStdData.Dict()\n"
  232. "dict.set('a', 1)\n"
  233. "dict.set('b', 2)\n"
  234. "dict.remove('b')\n"
  235. "a = dict.get('a')\n"
  236. "b = dict.get('b')\n");
  237. /* collect */
  238. int a = obj_getInt(pikaMain, "a");
  239. int b = obj_getInt(pikaMain, "b");
  240. /* assert */
  241. EXPECT_EQ(a, 1);
  242. EXPECT_EQ(b, -999999999);
  243. /* deinit */
  244. obj_deinit(pikaMain);
  245. EXPECT_EQ(pikaMemNow(), 0);
  246. }
  247. TEST(pikaMain, list_method) {
  248. /* init */
  249. pikaMemInfo.heapUsedMax = 0;
  250. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  251. /* run */
  252. obj_runDirect(pikaMain,
  253. "list = PikaStdData.List()\n"
  254. "list.append(7)\n"
  255. "list.append('eee')\n"
  256. "len = list.len()\n"
  257. "a1 = list.get(0)\n"
  258. "a2 = list.get(1)\n"
  259. "\n");
  260. /* collect */
  261. /* assert */
  262. /* deinit */
  263. // obj_deinit(globals);
  264. obj_deinit(pikaMain);
  265. EXPECT_EQ(pikaMemNow(), 0);
  266. }
  267. TEST(pikaMain, list_iter) {
  268. /* init */
  269. pikaMemInfo.heapUsedMax = 0;
  270. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  271. /* run */
  272. obj_runDirect(pikaMain,
  273. "list = PikaStdData.List()\n"
  274. "res = iter(list)\n"
  275. "\n");
  276. /* collect */
  277. PikaObj* res = (PikaObj*)obj_getPtr(pikaMain, "res");
  278. PikaObj* list = (PikaObj*)obj_getPtr(pikaMain, "list");
  279. int ref_cnt = obj_refcntNow(list);
  280. EXPECT_EQ(ref_cnt, 2);
  281. /* assert */
  282. EXPECT_EQ(res, list);
  283. /* deinit */
  284. // obj_deinit(globals);
  285. obj_deinit(pikaMain);
  286. EXPECT_EQ(pikaMemNow(), 0);
  287. }
  288. TEST(pikaMain, list_for_loop) {
  289. /* init */
  290. pikaMemInfo.heapUsedMax = 0;
  291. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  292. /* run */
  293. obj_runDirect(pikaMain,
  294. "list = PikaStdData.List()\n"
  295. "list.append(7)\n"
  296. "list.append('eee')\n"
  297. "len = list.len()\n"
  298. "for item in list:\n"
  299. " print(item)\n"
  300. " a = item\n"
  301. "\n");
  302. /* collect */
  303. char* a = obj_getStr(pikaMain, "a");
  304. /* assert */
  305. EXPECT_STREQ(a, "eee");
  306. /* deinit */
  307. // obj_deinit(globals);
  308. obj_deinit(pikaMain);
  309. EXPECT_EQ(pikaMemNow(), 0);
  310. }
  311. TEST(pikaMain, range) {
  312. /* init */
  313. pikaMemInfo.heapUsedMax = 0;
  314. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  315. /* run */
  316. obj_runDirect(pikaMain,
  317. "r = range(10, 0)\n"
  318. "r\n"
  319. "\n");
  320. /* collect */
  321. /* assert */
  322. obj_deinit(pikaMain);
  323. /* mem check */
  324. EXPECT_EQ(pikaMemNow(), 0);
  325. }
  326. TEST(pikaMain, for_in_range) {
  327. /* init */
  328. pikaMemInfo.heapUsedMax = 0;
  329. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  330. /* run */
  331. obj_runDirect(pikaMain,
  332. "a = 0\n"
  333. "for i in range(0, 10):\n"
  334. " print(i)\n"
  335. " a = a + i\n"
  336. "\n");
  337. /* collect */
  338. int a = obj_getInt(pikaMain, "a");
  339. /* assert */
  340. EXPECT_EQ(a, 45);
  341. /* deinit */
  342. obj_deinit(pikaMain);
  343. /* mem check */
  344. EXPECT_EQ(pikaMemNow(), 0);
  345. }
  346. TEST(pikaMain, for_for_in_range) {
  347. /* init */
  348. pikaMemInfo.heapUsedMax = 0;
  349. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  350. /* run */
  351. obj_runDirect(pikaMain,
  352. "a = 0\n"
  353. "for i in range(0, 10):\n"
  354. " for k in range(0, 3):\n"
  355. " print(k)\n"
  356. " a = a + k\n"
  357. "\n");
  358. /* collect */
  359. int a = obj_getInt(pikaMain, "a");
  360. /* assert */
  361. EXPECT_EQ(a, 30);
  362. /* deinit */
  363. obj_deinit(pikaMain);
  364. /* mem check */
  365. EXPECT_EQ(pikaMemNow(), 0);
  366. }
  367. TEST(pikaMain, for_if_break) {
  368. /* init */
  369. pikaMemInfo.heapUsedMax = 0;
  370. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  371. /* run */
  372. obj_runDirect(pikaMain,
  373. "a = 0\n"
  374. "for i in range(0, 10):\n"
  375. " if i == 5:\n"
  376. " break\n"
  377. " a = a + i\n"
  378. "\n");
  379. /* collect */
  380. int a = obj_getInt(pikaMain, "a");
  381. /* assert */
  382. EXPECT_EQ(a, 10);
  383. /* deinit */
  384. obj_deinit(pikaMain);
  385. /* mem check */
  386. EXPECT_EQ(pikaMemNow(), 0);
  387. }
  388. TEST(pikaMain, for_if_continue) {
  389. /* init */
  390. pikaMemInfo.heapUsedMax = 0;
  391. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  392. /* run */
  393. obj_run(pikaMain,
  394. "a = 0\n"
  395. "for i in range(0, 10):\n"
  396. " if i == 5:\n"
  397. " continue\n"
  398. " a = a + i\n"
  399. "\n");
  400. /* collect */
  401. int a = obj_getInt(pikaMain, "a");
  402. /* assert */
  403. EXPECT_EQ(a, 40);
  404. /* deinit */
  405. obj_deinit(pikaMain);
  406. /* mem check */
  407. EXPECT_EQ(pikaMemNow(), 0);
  408. }
  409. /* the log_buff of printf */
  410. extern char log_buff[LOG_BUFF_MAX][LOG_SIZE];
  411. TEST(pikaMain, print_in_def) {
  412. /* init */
  413. pikaMemInfo.heapUsedMax = 0;
  414. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  415. /* run */
  416. /* the test is used to fix too many print in def */
  417. /* clear the log_buff */
  418. memset(log_buff, 0, LOG_BUFF_MAX * LOG_SIZE);
  419. obj_runDirect(pikaMain,
  420. "def test_print():\n"
  421. " print('test')\n"
  422. "test_print()\n");
  423. /* collect */
  424. /* assert */
  425. /* should only print once, so the second log (log_buff[1]) shuold be '\0' */
  426. EXPECT_STREQ(log_buff[0], "test\r\n");
  427. EXPECT_STREQ(log_buff[1], "");
  428. /* deinit */
  429. obj_deinit(pikaMain);
  430. /* mem check */
  431. EXPECT_EQ(pikaMemNow(), 0);
  432. }
  433. TEST(pikaMain, prime_100) {
  434. /* init */
  435. pikaMemInfo.heapUsedMax = 0;
  436. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  437. /* run */
  438. obj_run(pikaMain,
  439. "num = 0\n"
  440. "i = 2\n"
  441. "while i < 100:\n"
  442. " j=2\n"
  443. " is_prime = 1\n"
  444. " while j < i:\n"
  445. " if i%j==0 :\n"
  446. " is_prime = 0\n"
  447. " break\n"
  448. " j+=1\n"
  449. " if is_prime:\n"
  450. " num = num + i\n"
  451. " i+=1\n"
  452. "\n");
  453. /* collect */
  454. int num = obj_getInt(pikaMain, "num");
  455. /* assert */
  456. EXPECT_EQ(num, 1060);
  457. /* deinit */
  458. obj_deinit(pikaMain);
  459. /* mem check */
  460. EXPECT_EQ(pikaMemNow(), 0);
  461. }
  462. /* the log_buff of printf */
  463. extern char log_buff[LOG_BUFF_MAX][LOG_SIZE];
  464. TEST(pikaMain, for_in_dict) {
  465. /* init */
  466. pikaMemInfo.heapUsedMax = 0;
  467. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  468. /* run */
  469. obj_run(pikaMain,
  470. "print('__start__')\n"
  471. "dict = PikaStdData.Dict()\n"
  472. "dict.set('a', 1)\n"
  473. "dict.set('b', 2)\n"
  474. "dict.remove('b')\n"
  475. "for item in dict:\n"
  476. " print(item)\n"
  477. "\n");
  478. /* collect */
  479. /* assert */
  480. EXPECT_STREQ(log_buff[0], "a\r\n");
  481. EXPECT_STREQ(log_buff[1], "__start__\r\n");
  482. /* deinit */
  483. obj_deinit(pikaMain);
  484. EXPECT_EQ(pikaMemNow(), 0);
  485. }
  486. TEST(pikaMain, str_add) {
  487. /* init */
  488. pikaMemInfo.heapUsedMax = 0;
  489. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  490. /* run */
  491. obj_run(pikaMain,
  492. "a = 'a'\n"
  493. "b = 'b'\n"
  494. "c = a + b\n"
  495. "d = 1\n"
  496. "e = 1.2\n"
  497. "g = c + str(d)\n"
  498. "h = c + str(d) + str(e)\n"
  499. "\n");
  500. /* collect */
  501. char* c = obj_getStr(pikaMain, "c");
  502. char* g = obj_getStr(pikaMain, "g");
  503. char* h = obj_getStr(pikaMain, "h");
  504. /* assert */
  505. EXPECT_STREQ(c, "ab");
  506. EXPECT_STREQ(g, "ab1");
  507. EXPECT_STREQ(h, "ab11.200000");
  508. /* deinit */
  509. obj_deinit(pikaMain);
  510. EXPECT_EQ(pikaMemNow(), 0);
  511. }
  512. TEST(pikaMain, str_add_print) {
  513. /* init */
  514. pikaMemInfo.heapUsedMax = 0;
  515. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  516. /* run */
  517. obj_run(pikaMain,
  518. "print('test: ' + str(2233))\n"
  519. "\n");
  520. /* collect */
  521. /* assert */
  522. EXPECT_STREQ(log_buff[0], "test: 2233\r\n");
  523. /* deinit */
  524. obj_deinit(pikaMain);
  525. EXPECT_EQ(pikaMemNow(), 0);
  526. }
  527. TEST(pikaMain, int_float_to_str) {
  528. /* init */
  529. pikaMemInfo.heapUsedMax = 0;
  530. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  531. /* run */
  532. obj_run(pikaMain,
  533. "a = str(1)\n"
  534. "b = str(1.2)\n"
  535. "\n");
  536. /* collect */
  537. char* a = obj_getStr(pikaMain, "a");
  538. char* b = obj_getStr(pikaMain, "b");
  539. /* assert */
  540. EXPECT_STREQ(a, "1");
  541. EXPECT_STREQ(b, "1.200000");
  542. /* deinit */
  543. obj_deinit(pikaMain);
  544. EXPECT_EQ(pikaMemNow(), 0);
  545. }
  546. TEST(pikaMain, str_eq) {
  547. /* init */
  548. pikaMemInfo.heapUsedMax = 0;
  549. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  550. /* run */
  551. obj_run(pikaMain,
  552. "a = '1ee'\n"
  553. "b = '1ee'\n"
  554. "c = '1e'\n"
  555. "if a==b:\n"
  556. " res1 = 1\n"
  557. "else:\n"
  558. " res1 = 0\n"
  559. "if a==c:\n"
  560. " res2 = 1\n"
  561. "else:\n"
  562. " res2 = 0\n"
  563. "\n");
  564. /* collect */
  565. int res1 = obj_getInt(pikaMain, "res1");
  566. int res2 = obj_getInt(pikaMain, "res2");
  567. /* assert */
  568. EXPECT_EQ(res1, 1);
  569. EXPECT_EQ(res2, 0);
  570. /* deinit */
  571. obj_deinit(pikaMain);
  572. EXPECT_EQ(pikaMemNow(), 0);
  573. }
  574. TEST(pikaMain, print_with_enter) {
  575. /* init */
  576. pikaMemInfo.heapUsedMax = 0;
  577. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  578. /* run */
  579. __platform_printf("BEGIN\n");
  580. obj_run(pikaMain, "print('test\\n')\n");
  581. /* collect */
  582. /* assert */
  583. EXPECT_STREQ(log_buff[0], "test\n\r\n");
  584. EXPECT_STREQ(log_buff[1], "BEGIN\n");
  585. /* deinit */
  586. obj_deinit(pikaMain);
  587. EXPECT_EQ(pikaMemNow(), 0);
  588. }
  589. TEST(pikaMain, print_with_2enter) {
  590. /* init */
  591. pikaMemInfo.heapUsedMax = 0;
  592. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  593. /* run */
  594. __platform_printf("BEGIN\n");
  595. obj_run(pikaMain, "print('test\\r\\n\\n')\n");
  596. /* collect */
  597. /* assert */
  598. EXPECT_STREQ(log_buff[0], "test\r\n\n\r\n");
  599. EXPECT_STREQ(log_buff[1], "BEGIN\n");
  600. /* deinit */
  601. obj_deinit(pikaMain);
  602. EXPECT_EQ(pikaMemNow(), 0);
  603. }
  604. TEST(pikaMain, print_ddd) {
  605. /* init */
  606. pikaMemInfo.heapUsedMax = 0;
  607. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  608. /* run */
  609. __platform_printf("BEGIN\n");
  610. obj_run(pikaMain, "print(\"[Info]: in Python config ...\")\n");
  611. /* collect */
  612. /* assert */
  613. EXPECT_STREQ(log_buff[0], "[Info]: in Python config ...\r\n");
  614. EXPECT_STREQ(log_buff[1], "BEGIN\n");
  615. /* deinit */
  616. obj_deinit(pikaMain);
  617. EXPECT_EQ(pikaMemNow(), 0);
  618. }
  619. TEST(pikaMain, for_in_string) {
  620. /* init */
  621. pikaMemInfo.heapUsedMax = 0;
  622. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  623. /* run */
  624. __platform_printf("BEGIN\n");
  625. obj_run(pikaMain,
  626. "s = PikaStdData.String('test')\n"
  627. "for c in s:\n"
  628. " print(c)\n"
  629. "\n");
  630. /* collect */
  631. /* assert */
  632. EXPECT_STREQ(log_buff[4], "BEGIN\n");
  633. EXPECT_STREQ(log_buff[3], "t\r\n");
  634. EXPECT_STREQ(log_buff[2], "e\r\n");
  635. EXPECT_STREQ(log_buff[1], "s\r\n");
  636. EXPECT_STREQ(log_buff[0], "t\r\n");
  637. /* deinit */
  638. obj_deinit(pikaMain);
  639. EXPECT_EQ(pikaMemNow(), 0);
  640. }
  641. TEST(pikaMain, string_no_init_arg) {
  642. /* init */
  643. pikaMemInfo.heapUsedMax = 0;
  644. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  645. /* run */
  646. __platform_printf("BEGIN\r\n");
  647. obj_run(pikaMain, "s = PikaStdData.String()\n");
  648. /* collect */
  649. /* assert */
  650. EXPECT_STREQ(log_buff[3], "BEGIN\r\n");
  651. EXPECT_STREQ(log_buff[2],
  652. "TypeError: __init__() takes 1 positional argument but 0 were "
  653. "given\r\n");
  654. /* deinit */
  655. obj_deinit(pikaMain);
  656. EXPECT_EQ(pikaMemNow(), 0);
  657. }
  658. TEST(pikaMain, obj_no_free) {
  659. /* init */
  660. pikaMemInfo.heapUsedMax = 0;
  661. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  662. /* run */
  663. __platform_printf("BEGIN\n");
  664. obj_run(pikaMain,
  665. "s = PikaStdData.String()\n"
  666. "s = PikaStdData.String()\n"
  667. "\n");
  668. /* collect */
  669. /* assert */
  670. /* deinit */
  671. obj_deinit(pikaMain);
  672. EXPECT_EQ(pikaMemNow(), 0);
  673. }
  674. TEST(pikaMain, list__setitem__) {
  675. /* init */
  676. pikaMemInfo.heapUsedMax = 0;
  677. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  678. /* run */
  679. __platform_printf("BEGIN\n");
  680. obj_run(pikaMain,
  681. "list = PikaStdData.List()\n"
  682. "list = __setitem__(list, 0, 2)\n"
  683. "res = __getitem__(list, 0)\n"
  684. "\n");
  685. /* collect */
  686. int res = obj_getInt(pikaMain, "res");
  687. /* assert */
  688. EXPECT_EQ(res, 2);
  689. /* deinit */
  690. obj_deinit(pikaMain);
  691. EXPECT_EQ(pikaMemNow(), 0);
  692. }
  693. TEST(pikaMain, string__getitem__) {
  694. /* init */
  695. pikaMemInfo.heapUsedMax = 0;
  696. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  697. /* run */
  698. __platform_printf("BEGIN\n");
  699. obj_run(pikaMain,
  700. "s = PikaStdData.String('test')\n"
  701. "res = __getitem__(s, 2)\n"
  702. "\n");
  703. /* collect */
  704. char* res = obj_getStr(pikaMain, "res");
  705. /* assert */
  706. EXPECT_STREQ(res, "s");
  707. /* deinit */
  708. obj_deinit(pikaMain);
  709. EXPECT_EQ(pikaMemNow(), 0);
  710. }
  711. TEST(pikaMain, dict__setitem__get) {
  712. /* init */
  713. pikaMemInfo.heapUsedMax = 0;
  714. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  715. /* run */
  716. __platform_printf("BEGIN\n");
  717. obj_run(pikaMain,
  718. "list = PikaStdData.Dict()\n"
  719. "list = __setitem__(list, 'a', 2)\n"
  720. "res = __getitem__(list, 'a')\n"
  721. "\n");
  722. /* collect */
  723. int res = obj_getInt(pikaMain, "res");
  724. /* assert */
  725. EXPECT_EQ(res, 2);
  726. /* deinit */
  727. obj_deinit(pikaMain);
  728. EXPECT_EQ(pikaMemNow(), 0);
  729. }
  730. TEST(pikaMain, str___getitem____setitem__) {
  731. /* init */
  732. pikaMemInfo.heapUsedMax = 0;
  733. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  734. /* run */
  735. __platform_printf("BEGIN\n");
  736. obj_run(pikaMain,
  737. "s = 'test'\n"
  738. "res = __getitem__(s, 2)\n"
  739. "s = __setitem__(s, 2, 'q')\n"
  740. "\n");
  741. /* collect */
  742. char* res = obj_getStr(pikaMain, "res");
  743. char* s = obj_getStr(pikaMain, "s");
  744. /* assert */
  745. EXPECT_STREQ(res, "s");
  746. EXPECT_STREQ(s, "teqt");
  747. /* deinit */
  748. obj_deinit(pikaMain);
  749. EXPECT_EQ(pikaMemNow(), 0);
  750. }
  751. #if PIKA_SYNTAX_SLICE_ENABLE
  752. TEST(pikaMain, str__index__) {
  753. /* init */
  754. pikaMemInfo.heapUsedMax = 0;
  755. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  756. /* run */
  757. __platform_printf("BEGIN\n");
  758. obj_run(pikaMain,
  759. "s = 'test'\n"
  760. "res = s[2]\n"
  761. "res2 = 'eqrt'[2]\n"
  762. "s[2] = 'q'\n"
  763. "\n");
  764. /* collect */
  765. char* res = obj_getStr(pikaMain, "res");
  766. char* res2 = obj_getStr(pikaMain, "res2");
  767. char* s = obj_getStr(pikaMain, "s");
  768. /* assert */
  769. EXPECT_STREQ(res, "s");
  770. EXPECT_STREQ(res2, "r");
  771. EXPECT_STREQ(s, "teqt");
  772. // EXPECT_STREQ(s, "teqt");
  773. /* deinit */
  774. obj_deinit(pikaMain);
  775. EXPECT_EQ(pikaMemNow(), 0);
  776. }
  777. #endif
  778. #if PIKA_SYNTAX_SLICE_ENABLE
  779. TEST(pikaMain, bytes__index__) {
  780. /* init */
  781. pikaMemInfo.heapUsedMax = 0;
  782. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  783. /* run */
  784. __platform_printf("BEGIN\n");
  785. obj_run(pikaMain,
  786. "s = b'test'\n"
  787. "res = s[2]\n"
  788. "res2 = b'eqrt'[2]\n"
  789. "s[2] = b'q'\n"
  790. "\n");
  791. /* collect */
  792. uint8_t* res = obj_getBytes(pikaMain, "res");
  793. uint8_t* res2 = obj_getBytes(pikaMain, "res2");
  794. uint8_t* s = obj_getBytes(pikaMain, "s");
  795. /* assert */
  796. EXPECT_STREQ((char*)res, "s");
  797. EXPECT_STREQ((char*)res2, "r");
  798. EXPECT_EQ(s[2], 'q');
  799. // EXPECT_STREQ(s, "teqt");
  800. /* deinit */
  801. obj_deinit(pikaMain);
  802. EXPECT_EQ(pikaMemNow(), 0);
  803. }
  804. #endif
  805. #if PIKA_SYNTAX_SLICE_ENABLE
  806. TEST(pikaMain, list_index) {
  807. /* init */
  808. pikaMemInfo.heapUsedMax = 0;
  809. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  810. /* run */
  811. __platform_printf("BEGIN\n");
  812. obj_run(pikaMain,
  813. "list = PikaStdData.List()\n"
  814. "list[0] = 2\n"
  815. "res = list[0]\n"
  816. "\n");
  817. /* collect */
  818. int res = obj_getInt(pikaMain, "res");
  819. /* assert */
  820. EXPECT_EQ(res, 2);
  821. /* deinit */
  822. obj_deinit(pikaMain);
  823. EXPECT_EQ(pikaMemNow(), 0);
  824. }
  825. #endif
  826. #if PIKA_SYNTAX_SLICE_ENABLE
  827. TEST(pikaMain, dict_index) {
  828. /* init */
  829. pikaMemInfo.heapUsedMax = 0;
  830. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  831. /* run */
  832. __platform_printf("BEGIN\n");
  833. obj_run(pikaMain,
  834. "dict = PikaStdData.Dict()\n"
  835. "dict['a'] = 2\n"
  836. "res = dict['a']\n"
  837. "\n");
  838. /* collect */
  839. int res = obj_getInt(pikaMain, "res");
  840. /* assert */
  841. EXPECT_EQ(res, 2);
  842. /* deinit */
  843. obj_deinit(pikaMain);
  844. EXPECT_EQ(pikaMemNow(), 0);
  845. }
  846. #endif
  847. TEST(pikaMain, task_run_once) {
  848. /* init */
  849. pikaMemInfo.heapUsedMax = 0;
  850. /* run */
  851. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  852. obj_run(pikaMain,
  853. "def todo1():\n"
  854. " print('task 1 running...')\n"
  855. "def todo2():\n"
  856. " print('task 2 running...')\n"
  857. "task = GTestTask.Task()\n"
  858. "task.call_always(todo1)\n"
  859. "task.call_always(todo2)\n"
  860. "task.run_once()\n"
  861. "\n");
  862. /* collect */
  863. /* assert */
  864. EXPECT_STREQ(log_buff[0], "task 2 running...\r\n");
  865. EXPECT_STREQ(log_buff[1], "task 1 running...\r\n");
  866. /* deinit */
  867. obj_deinit(pikaMain);
  868. EXPECT_EQ(pikaMemNow(), 0);
  869. }
  870. TEST(pikaMain, task_run_when) {
  871. /* init */
  872. pikaMemInfo.heapUsedMax = 0;
  873. /* run */
  874. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  875. obj_run(pikaMain,
  876. "def todo1():\n"
  877. " print('task 1 running...')\n"
  878. "def todo2():\n"
  879. " print('task 2 running...')\n"
  880. "def todo3():\n"
  881. " print('task 3 running...')\n"
  882. "def when3():\n"
  883. " return True\n"
  884. "task = GTestTask.Task()\n"
  885. "task.call_always(todo1)\n"
  886. "task.call_always(todo2)\n"
  887. "task.call_when(todo3, when3)\n"
  888. "task.run_once()\n"
  889. "\n");
  890. /* collect */
  891. /* assert */
  892. EXPECT_STREQ(log_buff[0], "task 3 running...\r\n");
  893. EXPECT_STREQ(log_buff[1], "task 2 running...\r\n");
  894. EXPECT_STREQ(log_buff[2], "task 1 running...\r\n");
  895. /* deinit */
  896. obj_deinit(pikaMain);
  897. EXPECT_EQ(pikaMemNow(), 0);
  898. }
  899. #if !PIKA_NANO_ENABLE
  900. TEST(pikaMain, task_run_period_until) {
  901. /* init */
  902. pikaMemInfo.heapUsedMax = 0;
  903. /* run */
  904. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  905. __platform_printf("BEGIN\r\n");
  906. obj_run(pikaMain,
  907. "import PikaStdLib\n"
  908. "mem = PikaStdLib.MemChecker()\n"
  909. "mem_now = 0\n"
  910. "mem_max = 0\n"
  911. "mem_lack = False\n"
  912. "def todo1():\n"
  913. " global mem_now\n"
  914. " global mem_lack\n"
  915. " global mem_max\n"
  916. " mem_max = mem.getMax()\n"
  917. " print('task 1 running...')\n"
  918. " if mem_now == 0:\n"
  919. " mem_now = mem.getNow()\n"
  920. " else:\n"
  921. " if mem.getNow() > mem_now + 1:\n"
  922. " mem_lack = True\n"
  923. " print(cformat('%0.8f', mem.getNow()))\n"
  924. "def todo2():\n"
  925. " print('task 2 running...')\n"
  926. "def todo3():\n"
  927. " print('task 3 running...')\n"
  928. "def when3():\n"
  929. " return True\n"
  930. "task = GTestTask.Task()\n"
  931. "task.call_period_ms(todo1, 200)\n"
  932. "task.call_period_ms(todo2, 500)\n"
  933. "# task.call_always(todo3)\n"
  934. "task.run_until_ms(1000)\n"
  935. "\n");
  936. /* collect */
  937. int mem_lack = obj_getInt(pikaMain, "mem_lack");
  938. /* assert */
  939. EXPECT_STREQ(log_buff[0], "task 1 running...\r\n");
  940. EXPECT_STREQ(log_buff[1], "task 1 running...\r\n");
  941. EXPECT_STREQ(log_buff[2], "task 2 running...\r\n");
  942. EXPECT_STREQ(log_buff[3], "task 1 running...\r\n");
  943. EXPECT_STREQ(log_buff[4], "task 1 running...\r\n");
  944. EXPECT_STREQ(log_buff[5], "task 2 running...\r\n");
  945. EXPECT_STREQ(log_buff[6], "task 1 running...\r\n");
  946. EXPECT_STREQ(log_buff[7], "BEGIN\r\n");
  947. EXPECT_EQ(mem_lack, 0);
  948. /* deinit */
  949. obj_deinit(pikaMain);
  950. EXPECT_EQ(pikaMemNow(), 0);
  951. }
  952. #endif
  953. TEST(pikaMain, fun_call) {
  954. /* init */
  955. pikaMemInfo.heapUsedMax = 0;
  956. /* run */
  957. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  958. __platform_printf("BEGIN\r\n");
  959. obj_run(pikaMain,
  960. "def fun(a,b,c,d,e):\n"
  961. " print(a)\n"
  962. " print(b)\n"
  963. " print(c)\n"
  964. " print(d)\n"
  965. " print(e)\n"
  966. "fun(10,20,30,40, 'xxx')\n"
  967. "\n");
  968. /* assert */
  969. EXPECT_STREQ(log_buff[0], "xxx\r\n");
  970. EXPECT_STREQ(log_buff[1], "40\r\n");
  971. EXPECT_STREQ(log_buff[2], "30\r\n");
  972. EXPECT_STREQ(log_buff[3], "20\r\n");
  973. EXPECT_STREQ(log_buff[4], "10\r\n");
  974. EXPECT_STREQ(log_buff[5], "BEGIN\r\n");
  975. /* deinit */
  976. obj_deinit(pikaMain);
  977. EXPECT_EQ(pikaMemNow(), 0);
  978. }
  979. TEST(pikaMain, global) {
  980. /* init */
  981. pikaMemInfo.heapUsedMax = 0;
  982. /* run */
  983. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  984. __platform_printf("BEGIN\r\n");
  985. obj_run(pikaMain,
  986. "x = 0\n"
  987. "a = 0\n"
  988. "z = 0\n"
  989. "def test_global():\n"
  990. " global x\n"
  991. " global y, z\n"
  992. " x = 1\n"
  993. " a = 1\n"
  994. " z = 1\n"
  995. " print('test global')\n"
  996. "test_global()\n"
  997. "\n");
  998. /* assert */
  999. int x = obj_getInt(pikaMain, "x");
  1000. int a = obj_getInt(pikaMain, "a");
  1001. int z = obj_getInt(pikaMain, "z");
  1002. EXPECT_EQ(x, 1);
  1003. EXPECT_EQ(a, 0);
  1004. EXPECT_EQ(z, 1);
  1005. /* deinit */
  1006. obj_deinit(pikaMain);
  1007. EXPECT_EQ(pikaMemNow(), 0);
  1008. }
  1009. TEST(pikaMain, complex_str) {
  1010. /* init */
  1011. pikaMemInfo.heapUsedMax = 0;
  1012. /* run */
  1013. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1014. __platform_printf("BEGIN\r\n");
  1015. obj_run(pikaMain, "print('test,test')\n");
  1016. /* assert */
  1017. EXPECT_STREQ(log_buff[0], "test,test\r\n");
  1018. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1019. /* deinit */
  1020. obj_deinit(pikaMain);
  1021. EXPECT_EQ(pikaMemNow(), 0);
  1022. }
  1023. TEST(pikaMain, synac_err_1) {
  1024. /* init */
  1025. pikaMemInfo.heapUsedMax = 0;
  1026. /* run */
  1027. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1028. __platform_printf("BEGIN\r\n");
  1029. obj_run(pikaMain, "print('testtest)\n");
  1030. /* assert */
  1031. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  1032. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1033. /* deinit */
  1034. obj_deinit(pikaMain);
  1035. EXPECT_EQ(pikaMemNow(), 0);
  1036. }
  1037. TEST(pikaMain, class_arg) {
  1038. /* init */
  1039. pikaMemInfo.heapUsedMax = 0;
  1040. /* run */
  1041. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1042. __platform_printf("BEGIN\r\n");
  1043. Args* buffs = New_strBuff();
  1044. char* pikaAsm = Parser_linesToAsm(buffs,
  1045. "class Test(TinyObj):\n"
  1046. " x = 1\n"
  1047. "\n"
  1048. "test = Test()\n"
  1049. "print(test.x)\n");
  1050. printf("%s", pikaAsm);
  1051. pikaVM_runAsm(pikaMain, pikaAsm);
  1052. /* assert */
  1053. EXPECT_STREQ(log_buff[0], "1\r\n");
  1054. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1055. /* deinit */
  1056. obj_deinit(pikaMain);
  1057. args_deinit(buffs);
  1058. EXPECT_EQ(pikaMemNow(), 0);
  1059. }
  1060. TEST(pikaMain, class_def) {
  1061. /* init */
  1062. pikaMemInfo.heapUsedMax = 0;
  1063. /* run */
  1064. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1065. __platform_printf("BEGIN\r\n");
  1066. obj_run(self,
  1067. "class Test():\n"
  1068. " x = 1\n"
  1069. " def hello(self, x):\n"
  1070. " return x + 2\n"
  1071. "test = Test()\n"
  1072. "x = test.hello(test.x)\n"
  1073. "print(x)\n");
  1074. /* collect */
  1075. PikaObj* test = obj_getObj(self, "test");
  1076. EXPECT_EQ(obj_refcntNow(test), 1);
  1077. /* assert */
  1078. EXPECT_STREQ(log_buff[0], "3\r\n");
  1079. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1080. /* deinit */
  1081. obj_deinit(self);
  1082. EXPECT_EQ(pikaMemNow(), 0);
  1083. }
  1084. TEST(pikaMain, class_def_print) {
  1085. /* init */
  1086. pikaMemInfo.heapUsedMax = 0;
  1087. /* run */
  1088. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1089. __platform_printf("BEGIN\r\n");
  1090. obj_run(self,
  1091. "class Test():\n"
  1092. " x = 1\n"
  1093. " def hi(self):\n"
  1094. " print('hi' + str(self.x))\n"
  1095. "test = Test()\n"
  1096. "test.hi()\n");
  1097. /* assert */
  1098. EXPECT_STREQ(log_buff[0], "hi1\r\n");
  1099. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1100. /* deinit */
  1101. obj_deinit(self);
  1102. EXPECT_EQ(pikaMemNow(), 0);
  1103. }
  1104. TEST(pikaMain, class_demo_1) {
  1105. /* init */
  1106. pikaMemInfo.heapUsedMax = 0;
  1107. /* run */
  1108. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1109. __platform_printf("BEGIN\r\n");
  1110. obj_run(self,
  1111. "class MyClass:\n"
  1112. " i = 12345\n"
  1113. " def f(self):\n"
  1114. " return 'hello world'\n"
  1115. " \n"
  1116. "x = MyClass()\n"
  1117. " \n"
  1118. "print(x.i)\n"
  1119. "print(x.f())\n");
  1120. /* assert */
  1121. EXPECT_STREQ(log_buff[0], "hello world\r\n");
  1122. EXPECT_STREQ(log_buff[1], "12345\r\n");
  1123. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1124. /* deinit */
  1125. obj_deinit(self);
  1126. EXPECT_EQ(pikaMemNow(), 0);
  1127. }
  1128. TEST(pikaMain, class_demo_2) {
  1129. /* init */
  1130. pikaMemInfo.heapUsedMax = 0;
  1131. /* run */
  1132. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1133. __platform_printf("BEGIN\r\n");
  1134. obj_run(self,
  1135. "class Complex:\n"
  1136. " def __init__(self):\n"
  1137. " self.r = 3.0\n"
  1138. " self.i = -4.5\n"
  1139. "x = Complex()\n"
  1140. "print(x.r)\n"
  1141. "print(x.i)\n");
  1142. /* assert */
  1143. EXPECT_STREQ(log_buff[0], "-4.500000\r\n");
  1144. EXPECT_STREQ(log_buff[1], "3.000000\r\n");
  1145. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1146. /* deinit */
  1147. obj_deinit(self);
  1148. EXPECT_EQ(pikaMemNow(), 0);
  1149. }
  1150. TEST(pikaMain, class_demo_3) {
  1151. /* init */
  1152. pikaMemInfo.heapUsedMax = 0;
  1153. /* run */
  1154. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1155. __platform_printf("BEGIN\r\n");
  1156. obj_run(self,
  1157. "class people:\n"
  1158. " def speak(self):\n"
  1159. " print('i am a people')\n"
  1160. " \n"
  1161. "class student(people):\n"
  1162. " def speak(self):\n"
  1163. " print('i am a student')\n"
  1164. " \n"
  1165. "p = people()\n"
  1166. "s = student()\n"
  1167. "p.speak()\n"
  1168. "s.speak()\n");
  1169. /* assert */
  1170. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1171. EXPECT_STREQ(log_buff[1], "i am a people\r\n");
  1172. EXPECT_STREQ(log_buff[0], "i am a student\r\n");
  1173. /* deinit */
  1174. obj_deinit(self);
  1175. EXPECT_EQ(pikaMemNow(), 0);
  1176. }
  1177. TEST(pikaMain, class_demo_save_asm) {
  1178. /* init */
  1179. pikaMemInfo.heapUsedMax = 0;
  1180. /* run */
  1181. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1182. __platform_printf("BEGIN\r\n");
  1183. obj_run(self,
  1184. "class people:\n"
  1185. " def speak(self):\n"
  1186. " print('i am a people')\n"
  1187. " \n"
  1188. "class student(people):\n"
  1189. " def speak(self):\n"
  1190. " print('i am a student')\n"
  1191. " \n");
  1192. obj_run(self,
  1193. "p = people()\n"
  1194. "s = student()\n"
  1195. "p.speak()\n"
  1196. "s.speak()\n");
  1197. /* assert */
  1198. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1199. EXPECT_STREQ(log_buff[1], "i am a people\r\n");
  1200. EXPECT_STREQ(log_buff[0], "i am a student\r\n");
  1201. /* deinit */
  1202. obj_deinit(self);
  1203. EXPECT_EQ(pikaMemNow(), 0);
  1204. }
  1205. TEST(pikaMain, for_if_continue_byte_code) {
  1206. /* init */
  1207. pikaMemInfo.heapUsedMax = 0;
  1208. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1209. /* run */
  1210. pikaVM_run(pikaMain,
  1211. "a = 0\n"
  1212. "for i in range(0, 10):\n"
  1213. " if i == 5:\n"
  1214. " continue\n"
  1215. " a = a + i\n"
  1216. "\n");
  1217. /* collect */
  1218. int a = obj_getInt(pikaMain, "a");
  1219. /* assert */
  1220. EXPECT_EQ(a, 40);
  1221. /* deinit */
  1222. obj_deinit(pikaMain);
  1223. /* mem check */
  1224. EXPECT_EQ(pikaMemNow(), 0);
  1225. }
  1226. TEST(pikaMain, print_in_def_byte_code) {
  1227. /* init */
  1228. pikaMemInfo.heapUsedMax = 0;
  1229. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1230. /* run */
  1231. __platform_printf("BEGIN\r\n");
  1232. pikaVM_run(pikaMain,
  1233. "def test_print():\n"
  1234. " print('test')\n"
  1235. "test_print()\n");
  1236. /* collect */
  1237. /* assert */
  1238. /* should only print once, so the second log (log_buff[1]) shuold be '\0' */
  1239. EXPECT_STREQ(log_buff[0], "test\r\n");
  1240. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1241. /* deinit */
  1242. obj_deinit(pikaMain);
  1243. /* mem check */
  1244. EXPECT_EQ(pikaMemNow(), 0);
  1245. }
  1246. TEST(pikaMain, class_demo_1_file) {
  1247. /* init */
  1248. pikaMemInfo.heapUsedMax = 0;
  1249. /* run */
  1250. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1251. __platform_printf("BEGIN\r\n");
  1252. char lines[] =
  1253. "class MyClass:\n"
  1254. " i = 12345\n"
  1255. " def f(self):\n"
  1256. " return 'hello world'\n"
  1257. " \n"
  1258. "x = MyClass()\n"
  1259. " \n"
  1260. "print(x.i)\n"
  1261. "print(x.f())\n";
  1262. pikaCompile("pika_bytecode.bin", lines);
  1263. char bytecodebuff[4096] = {0};
  1264. FILE* f = __platform_fopen("pika_bytecode.bin", "rb");
  1265. printf("size: %ld\n", fread(bytecodebuff, 1, 4096, f));
  1266. __platform_printf("BEGIN\r\n");
  1267. pikaVM_runByteCode(self, (uint8_t*)bytecodebuff);
  1268. /* assert */
  1269. EXPECT_STREQ(log_buff[0], "hello world\r\n");
  1270. EXPECT_STREQ(log_buff[1], "12345\r\n");
  1271. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1272. /* deinit */
  1273. obj_deinit(self);
  1274. __platform_fclose(f);
  1275. EXPECT_EQ(pikaMemNow(), 0);
  1276. }
  1277. // #if !PIKA_NANO_ENABLE
  1278. // TEST(pikaMain, get_native_method) {
  1279. // /* init */
  1280. // pikaMemInfo.heapUsedMax = 0;
  1281. // PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1282. // __platform_printf("BEGIN\r\n");
  1283. // /* do some thing */
  1284. // Args args = {0};
  1285. // args_setStr(&args, "val", "t");
  1286. // obj_runNativeMethod(pikaMain, "ord", &args);
  1287. // args_deinit_stack(&args);
  1288. // /* assert */
  1289. // EXPECT_STREQ(log_buff[0], "test\r\n");
  1290. // EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1291. // /* deinit */
  1292. // obj_deinit(pikaMain);
  1293. // EXPECT_EQ(pikaMemNow(), 0);
  1294. // }
  1295. // #endif
  1296. TEST(pikaMain, hex_list) {
  1297. /* init */
  1298. pikaMemInfo.heapUsedMax = 0;
  1299. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1300. /* run */
  1301. __platform_printf("BEGIN\r\n");
  1302. pikaVM_run(pikaMain,
  1303. "bytes = PikaStdData.List()\n"
  1304. "bytes.append(0x0F)\n"
  1305. "bytes.append(0xFF)\n"
  1306. "sum = 0\n"
  1307. "for byte in bytes:\n"
  1308. " sum = sum + byte\n"
  1309. "\n"
  1310. "print(sum)\n");
  1311. /* collect */
  1312. /* assert */
  1313. EXPECT_STREQ(log_buff[0], "270\r\n");
  1314. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1315. /* deinit */
  1316. obj_deinit(pikaMain);
  1317. /* mem check */
  1318. EXPECT_EQ(pikaMemNow(), 0);
  1319. }
  1320. TEST(pikaMain, bytearray) {
  1321. /* init */
  1322. pikaMemInfo.heapUsedMax = 0;
  1323. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1324. /* run */
  1325. __platform_printf("BEGIN\r\n");
  1326. pikaVM_run(pikaMain,
  1327. "bytes = PikaStdData.ByteArray(b'test')\n"
  1328. "sum = 0\n"
  1329. "for byte in bytes:\n"
  1330. " sum += byte\n"
  1331. "print(bytes)\n"
  1332. "\n");
  1333. /* collect */
  1334. int sum = obj_getInt(pikaMain, "sum");
  1335. /* assert */
  1336. EXPECT_EQ(sum, 448);
  1337. EXPECT_STREQ(log_buff[0], "bytearray(b'\\x74\\x65\\x73\\x74')\r\n");
  1338. /* deinit */
  1339. obj_deinit(pikaMain);
  1340. /* mem check */
  1341. EXPECT_EQ(pikaMemNow(), 0);
  1342. }
  1343. TEST(pikaMain, not_4_space) {
  1344. /* init */
  1345. pikaMemInfo.heapUsedMax = 0;
  1346. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1347. /* run */
  1348. __platform_printf("BEGIN\r\n");
  1349. pikaVM_run(pikaMain, " print('test')\n");
  1350. /* collect */
  1351. /* assert */
  1352. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  1353. EXPECT_STREQ(
  1354. log_buff[1],
  1355. "IndentationError: unexpected indent, only support 4 spaces\r\n");
  1356. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1357. /* deinit */
  1358. obj_deinit(pikaMain);
  1359. /* mem check */
  1360. EXPECT_EQ(pikaMemNow(), 0);
  1361. }
  1362. TEST(pikaMain, self_operator) {
  1363. /* init */
  1364. pikaMemInfo.heapUsedMax = 0;
  1365. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1366. /* run */
  1367. __platform_printf("BEGIN\r\n");
  1368. pikaVM_run(pikaMain,
  1369. "a = 1\n"
  1370. "b = 2\n"
  1371. "c = 3\n"
  1372. "d = 4\n"
  1373. "a += 2\n"
  1374. "b -= 1+1-3\n"
  1375. "c *= 4\n"
  1376. "d /= 2\n");
  1377. /* collect */
  1378. int a = obj_getInt(pikaMain, "a");
  1379. int b = obj_getInt(pikaMain, "b");
  1380. int c = obj_getInt(pikaMain, "c");
  1381. int d = obj_getInt(pikaMain, "d");
  1382. /* assert */
  1383. EXPECT_EQ(a, 3);
  1384. EXPECT_EQ(b, 3);
  1385. EXPECT_EQ(c, 12);
  1386. EXPECT_EQ(d, 2);
  1387. /* deinit */
  1388. obj_deinit(pikaMain);
  1389. /* mem check */
  1390. EXPECT_EQ(pikaMemNow(), 0);
  1391. }
  1392. TEST(pikaMain, class_demo_2_initwitharg) {
  1393. /* init */
  1394. pikaMemInfo.heapUsedMax = 0;
  1395. /* run */
  1396. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1397. __platform_printf("BEGIN\r\n");
  1398. obj_run(self,
  1399. "class Complex:\n"
  1400. " def __init__(self, r, i):\n"
  1401. " self.r = r\n"
  1402. " self.i = i\n"
  1403. "x = Complex(3.0, -4.5)\n"
  1404. "print(x.r)\n"
  1405. "print(x.i)\n");
  1406. /* assert */
  1407. EXPECT_STREQ(log_buff[0], "-4.500000\r\n");
  1408. EXPECT_STREQ(log_buff[1], "3.000000\r\n");
  1409. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1410. /* deinit */
  1411. obj_deinit(self);
  1412. EXPECT_EQ(pikaMemNow(), 0);
  1413. }
  1414. TEST(pikaMain, def_args_err) {
  1415. /* init */
  1416. pikaMemInfo.heapUsedMax = 0;
  1417. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1418. /* run */
  1419. obj_run(pikaMain, "print()\n");
  1420. // obj_run(pikaMain, "printNoEnd()\n");
  1421. /* collect */
  1422. /* assert */
  1423. /* deinit */
  1424. obj_deinit(pikaMain);
  1425. EXPECT_EQ(pikaMemNow(), 0);
  1426. }
  1427. // TEST(pikaMain, class_args_err) {
  1428. // /* init */
  1429. // pikaMemInfo.heapUsedMax = 0;
  1430. // PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1431. // /* run */
  1432. // obj_run(pikaMain, "mem = PikaStdLib.MemChecker(0)\n");
  1433. // /* collect */
  1434. // /* assert */
  1435. // EXPECT_STREQ(log_buff[4],
  1436. // "TypeError: PikaStdLib.MemChecker() takes no
  1437. // arguments\r\n");
  1438. // /* deinit */
  1439. // obj_deinit(pikaMain);
  1440. // EXPECT_EQ(pikaMemNow(), 0);
  1441. // }
  1442. TEST(pikaMain, int_) {
  1443. /* init */
  1444. pikaMemInfo.heapUsedMax = 0;
  1445. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1446. /* run */
  1447. obj_run(pikaMain, "a = int('3')\n");
  1448. /* collect */
  1449. int a = obj_getInt(pikaMain, "a");
  1450. /* assert */
  1451. EXPECT_EQ(a, 3);
  1452. /* deinit */
  1453. obj_deinit(pikaMain);
  1454. EXPECT_EQ(pikaMemNow(), 0);
  1455. }
  1456. TEST(pikaMain, len_) {
  1457. /* init */
  1458. pikaMemInfo.heapUsedMax = 0;
  1459. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1460. /* run */
  1461. obj_run(pikaMain, "a = len('331dd')\n");
  1462. /* collect */
  1463. int a = obj_getInt(pikaMain, "a");
  1464. /* assert */
  1465. EXPECT_EQ(a, 5);
  1466. /* deinit */
  1467. obj_deinit(pikaMain);
  1468. EXPECT_EQ(pikaMemNow(), 0);
  1469. }
  1470. TEST(pikaMain, def_in_shell) {
  1471. /* init */
  1472. pikaMemInfo.heapUsedMax = 0;
  1473. /* run */
  1474. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1475. /* skip the first obj_run */
  1476. __platform_printf("BEGIN\r\n");
  1477. /* as run in shell */
  1478. obj_run(pikaMain,
  1479. "def test():\n"
  1480. " print('test')\n"
  1481. "\n");
  1482. obj_run(pikaMain,
  1483. "def test2():\n"
  1484. " print('test2')\n"
  1485. "\n");
  1486. obj_run(pikaMain, "test()\n");
  1487. obj_run(pikaMain, "test2()\n");
  1488. /* collect */
  1489. /* assert */
  1490. EXPECT_STREQ("test2\r\n", log_buff[0]);
  1491. EXPECT_STREQ("test\r\n", log_buff[1]);
  1492. EXPECT_STREQ("BEGIN\r\n", log_buff[2]);
  1493. /* deinit */
  1494. obj_deinit(pikaMain);
  1495. EXPECT_EQ(pikaMemNow(), 0);
  1496. }
  1497. TEST(pikaMain, def_in_shell_override) {
  1498. /* init */
  1499. pikaMemInfo.heapUsedMax = 0;
  1500. /* run */
  1501. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1502. /* skip the first obj_run */
  1503. __platform_printf("BEGIN\r\n");
  1504. /* as run in shell */
  1505. obj_run(pikaMain,
  1506. "def test():\n"
  1507. " print('test')\n"
  1508. "\n");
  1509. obj_run(pikaMain, "test()\n");
  1510. obj_run(pikaMain,
  1511. "def test():\n"
  1512. " print('test2')\n"
  1513. "\n");
  1514. obj_run(pikaMain, "test()\n");
  1515. /* collect */
  1516. /* assert */
  1517. EXPECT_STREQ("test2\r\n", log_buff[0]);
  1518. EXPECT_STREQ("test\r\n", log_buff[1]);
  1519. EXPECT_STREQ("BEGIN\r\n", log_buff[2]);
  1520. /* deinit */
  1521. obj_deinit(pikaMain);
  1522. EXPECT_EQ(pikaMemNow(), 0);
  1523. }
  1524. TEST(pikaMain, class_in_shell) {
  1525. /* init */
  1526. pikaMemInfo.heapUsedMax = 0;
  1527. /* run */
  1528. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1529. /* skip the first obj_run */
  1530. __platform_printf("BEGIN\r\n");
  1531. /* as run in shell */
  1532. obj_run(pikaMain,
  1533. "class Test():\n"
  1534. " def hi(self):\n"
  1535. " print('hi')\n"
  1536. "\n");
  1537. obj_run(pikaMain,
  1538. "t = Test()\n"
  1539. "t.hi()\n");
  1540. /* collect */
  1541. /* assert */
  1542. EXPECT_STREQ("hi\r\n", log_buff[0]);
  1543. EXPECT_STREQ("BEGIN\r\n", log_buff[1]);
  1544. /* deinit */
  1545. obj_deinit(pikaMain);
  1546. EXPECT_EQ(pikaMemNow(), 0);
  1547. }
  1548. TEST(pikaMain, class_TinyObj) {
  1549. /* init */
  1550. pikaMemInfo.heapUsedMax = 0;
  1551. /* run */
  1552. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1553. /* skip the first obj_run */
  1554. __platform_printf("BEGIN\r\n");
  1555. /* as run in shell */
  1556. obj_run(pikaMain,
  1557. "class MyClass( TinyObj ):\n"
  1558. " def __init__(self):\n"
  1559. " self.val = 123\n"
  1560. "\n");
  1561. obj_run(pikaMain,
  1562. "x = MyClass()\n"
  1563. "x.val\n");
  1564. /* collect */
  1565. /* assert */
  1566. EXPECT_STREQ("123\r\n", log_buff[0]);
  1567. EXPECT_STREQ("BEGIN\r\n", log_buff[1]);
  1568. /* deinit */
  1569. obj_deinit(pikaMain);
  1570. EXPECT_EQ(pikaMemNow(), 0);
  1571. }
  1572. TEST(pikaMain, list_sysobj) {
  1573. /* init */
  1574. pikaMemInfo.heapUsedMax = 0;
  1575. /* run */
  1576. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1577. /* skip the first obj_run */
  1578. obj_run(pikaMain,
  1579. "a = list()\n"
  1580. "a.len()\n");
  1581. /* as run in shell */
  1582. /* collect */
  1583. /* assert */
  1584. /* deinit */
  1585. obj_deinit(pikaMain);
  1586. EXPECT_EQ(pikaMemNow(), 0);
  1587. }
  1588. #if PIKA_BUILTIN_STRUCT_ENABLE
  1589. TEST(pikaMain, list_init) {
  1590. /* init */
  1591. pikaMemInfo.heapUsedMax = 0;
  1592. /* run */
  1593. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1594. __platform_printf("BEGIN\r\n");
  1595. obj_run(pikaMain,
  1596. "a = [1, 2, 3]\n"
  1597. "a.len()\n");
  1598. /* as run in shell */
  1599. /* collect */
  1600. /* assert */
  1601. EXPECT_STREQ("3\r\n", log_buff[0]);
  1602. EXPECT_STREQ("BEGIN\r\n", log_buff[1]);
  1603. /* deinit */
  1604. obj_deinit(pikaMain);
  1605. EXPECT_EQ(pikaMemNow(), 0);
  1606. }
  1607. #endif
  1608. TEST(pikaMain, ctypes) {
  1609. /* init */
  1610. pikaMemInfo.heapUsedMax = 0;
  1611. /* run */
  1612. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1613. char* lines =
  1614. "t = ctypes.Test()\n"
  1615. "sendbuf = b'\\x03\\x04\\x33\\x00\\x05'\n"
  1616. "rlen = ctypes.c_uint(0)\n"
  1617. "rcvbuf = ctypes.c_char_p('')\n"
  1618. "res = t.dc_cpuapdu_hex(5, sendbuf, rlen, rcvbuf)\n"
  1619. "t.print_rcv(rcvbuf)\n";
  1620. __platform_printf("%s\n", lines);
  1621. __platform_printf("BEGIN\r\n");
  1622. obj_run(pikaMain, lines);
  1623. /* as run in shell */
  1624. /* collect */
  1625. Arg* sendbuf = obj_getArg(pikaMain, "sendbuf");
  1626. Arg* rcvbuf = obj_getArg(pikaMain, "rcvbuf.value");
  1627. /* assert */
  1628. EXPECT_EQ(arg_getBytesSize(sendbuf), 5);
  1629. EXPECT_EQ(arg_getBytesSize(rcvbuf), 6);
  1630. /* deinit */
  1631. obj_deinit(pikaMain);
  1632. EXPECT_EQ(pikaMemNow(), 0);
  1633. }
  1634. TEST(pikaMain, import_err) {
  1635. /* init */
  1636. pikaMemInfo.heapUsedMax = 0;
  1637. /* run */
  1638. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1639. char* lines = "import qqpe\n";
  1640. __platform_printf("%s\n", lines);
  1641. __platform_printf("BEGIN\r\n");
  1642. obj_run(pikaMain, lines);
  1643. /* as run in shell */
  1644. /* collect */
  1645. /* assert */
  1646. EXPECT_STREQ(log_buff[2],
  1647. "ModuleNotFoundError: No module named 'qqpe'\r\n");
  1648. /* deinit */
  1649. obj_deinit(pikaMain);
  1650. EXPECT_EQ(pikaMemNow(), 0);
  1651. }
  1652. TEST(pikaMain, cmodule_in_py) {
  1653. /* init */
  1654. pikaMemInfo.heapUsedMax = 0;
  1655. /* run */
  1656. PikaObj* pikaMain = pikaScriptInit();
  1657. char* lines =
  1658. "import test_cmodule\n"
  1659. "test_cmodule.test_mem()\n";
  1660. __platform_printf("%s\n", lines);
  1661. __platform_printf("BEGIN\r\n");
  1662. obj_run(pikaMain, lines);
  1663. /* as run in shell */
  1664. /* collect */
  1665. /* assert */
  1666. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1667. EXPECT_STREQ(log_buff[1], "mem used max:\r\n");
  1668. /* deinit */
  1669. obj_deinit(pikaMain);
  1670. EXPECT_EQ(pikaMemNow(), 0);
  1671. }
  1672. TEST(pikaMain, class___str___print) {
  1673. /* init */
  1674. pikaMemInfo.heapUsedMax = 0;
  1675. /* run */
  1676. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1677. __platform_printf("BEGIN\r\n");
  1678. obj_run(self,
  1679. "class Test():\n"
  1680. " def __str__(self):\n"
  1681. " return 'test'\n"
  1682. "test = Test()\n"
  1683. "print(test)\n");
  1684. /* assert */
  1685. EXPECT_STREQ(log_buff[0], "test\r\n");
  1686. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1687. /* deinit */
  1688. obj_deinit(self);
  1689. EXPECT_EQ(pikaMemNow(), 0);
  1690. }
  1691. TEST(pikaMain, CModule__str__) {
  1692. /* init */
  1693. pikaMemInfo.heapUsedMax = 0;
  1694. /* run */
  1695. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1696. __platform_printf("BEGIN\r\n");
  1697. obj_run(self,
  1698. "op = PikaMath.Operator()\n"
  1699. "print(op)\n"
  1700. "print('obj: ' + str(op))\n");
  1701. /* assert */
  1702. EXPECT_STREQ(log_buff[0], "obj: test\r\n");
  1703. EXPECT_STREQ(log_buff[1], "test\r\n");
  1704. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1705. /* deinit */
  1706. obj_deinit(self);
  1707. EXPECT_STREQ(log_buff[0], "del operator...\r\n");
  1708. EXPECT_EQ(pikaMemNow(), 0);
  1709. }
  1710. #if !PIKA_NANO_ENABLE
  1711. TEST(pikaMain, builtin_hex) {
  1712. /* init */
  1713. pikaMemInfo.heapUsedMax = 0;
  1714. /* run */
  1715. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1716. __platform_printf("BEGIN\r\n");
  1717. obj_run(self,
  1718. "hex(3)\n"
  1719. "hex(-3)\n"
  1720. "hex(1234)\n"
  1721. "hex(-12039)\n");
  1722. /* assert */
  1723. EXPECT_STREQ(log_buff[4], "BEGIN\r\n");
  1724. EXPECT_STREQ(log_buff[3], "'0x03'\r\n");
  1725. EXPECT_STREQ(log_buff[2], "'-0x03'\r\n");
  1726. EXPECT_STREQ(log_buff[1], "'0x4d2'\r\n");
  1727. EXPECT_STREQ(log_buff[0], "'-0x2f07'\r\n");
  1728. /* deinit */
  1729. obj_deinit(self);
  1730. EXPECT_EQ(pikaMemNow(), 0);
  1731. }
  1732. #endif
  1733. #if !PIKA_NANO_ENABLE
  1734. TEST(pikaMain, builtin_ord) {
  1735. /* init */
  1736. pikaMemInfo.heapUsedMax = 0;
  1737. /* run */
  1738. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1739. __platform_printf("BEGIN\r\n");
  1740. obj_run(self, "ord('a')\n");
  1741. /* assert */
  1742. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1743. EXPECT_STREQ(log_buff[0], "97\r\n");
  1744. /* deinit */
  1745. obj_deinit(self);
  1746. EXPECT_EQ(pikaMemNow(), 0);
  1747. }
  1748. #endif
  1749. #if !PIKA_NANO_ENABLE
  1750. TEST(pikaMain, builtin_chr) {
  1751. /* init */
  1752. pikaMemInfo.heapUsedMax = 0;
  1753. /* run */
  1754. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1755. __platform_printf("BEGIN\r\n");
  1756. obj_run(self, "chr(97)\n");
  1757. /* assert */
  1758. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1759. EXPECT_STREQ(log_buff[0], "'a'\r\n");
  1760. /* deinit */
  1761. obj_deinit(self);
  1762. EXPECT_EQ(pikaMemNow(), 0);
  1763. }
  1764. #endif
  1765. TEST(pikaMain, iteral_oct) {
  1766. /* init */
  1767. pikaMemInfo.heapUsedMax = 0;
  1768. /* run */
  1769. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1770. __platform_printf("BEGIN\r\n");
  1771. obj_run(self, "0o100\n");
  1772. /* assert */
  1773. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1774. EXPECT_STREQ(log_buff[0], "64\r\n");
  1775. /* deinit */
  1776. obj_deinit(self);
  1777. EXPECT_EQ(pikaMemNow(), 0);
  1778. }
  1779. TEST(pikaMain, iteral_bin) {
  1780. /* init */
  1781. pikaMemInfo.heapUsedMax = 0;
  1782. /* run */
  1783. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1784. __platform_printf("BEGIN\r\n");
  1785. obj_run(self, "0b100\n");
  1786. /* assert */
  1787. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1788. EXPECT_STREQ(log_buff[0], "4\r\n");
  1789. /* deinit */
  1790. obj_deinit(self);
  1791. EXPECT_EQ(pikaMemNow(), 0);
  1792. }
  1793. TEST(pikaMain, REPL_push_mode) {
  1794. /* init */
  1795. pikaMemInfo.heapUsedMax = 0;
  1796. /* run */
  1797. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1798. __platform_printf("BEGIN\r\n");
  1799. obj_runCharInit(self);
  1800. char lines[] =
  1801. "print('test')\n"
  1802. "for i in range(0, 10):\n"
  1803. " i\n"
  1804. "\n";
  1805. for (size_t i = 0; i < strGetSize(lines); i++) {
  1806. obj_runChar(self, lines[i]);
  1807. }
  1808. /* assert */
  1809. EXPECT_STREQ(log_buff[18], "BEGIN\r\n");
  1810. EXPECT_STREQ(log_buff[14], "test\r\n");
  1811. EXPECT_STREQ(log_buff[11], "... ");
  1812. EXPECT_STREQ(log_buff[1], "9\r\n");
  1813. EXPECT_STREQ(log_buff[0], ">>> ");
  1814. /* deinit */
  1815. obj_deinit(self);
  1816. EXPECT_EQ(pikaMemNow(), 0);
  1817. }
  1818. #if PIKA_SYNTAX_SLICE_ENABLE
  1819. TEST(pikaMain, int_from_bytes) {
  1820. /* init */
  1821. pikaMemInfo.heapUsedMax = 0;
  1822. /* run */
  1823. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1824. __platform_printf("BEGIN\r\n");
  1825. obj_run(self, "int(b'test'[0])\n");
  1826. /* assert */
  1827. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1828. EXPECT_STREQ(log_buff[0], "116\r\n");
  1829. /* deinit */
  1830. obj_deinit(self);
  1831. EXPECT_EQ(pikaMemNow(), 0);
  1832. }
  1833. #endif
  1834. // TEST(pikaMain, big_bytes) {
  1835. // /* init */
  1836. // pikaMemInfo.heapUsedMax = 0;
  1837. // /* run */
  1838. // PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1839. // __platform_printf("BEGIN\r\n");
  1840. // obj_run(self, "b = bytes(8192)\n");
  1841. // /* assert */
  1842. // /* deinit */
  1843. // obj_deinit(self);
  1844. // EXPECT_EQ(pikaMemNow(), 0);
  1845. // }
  1846. #if PIKA_SYNTAX_SLICE_ENABLE
  1847. TEST(pikaMain, utils_int_to_bytes) {
  1848. /* init */
  1849. pikaMemInfo.heapUsedMax = 0;
  1850. /* run */
  1851. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1852. __platform_printf("BEGIN\r\n");
  1853. obj_run(self,
  1854. "ut = PikaStdData.Utils()\n"
  1855. "b = b'test'\n"
  1856. "b[0] = ut.int_to_bytes(0x35)\n"
  1857. "b\n");
  1858. /* assert */
  1859. EXPECT_STREQ(log_buff[0], "b'\\x35\\x65\\x73\\x74'\r\n");
  1860. /* deinit */
  1861. obj_deinit(self);
  1862. EXPECT_EQ(pikaMemNow(), 0);
  1863. }
  1864. #endif
  1865. #if PIKA_SYNTAX_SLICE_ENABLE
  1866. TEST(pikaMain, neg_index) {
  1867. /* init */
  1868. pikaMemInfo.heapUsedMax = 0;
  1869. /* run */
  1870. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1871. __platform_printf("BEGIN\r\n");
  1872. obj_run(self,
  1873. "'test'[-2]\n"
  1874. "b'test'[-2]\n");
  1875. /* assert */
  1876. EXPECT_STREQ(log_buff[0], "b'\\x73'\r\n");
  1877. EXPECT_STREQ(log_buff[1], "'s'\r\n");
  1878. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1879. /* deinit */
  1880. obj_deinit(self);
  1881. EXPECT_EQ(pikaMemNow(), 0);
  1882. }
  1883. #endif
  1884. #if PIKA_SYNTAX_SLICE_ENABLE
  1885. TEST(pikaMain, slice1) {
  1886. /* init */
  1887. pikaMemInfo.heapUsedMax = 0;
  1888. /* run */
  1889. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1890. __platform_printf("BEGIN\r\n");
  1891. obj_run(self, "'test'[1:3]\n");
  1892. /* assert */
  1893. EXPECT_STREQ(log_buff[0], "'es'\r\n");
  1894. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1895. /* deinit */
  1896. obj_deinit(self);
  1897. EXPECT_EQ(pikaMemNow(), 0);
  1898. }
  1899. #endif
  1900. #if PIKA_SYNTAX_SLICE_ENABLE
  1901. TEST(pikaMain, slice2) {
  1902. /* init */
  1903. pikaMemInfo.heapUsedMax = 0;
  1904. /* run */
  1905. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1906. __platform_printf("BEGIN\r\n");
  1907. obj_run(self, "b'test'[1:3]\n");
  1908. /* assert */
  1909. EXPECT_STREQ(log_buff[0], "b'\\x65\\x73'\r\n");
  1910. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1911. /* deinit */
  1912. obj_deinit(self);
  1913. EXPECT_EQ(pikaMemNow(), 0);
  1914. }
  1915. #endif
  1916. TEST(pikaMain, c_buffer) {
  1917. /* init */
  1918. pikaMemInfo.heapUsedMax = 0;
  1919. /* run */
  1920. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1921. __platform_printf("BEGIN\r\n");
  1922. obj_run(self, "version = ctypes.c_buffer(b\"\", 128)\n");
  1923. /* assert */
  1924. /* deinit */
  1925. obj_deinit(self);
  1926. EXPECT_EQ(pikaMemNow(), 0);
  1927. }
  1928. TEST(pikaMain, str_add1) {
  1929. /* init */
  1930. pikaMemInfo.heapUsedMax = 0;
  1931. /* run */
  1932. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1933. __platform_printf("BEGIN\r\n");
  1934. obj_run(self,
  1935. "i = 32\n"
  1936. "msg = \"device_names[\" + str(i) + \"]:\"\n"
  1937. "msg\n");
  1938. /* assert */
  1939. EXPECT_STREQ(log_buff[0], "'device_names[32]:'\r\n");
  1940. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1941. /* deinit */
  1942. obj_deinit(self);
  1943. EXPECT_EQ(pikaMemNow(), 0);
  1944. }
  1945. #if PIKA_SYNTAX_SLICE_ENABLE
  1946. TEST(pikaMain, slice_eei) {
  1947. /* init */
  1948. pikaMemInfo.heapUsedMax = 0;
  1949. /* run */
  1950. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1951. __platform_printf("BEGIN\r\n");
  1952. obj_run(self, "'test'[1:-1]\n");
  1953. /* assert */
  1954. EXPECT_STREQ(log_buff[0], "'est'\r\n");
  1955. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1956. /* deinit */
  1957. obj_deinit(self);
  1958. EXPECT_EQ(pikaMemNow(), 0);
  1959. }
  1960. #endif
  1961. #if PIKA_SYNTAX_SLICE_ENABLE
  1962. TEST(pikaMain, slice_a97) {
  1963. /* init */
  1964. pikaMemInfo.heapUsedMax = 0;
  1965. /* run */
  1966. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1967. __platform_printf("BEGIN\r\n");
  1968. obj_run(self, "'test'[:-2]\n");
  1969. /* assert */
  1970. EXPECT_STREQ(log_buff[0], "'tes'\r\n");
  1971. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1972. /* deinit */
  1973. obj_deinit(self);
  1974. EXPECT_EQ(pikaMemNow(), 0);
  1975. }
  1976. #endif
  1977. #if PIKA_SYNTAX_SLICE_ENABLE
  1978. TEST(pikaMain, slice_a90) {
  1979. /* init */
  1980. pikaMemInfo.heapUsedMax = 0;
  1981. /* run */
  1982. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1983. __platform_printf("BEGIN\r\n");
  1984. obj_run(self, "'test'[1:]\n");
  1985. /* assert */
  1986. EXPECT_STREQ(log_buff[0], "'est'\r\n");
  1987. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1988. /* deinit */
  1989. obj_deinit(self);
  1990. EXPECT_EQ(pikaMemNow(), 0);
  1991. }
  1992. #endif
  1993. #if PIKA_SYNTAX_SLICE_ENABLE
  1994. TEST(pikaMain, slice_a12) {
  1995. /* init */
  1996. pikaMemInfo.heapUsedMax = 0;
  1997. /* run */
  1998. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1999. __platform_printf("BEGIN\r\n");
  2000. obj_run(self, "b'test'[1:]\n");
  2001. /* assert */
  2002. EXPECT_STREQ(log_buff[0], "b'\\x65\\x73\\x74'\r\n");
  2003. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2004. /* deinit */
  2005. obj_deinit(self);
  2006. EXPECT_EQ(pikaMemNow(), 0);
  2007. }
  2008. #endif
  2009. TEST(pikaMain, str_string) {
  2010. /* init */
  2011. pikaMemInfo.heapUsedMax = 0;
  2012. /* run */
  2013. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2014. __platform_printf("BEGIN\r\n");
  2015. obj_run(self,
  2016. "a = PikaStdData.String('test')\n"
  2017. "print(a)\n");
  2018. /* assert */
  2019. EXPECT_STREQ(log_buff[0], "test\r\n");
  2020. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2021. /* deinit */
  2022. obj_deinit(self);
  2023. EXPECT_EQ(pikaMemNow(), 0);
  2024. }
  2025. #if PIKA_SYNTAX_SLICE_ENABLE
  2026. TEST(pikaMain, string_index) {
  2027. /* init */
  2028. pikaMemInfo.heapUsedMax = 0;
  2029. /* run */
  2030. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2031. __platform_printf("BEGIN\r\n");
  2032. obj_run(self,
  2033. "a = PikaStdData.String('test')\n"
  2034. "a[1]\n"
  2035. "a[1] = 'q'\n"
  2036. "print(a)\n");
  2037. /* assert */
  2038. EXPECT_STREQ(log_buff[0], "tqst\r\n");
  2039. EXPECT_STREQ(log_buff[1], "'e'\r\n");
  2040. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  2041. /* deinit */
  2042. obj_deinit(self);
  2043. EXPECT_EQ(pikaMemNow(), 0);
  2044. }
  2045. #endif
  2046. TEST(pikaMain, list_for_append) {
  2047. /* init */
  2048. pikaMemInfo.heapUsedMax = 0;
  2049. /* run */
  2050. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2051. __platform_printf("BEGIN\r\n");
  2052. obj_run(self,
  2053. "rcv_buf = PikaStdData.List()\n"
  2054. "for i in range(0, 1024):\n"
  2055. " rcv_buf.append(0)\n"
  2056. "rcv_buf.len()\n");
  2057. /* assert */
  2058. EXPECT_STREQ(log_buff[0], "1024\r\n");
  2059. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2060. /* deinit */
  2061. obj_deinit(self);
  2062. EXPECT_EQ(pikaMemNow(), 0);
  2063. }
  2064. TEST(pikaMain, string_str) {
  2065. /* init */
  2066. pikaMemInfo.heapUsedMax = 0;
  2067. /* run */
  2068. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2069. __platform_printf("BEGIN\r\n");
  2070. obj_run(self, "str(PikaStdData.String('test'))\n");
  2071. /* assert */
  2072. EXPECT_STREQ(log_buff[0], "'test'\r\n");
  2073. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2074. /* deinit */
  2075. obj_deinit(self);
  2076. EXPECT_EQ(pikaMemNow(), 0);
  2077. }
  2078. TEST(pikaMain, module_import_from_module) {
  2079. pikaMemInfo.heapUsedMax = 0;
  2080. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2081. extern unsigned char pikaModules_py_a[];
  2082. obj_linkLibrary(self, pikaModules_py_a);
  2083. __platform_printf("BEGIN\r\n");
  2084. obj_run(self,
  2085. "import test_module1\n"
  2086. "test_module1.test_module_import()\n");
  2087. EXPECT_STREQ(log_buff[0], "test_module_2_hello\r\n");
  2088. EXPECT_STREQ(log_buff[1], "in test module 2\r\n");
  2089. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  2090. obj_deinit(self);
  2091. EXPECT_EQ(pikaMemNow(), 0);
  2092. }
  2093. TEST(pikaMain, module_1_module2_test) {
  2094. pikaMemInfo.heapUsedMax = 0;
  2095. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2096. extern unsigned char pikaModules_py_a[];
  2097. obj_linkLibrary(self, pikaModules_py_a);
  2098. __platform_printf("BEGIN\r\n");
  2099. obj_run(self,
  2100. "import test_module1\n"
  2101. "test_module1.test_module2.mytest()\n");
  2102. EXPECT_STREQ(log_buff[0], "test_module_2_hello\r\n");
  2103. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2104. obj_deinit(self);
  2105. EXPECT_EQ(pikaMemNow(), 0);
  2106. }
  2107. TEST(pikaMain, string_startswith) {
  2108. /* init */
  2109. pikaMemInfo.heapUsedMax = 0;
  2110. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2111. /* run */
  2112. obj_run(pikaMain,
  2113. "a = PikaStdData.String('test')\n"
  2114. "res1 = a.startswith('te')\n"
  2115. "res2 = a.startswith('st')\n");
  2116. /* collect */
  2117. int res1 = obj_getInt(pikaMain, "res1");
  2118. int res2 = obj_getInt(pikaMain, "res2");
  2119. /* assert */
  2120. EXPECT_EQ(res1, 1);
  2121. EXPECT_EQ(res2, 0);
  2122. /* deinit */
  2123. obj_deinit(pikaMain);
  2124. EXPECT_EQ(pikaMemNow(), 0);
  2125. }
  2126. TEST(pikaMain, string_endswith) {
  2127. /* init */
  2128. pikaMemInfo.heapUsedMax = 0;
  2129. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2130. /* run */
  2131. obj_run(pikaMain,
  2132. "a = PikaStdData.String('test')\n"
  2133. "res1 = a.endswith('te')\n"
  2134. "res2 = a.endswith('st')\n");
  2135. /* collect */
  2136. int res1 = obj_getInt(pikaMain, "res1");
  2137. int res2 = obj_getInt(pikaMain, "res2");
  2138. /* assert */
  2139. EXPECT_EQ(res1, 0);
  2140. EXPECT_EQ(res2, 1);
  2141. /* deinit */
  2142. obj_deinit(pikaMain);
  2143. EXPECT_EQ(pikaMemNow(), 0);
  2144. }
  2145. TEST(pikaMain, string_isdigit) {
  2146. /* init */
  2147. pikaMemInfo.heapUsedMax = 0;
  2148. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2149. /* run */
  2150. obj_run(pikaMain,
  2151. "a = PikaStdData.String('test')\n"
  2152. "b = PikaStdData.String('1234')\n"
  2153. "c = PikaStdData.String('test1234')\n"
  2154. "res1 = a.isdigit()\n"
  2155. "res2 = b.isdigit()\n"
  2156. "res3 = c.isdigit()\n");
  2157. /* collect */
  2158. int res1 = obj_getInt(pikaMain, "res1");
  2159. int res2 = obj_getInt(pikaMain, "res2");
  2160. int res3 = obj_getInt(pikaMain, "res3");
  2161. /* assert */
  2162. EXPECT_EQ(res1, 0);
  2163. EXPECT_EQ(res2, 1);
  2164. EXPECT_EQ(res3, 0);
  2165. /* deinit */
  2166. obj_deinit(pikaMain);
  2167. EXPECT_EQ(pikaMemNow(), 0);
  2168. }
  2169. TEST(pikaMain, string_islower) {
  2170. /* init */
  2171. pikaMemInfo.heapUsedMax = 0;
  2172. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2173. /* run */
  2174. obj_run(pikaMain,
  2175. "a = PikaStdData.String('test')\n"
  2176. "b = PikaStdData.String('1234')\n"
  2177. "c = PikaStdData.String('Test')\n"
  2178. "res1 = a.islower()\n"
  2179. "res2 = b.islower()\n"
  2180. "res3 = c.islower()\n");
  2181. /* collect */
  2182. int res1 = obj_getInt(pikaMain, "res1");
  2183. int res2 = obj_getInt(pikaMain, "res2");
  2184. int res3 = obj_getInt(pikaMain, "res3");
  2185. /* assert */
  2186. EXPECT_EQ(res1, 1);
  2187. EXPECT_EQ(res2, 0);
  2188. EXPECT_EQ(res3, 0);
  2189. /* deinit */
  2190. obj_deinit(pikaMain);
  2191. EXPECT_EQ(pikaMemNow(), 0);
  2192. }
  2193. TEST(pikaMain, string_isalnum) {
  2194. /* init */
  2195. pikaMemInfo.heapUsedMax = 0;
  2196. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2197. /* run */
  2198. obj_run(pikaMain,
  2199. "a = PikaStdData.String('test1234')\n"
  2200. "b = PikaStdData.String('1234')\n"
  2201. "c = PikaStdData.String(' ')\n"
  2202. "res1 = a.isalnum()\n"
  2203. "res2 = b.isalnum()\n"
  2204. "res3 = c.isalnum()\n");
  2205. /* collect */
  2206. int res1 = obj_getInt(pikaMain, "res1");
  2207. int res2 = obj_getInt(pikaMain, "res2");
  2208. int res3 = obj_getInt(pikaMain, "res3");
  2209. /* assert */
  2210. EXPECT_EQ(res1, 1);
  2211. EXPECT_EQ(res2, 1);
  2212. EXPECT_EQ(res3, 0);
  2213. /* deinit */
  2214. obj_deinit(pikaMain);
  2215. EXPECT_EQ(pikaMemNow(), 0);
  2216. }
  2217. TEST(pikaMain, string_isalpha) {
  2218. /* init */
  2219. pikaMemInfo.heapUsedMax = 0;
  2220. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2221. /* run */
  2222. obj_run(pikaMain,
  2223. "a = PikaStdData.String('test')\n"
  2224. "b = PikaStdData.String('1234')\n"
  2225. "c = PikaStdData.String(' ')\n"
  2226. "res1 = a.isalpha()\n"
  2227. "res2 = b.isalpha()\n"
  2228. "res3 = c.isalpha()\n");
  2229. /* collect */
  2230. int res1 = obj_getInt(pikaMain, "res1");
  2231. int res2 = obj_getInt(pikaMain, "res2");
  2232. int res3 = obj_getInt(pikaMain, "res3");
  2233. /* assert */
  2234. EXPECT_EQ(res1, 1);
  2235. EXPECT_EQ(res2, 0);
  2236. EXPECT_EQ(res3, 0);
  2237. /* deinit */
  2238. obj_deinit(pikaMain);
  2239. EXPECT_EQ(pikaMemNow(), 0);
  2240. }
  2241. TEST(pikaMain, string_isspace) {
  2242. /* init */
  2243. pikaMemInfo.heapUsedMax = 0;
  2244. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2245. /* run */
  2246. obj_run(pikaMain,
  2247. "a = PikaStdData.String('test')\n"
  2248. "b = PikaStdData.String(' ')\n"
  2249. "res1 = a.isspace()\n"
  2250. "res2 = b.isspace()\n");
  2251. /* collect */
  2252. int res1 = obj_getInt(pikaMain, "res1");
  2253. int res2 = obj_getInt(pikaMain, "res2");
  2254. /* assert */
  2255. EXPECT_EQ(res1, 0);
  2256. EXPECT_EQ(res2, 1);
  2257. /* deinit */
  2258. obj_deinit(pikaMain);
  2259. EXPECT_EQ(pikaMemNow(), 0);
  2260. }
  2261. TEST(pikaMain, print_obj) {
  2262. /* init */
  2263. pikaMemInfo.heapUsedMax = 0;
  2264. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2265. /* run */
  2266. __platform_printf("BEGIN\r\n");
  2267. obj_run(pikaMain,
  2268. "mem = PikaStdLib.MemChecker()\n"
  2269. "mem\n"
  2270. "print(mem)\n"
  2271. "res = str(mem)\n");
  2272. /* collect */
  2273. char* res = obj_getStr(pikaMain, "res");
  2274. /* assert */
  2275. EXPECT_EQ(strIsStartWith(log_buff[0], "<object at "), true);
  2276. EXPECT_EQ(strIsStartWith(log_buff[1], "<object at "), true);
  2277. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  2278. EXPECT_EQ(strIsStartWith(res, "<object at "), true);
  2279. /* deinit */
  2280. obj_deinit(pikaMain);
  2281. EXPECT_EQ(pikaMemNow(), 0);
  2282. }
  2283. TEST(pikaMain, num_issue1) {
  2284. /* init */
  2285. pikaMemInfo.heapUsedMax = 0;
  2286. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2287. /* run */
  2288. __platform_printf("BEGIN\r\n");
  2289. char* lines = "(((1 + (2 * 3)/(4 + 5))*(6 - 7) + (8 + 9) * 10)/11) - 12\n";
  2290. obj_run(pikaMain, lines);
  2291. /* collect */
  2292. /* assert */
  2293. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2294. EXPECT_STREQ(log_buff[0], "3.303030\r\n");
  2295. /* deinit */
  2296. obj_deinit(pikaMain);
  2297. EXPECT_EQ(pikaMemNow(), 0);
  2298. }
  2299. TEST(pikaMain, returnNullString) {
  2300. /* init */
  2301. pikaMemInfo.heapUsedMax = 0;
  2302. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2303. /* run */
  2304. __platform_printf("BEGIN\r\n");
  2305. char* lines =
  2306. "import GTestTask\n"
  2307. "task = GTestTask.Task()\n"
  2308. "def test_return_none():\n"
  2309. " s = task.returnNullString()\n"
  2310. " return s\n"
  2311. "s = test_return_none()\n"
  2312. "print(s)\n"
  2313. "if None == s:\n"
  2314. " print('s is None')\n";
  2315. obj_run(pikaMain, lines);
  2316. /* collect */
  2317. Arg* s = obj_getArg(pikaMain, "s");
  2318. /* assert */
  2319. EXPECT_EQ((uintptr_t)s, NULL);
  2320. /* deinit */
  2321. obj_deinit(pikaMain);
  2322. EXPECT_EQ(pikaMemNow(), 0);
  2323. }
  2324. TEST(pikaMain, for_loop_issue_1b2a3f1bdf) {
  2325. char* lines =
  2326. "res = 0\n"
  2327. "for i in range(0, 10):\n"
  2328. " if i < 3:\n"
  2329. " continue\n"
  2330. " if i > 7:\n"
  2331. " continue\n"
  2332. " for i in range(i, i+3):\n"
  2333. " res += i\n"
  2334. "\n";
  2335. Args* buffs = New_strBuff();
  2336. __platform_printf("%s\n", lines);
  2337. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  2338. __platform_printf("%s", pikaAsm);
  2339. args_deinit(buffs);
  2340. EXPECT_EQ(pikaMemNow(), 0);
  2341. /* init */
  2342. pikaMemInfo.heapUsedMax = 0;
  2343. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2344. /* run */
  2345. __platform_printf("BEGIN\r\n");
  2346. obj_run(pikaMain, lines);
  2347. /* collect */
  2348. int res = obj_getInt(pikaMain, "res");
  2349. /* assert */
  2350. EXPECT_EQ(res, 90);
  2351. /* deinit */
  2352. obj_deinit(pikaMain);
  2353. EXPECT_EQ(pikaMemNow(), 0);
  2354. }
  2355. TEST(pikaMain, syantex_issue123lkjxi) {
  2356. char* lines = "if i < 3\n";
  2357. /* init */
  2358. pikaMemInfo.heapUsedMax = 0;
  2359. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2360. /* run */
  2361. __platform_printf("BEGIN\r\n");
  2362. obj_run(pikaMain, lines);
  2363. /* collect */
  2364. /* assert */
  2365. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  2366. /* deinit */
  2367. obj_deinit(pikaMain);
  2368. EXPECT_EQ(pikaMemNow(), 0);
  2369. }
  2370. TEST(pikaMain, syantex_issue_fae13) {
  2371. char* lines = "for i < 3\n";
  2372. /* init */
  2373. pikaMemInfo.heapUsedMax = 0;
  2374. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2375. /* run */
  2376. __platform_printf("BEGIN\r\n");
  2377. obj_run(pikaMain, lines);
  2378. /* collect */
  2379. /* assert */
  2380. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  2381. /* deinit */
  2382. obj_deinit(pikaMain);
  2383. EXPECT_EQ(pikaMemNow(), 0);
  2384. }
  2385. TEST(pikaMain, syantex_issue_1289341) {
  2386. char* lines = "class lkj\n";
  2387. /* init */
  2388. pikaMemInfo.heapUsedMax = 0;
  2389. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2390. /* run */
  2391. __platform_printf("BEGIN\r\n");
  2392. obj_run(pikaMain, lines);
  2393. /* collect */
  2394. /* assert */
  2395. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  2396. /* deinit */
  2397. obj_deinit(pikaMain);
  2398. EXPECT_EQ(pikaMemNow(), 0);
  2399. }
  2400. TEST(pikaMain, syantex_issue_183571) {
  2401. char* lines = "def test\n";
  2402. /* init */
  2403. pikaMemInfo.heapUsedMax = 0;
  2404. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2405. /* run */
  2406. __platform_printf("BEGIN\r\n");
  2407. obj_run(pikaMain, lines);
  2408. /* collect */
  2409. /* assert */
  2410. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  2411. /* deinit */
  2412. obj_deinit(pikaMain);
  2413. EXPECT_EQ(pikaMemNow(), 0);
  2414. }
  2415. TEST(pikaMain, dump_issue_12l3kjioa) {
  2416. char* lines =
  2417. "if i = 1:\n"
  2418. " print('test')\n"
  2419. "\n";
  2420. /* init */
  2421. pikaMemInfo.heapUsedMax = 0;
  2422. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2423. /* run */
  2424. __platform_printf("BEGIN\r\n");
  2425. obj_run(pikaMain, lines);
  2426. /* collect */
  2427. /* assert */
  2428. /* deinit */
  2429. obj_deinit(pikaMain);
  2430. EXPECT_EQ(pikaMemNow(), 0);
  2431. }
  2432. #if PIKA_SYNTAX_SLICE_ENABLE
  2433. TEST(pikaMain, issue_fa13f4) {
  2434. char* lines =
  2435. "d = {'language':['c','python']}\n"
  2436. "d['language'].append('Java')\n"
  2437. "print(d)\n";
  2438. /* init */
  2439. pikaMemInfo.heapUsedMax = 0;
  2440. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2441. /* run */
  2442. __platform_printf("BEGIN\r\n");
  2443. obj_run(pikaMain, lines);
  2444. /* collect */
  2445. /* assert */
  2446. EXPECT_STREQ(log_buff[0], "{'language': ['c', 'python', 'Java']}\r\n");
  2447. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2448. /* deinit */
  2449. obj_deinit(pikaMain);
  2450. EXPECT_EQ(pikaMemNow(), 0);
  2451. }
  2452. #endif
  2453. TEST(pikaMain, callback_issue_I5L1MI) {
  2454. char* lines =
  2455. "def callback(func):\n"
  2456. " func()\n"
  2457. "\n"
  2458. "def test():\n"
  2459. " print('hello')\n"
  2460. "\n"
  2461. "f = test\n"
  2462. "callback(f)\n";
  2463. /* init */
  2464. pikaMemInfo.heapUsedMax = 0;
  2465. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2466. /* run */
  2467. __platform_printf("BEGIN\r\n");
  2468. obj_run(pikaMain, lines);
  2469. /* collect */
  2470. /* assert */
  2471. EXPECT_STREQ(log_buff[0], "hello\r\n");
  2472. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2473. /* deinit */
  2474. obj_deinit(pikaMain);
  2475. EXPECT_EQ(pikaMemNow(), 0);
  2476. }