pikaMain-test.cpp 79 KB

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