pikaMain-test.cpp 84 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116
  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\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. int res = obj_getInt(pikaMain, "res");
  796. int res2 = obj_getInt(pikaMain, "res2");
  797. uint8_t* s = obj_getBytes(pikaMain, "s");
  798. /* assert */
  799. EXPECT_EQ(res, 115);
  800. EXPECT_EQ(res2, 114);
  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], PIKA_ERR_STRING_SYNTAX_ERROR "\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 = pika_lines2Asm(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], PIKA_ERR_STRING_SYNTAX_ERROR "\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], "ModuleNotFoundError: No module named 'qqpe'\n");
  1645. /* deinit */
  1646. obj_deinit(pikaMain);
  1647. EXPECT_EQ(pikaMemNow(), 0);
  1648. }
  1649. TEST(pikaMain, cmodule_in_py) {
  1650. /* init */
  1651. g_PikaMemInfo.heapUsedMax = 0;
  1652. /* run */
  1653. PikaObj* pikaMain = pikaScriptInit();
  1654. char* lines =
  1655. "import test_cmodule\n"
  1656. "test_cmodule.test_mem()\n";
  1657. __platform_printf("%s\n", lines);
  1658. __platform_printf("BEGIN\r\n");
  1659. obj_run(pikaMain, lines);
  1660. /* as run in shell */
  1661. /* collect */
  1662. /* assert */
  1663. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1664. EXPECT_STREQ(log_buff[1], "mem used max:\r\n");
  1665. /* deinit */
  1666. obj_deinit(pikaMain);
  1667. EXPECT_EQ(pikaMemNow(), 0);
  1668. }
  1669. TEST(pikaMain, class___str___print) {
  1670. /* init */
  1671. g_PikaMemInfo.heapUsedMax = 0;
  1672. /* run */
  1673. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1674. __platform_printf("BEGIN\r\n");
  1675. obj_run(self,
  1676. "class Test():\n"
  1677. " def __str__(self):\n"
  1678. " return 'test'\n"
  1679. "test = Test()\n"
  1680. "print(test)\n");
  1681. /* assert */
  1682. EXPECT_STREQ(log_buff[0], "test\r\n");
  1683. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1684. /* deinit */
  1685. obj_deinit(self);
  1686. EXPECT_EQ(pikaMemNow(), 0);
  1687. }
  1688. TEST(pikaMain, CModule__str__) {
  1689. /* init */
  1690. g_PikaMemInfo.heapUsedMax = 0;
  1691. /* run */
  1692. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1693. __platform_printf("BEGIN\r\n");
  1694. obj_run(self,
  1695. "op = PikaMath.Operator()\n"
  1696. "print(op)\n"
  1697. "print('obj: ' + str(op))\n");
  1698. /* assert */
  1699. EXPECT_STREQ(log_buff[0], "obj: test\r\n");
  1700. EXPECT_STREQ(log_buff[1], "test\r\n");
  1701. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1702. /* deinit */
  1703. obj_deinit(self);
  1704. EXPECT_STREQ(log_buff[0], "del operator...\r\n");
  1705. EXPECT_EQ(pikaMemNow(), 0);
  1706. }
  1707. #if !PIKA_NANO_ENABLE
  1708. TEST(pikaMain, builtin_hex) {
  1709. /* init */
  1710. g_PikaMemInfo.heapUsedMax = 0;
  1711. /* run */
  1712. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1713. __platform_printf("BEGIN\r\n");
  1714. obj_run(self,
  1715. "hex(3)\n"
  1716. "hex(-3)\n"
  1717. "hex(1234)\n"
  1718. "hex(-12039)\n");
  1719. /* assert */
  1720. EXPECT_STREQ(log_buff[4], "BEGIN\r\n");
  1721. EXPECT_STREQ(log_buff[3], "'0x03'\r\n");
  1722. EXPECT_STREQ(log_buff[2], "'-0x03'\r\n");
  1723. EXPECT_STREQ(log_buff[1], "'0x4d2'\r\n");
  1724. EXPECT_STREQ(log_buff[0], "'-0x2f07'\r\n");
  1725. /* deinit */
  1726. obj_deinit(self);
  1727. EXPECT_EQ(pikaMemNow(), 0);
  1728. }
  1729. #endif
  1730. #if !PIKA_NANO_ENABLE
  1731. TEST(pikaMain, builtin_ord) {
  1732. /* init */
  1733. g_PikaMemInfo.heapUsedMax = 0;
  1734. /* run */
  1735. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1736. __platform_printf("BEGIN\r\n");
  1737. obj_run(self, "ord('a')\n");
  1738. /* assert */
  1739. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1740. EXPECT_STREQ(log_buff[0], "97\r\n");
  1741. /* deinit */
  1742. obj_deinit(self);
  1743. EXPECT_EQ(pikaMemNow(), 0);
  1744. }
  1745. #endif
  1746. #if !PIKA_NANO_ENABLE
  1747. TEST(pikaMain, builtin_chr) {
  1748. /* init */
  1749. g_PikaMemInfo.heapUsedMax = 0;
  1750. /* run */
  1751. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1752. __platform_printf("BEGIN\r\n");
  1753. obj_run(self, "chr(97)\n");
  1754. /* assert */
  1755. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1756. EXPECT_STREQ(log_buff[0], "'a'\r\n");
  1757. /* deinit */
  1758. obj_deinit(self);
  1759. EXPECT_EQ(pikaMemNow(), 0);
  1760. }
  1761. #endif
  1762. TEST(pikaMain, iteral_oct) {
  1763. /* init */
  1764. g_PikaMemInfo.heapUsedMax = 0;
  1765. /* run */
  1766. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1767. __platform_printf("BEGIN\r\n");
  1768. obj_run(self, "0o100\n");
  1769. /* assert */
  1770. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1771. EXPECT_STREQ(log_buff[0], "64\r\n");
  1772. /* deinit */
  1773. obj_deinit(self);
  1774. EXPECT_EQ(pikaMemNow(), 0);
  1775. }
  1776. TEST(pikaMain, iteral_bin) {
  1777. /* init */
  1778. g_PikaMemInfo.heapUsedMax = 0;
  1779. /* run */
  1780. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1781. __platform_printf("BEGIN\r\n");
  1782. obj_run(self, "0b100\n");
  1783. /* assert */
  1784. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1785. EXPECT_STREQ(log_buff[0], "4\r\n");
  1786. /* deinit */
  1787. obj_deinit(self);
  1788. EXPECT_EQ(pikaMemNow(), 0);
  1789. }
  1790. TEST(pikaMain, REPL_push_mode) {
  1791. /* init */
  1792. g_PikaMemInfo.heapUsedMax = 0;
  1793. /* run */
  1794. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1795. __platform_printf("BEGIN\r\n");
  1796. char lines[] =
  1797. "print('test')\n"
  1798. "for i in range(0, 10):\n"
  1799. " i\n"
  1800. "\n";
  1801. for (size_t i = 0; i < strGetSize(lines); i++) {
  1802. obj_runChar(self, lines[i]);
  1803. }
  1804. /* assert */
  1805. // EXPECT_STREQ(log_buff[18], "BEGIN\r\n");
  1806. EXPECT_STREQ(log_buff[14], "test\r\n");
  1807. EXPECT_STREQ(log_buff[11], "... ");
  1808. EXPECT_STREQ(log_buff[1], "9\r\n");
  1809. EXPECT_STREQ(log_buff[0], ">>> ");
  1810. /* deinit */
  1811. obj_deinit(self);
  1812. EXPECT_EQ(pikaMemNow(), 0);
  1813. }
  1814. TEST(pikaMain, REPL_push_mode_err) {
  1815. /* init */
  1816. g_PikaMemInfo.heapUsedMax = 0;
  1817. /* run */
  1818. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1819. __platform_printf("BEGIN\r\n");
  1820. char lines[] =
  1821. "def test:\n"
  1822. " print('test')\n"
  1823. "\n";
  1824. for (size_t i = 0; i < strGetSize(lines); i++) {
  1825. obj_runChar(self, lines[i]);
  1826. }
  1827. /* assert */
  1828. EXPECT_STREQ(log_buff[1], PIKA_ERR_STRING_SYNTAX_ERROR "\n");
  1829. /* deinit */
  1830. obj_deinit(self);
  1831. EXPECT_EQ(pikaMemNow(), 0);
  1832. }
  1833. #if 0
  1834. TEST(pikaMain, REPL_input) {
  1835. /* init */
  1836. g_PikaMemInfo.heapUsedMax = 0;
  1837. /* run */
  1838. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1839. __platform_printf("BEGIN\r\n");
  1840. char lines[] =
  1841. "res = input('input:\\n')\n"
  1842. "1234\n"
  1843. "print(res)\n";
  1844. for (size_t i = 0; i < strGetSize(lines); i++) {
  1845. obj_runChar(self, lines[i]);
  1846. }
  1847. /* assert */
  1848. /* deinit */
  1849. obj_deinit(self);
  1850. EXPECT_EQ(pikaMemNow(), 0);
  1851. }
  1852. #endif
  1853. #if PIKA_SYNTAX_SLICE_ENABLE
  1854. TEST(pikaMain, int_from_bytes) {
  1855. /* init */
  1856. g_PikaMemInfo.heapUsedMax = 0;
  1857. /* run */
  1858. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1859. __platform_printf("BEGIN\r\n");
  1860. obj_run(self, "int(b'test'[0])\n");
  1861. /* assert */
  1862. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1863. EXPECT_STREQ(log_buff[0], "116\r\n");
  1864. /* deinit */
  1865. obj_deinit(self);
  1866. EXPECT_EQ(pikaMemNow(), 0);
  1867. }
  1868. #endif
  1869. // TEST(pikaMain, big_bytes) {
  1870. // /* init */
  1871. // g_PikaMemInfo.heapUsedMax = 0;
  1872. // /* run */
  1873. // PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1874. // __platform_printf("BEGIN\r\n");
  1875. // obj_run(self, "b = bytes(8192)\n");
  1876. // /* assert */
  1877. // /* deinit */
  1878. // obj_deinit(self);
  1879. // EXPECT_EQ(pikaMemNow(), 0);
  1880. // }
  1881. #if PIKA_SYNTAX_SLICE_ENABLE
  1882. TEST(pikaMain, utils_int_to_bytes) {
  1883. /* init */
  1884. g_PikaMemInfo.heapUsedMax = 0;
  1885. /* run */
  1886. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1887. __platform_printf("BEGIN\r\n");
  1888. obj_run(self,
  1889. "ut = PikaStdData.Utils()\n"
  1890. "b = b'test'\n"
  1891. "b[0] = ut.int_to_bytes(0x35)\n"
  1892. "b\n");
  1893. /* assert */
  1894. EXPECT_STREQ(log_buff[0], "b'\\x35\\x65\\x73\\x74'\r\n");
  1895. /* deinit */
  1896. obj_deinit(self);
  1897. EXPECT_EQ(pikaMemNow(), 0);
  1898. }
  1899. #endif
  1900. #if PIKA_SYNTAX_SLICE_ENABLE
  1901. TEST(pikaMain, neg_index) {
  1902. /* init */
  1903. g_PikaMemInfo.heapUsedMax = 0;
  1904. /* run */
  1905. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1906. __platform_printf("BEGIN\r\n");
  1907. obj_run(self,
  1908. "'test'[-2]\n"
  1909. "b'test'[-2]\n");
  1910. /* assert */
  1911. EXPECT_STREQ(log_buff[0], "115\r\n");
  1912. EXPECT_STREQ(log_buff[1], "'s'\r\n");
  1913. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1914. /* deinit */
  1915. obj_deinit(self);
  1916. EXPECT_EQ(pikaMemNow(), 0);
  1917. }
  1918. #endif
  1919. #if PIKA_SYNTAX_SLICE_ENABLE
  1920. TEST(pikaMain, slice1) {
  1921. /* init */
  1922. g_PikaMemInfo.heapUsedMax = 0;
  1923. /* run */
  1924. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1925. __platform_printf("BEGIN\r\n");
  1926. obj_run(self, "'test'[1:3]\n");
  1927. /* assert */
  1928. EXPECT_STREQ(log_buff[0], "'es'\r\n");
  1929. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1930. /* deinit */
  1931. obj_deinit(self);
  1932. EXPECT_EQ(pikaMemNow(), 0);
  1933. }
  1934. #endif
  1935. #if PIKA_SYNTAX_SLICE_ENABLE
  1936. TEST(pikaMain, slice2) {
  1937. /* init */
  1938. g_PikaMemInfo.heapUsedMax = 0;
  1939. /* run */
  1940. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1941. __platform_printf("BEGIN\r\n");
  1942. obj_run(self, "b'test'[1:3]\n");
  1943. /* assert */
  1944. EXPECT_STREQ(log_buff[0], "b'\\x65\\x73'\r\n");
  1945. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1946. /* deinit */
  1947. obj_deinit(self);
  1948. EXPECT_EQ(pikaMemNow(), 0);
  1949. }
  1950. #endif
  1951. TEST(pikaMain, c_buffer) {
  1952. /* init */
  1953. g_PikaMemInfo.heapUsedMax = 0;
  1954. /* run */
  1955. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1956. __platform_printf("BEGIN\r\n");
  1957. obj_run(self, "version = ctypes.c_buffer(b\"\", 128)\n");
  1958. /* assert */
  1959. /* deinit */
  1960. obj_deinit(self);
  1961. EXPECT_EQ(pikaMemNow(), 0);
  1962. }
  1963. TEST(pikaMain, str_add1) {
  1964. /* init */
  1965. g_PikaMemInfo.heapUsedMax = 0;
  1966. /* run */
  1967. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1968. __platform_printf("BEGIN\r\n");
  1969. obj_run(self,
  1970. "i = 32\n"
  1971. "msg = \"device_names[\" + str(i) + \"]:\"\n"
  1972. "msg\n");
  1973. /* assert */
  1974. EXPECT_STREQ(log_buff[0], "'device_names[32]:'\r\n");
  1975. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1976. /* deinit */
  1977. obj_deinit(self);
  1978. EXPECT_EQ(pikaMemNow(), 0);
  1979. }
  1980. #if PIKA_SYNTAX_SLICE_ENABLE
  1981. TEST(pikaMain, slice_eei) {
  1982. /* init */
  1983. g_PikaMemInfo.heapUsedMax = 0;
  1984. /* run */
  1985. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1986. __platform_printf("BEGIN\r\n");
  1987. obj_run(self, "'test'[1:-1]\n");
  1988. /* assert */
  1989. EXPECT_STREQ(log_buff[0], "'es'\r\n");
  1990. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1991. /* deinit */
  1992. obj_deinit(self);
  1993. EXPECT_EQ(pikaMemNow(), 0);
  1994. }
  1995. #endif
  1996. #if PIKA_SYNTAX_SLICE_ENABLE
  1997. TEST(pikaMain, slice_a97) {
  1998. /* init */
  1999. g_PikaMemInfo.heapUsedMax = 0;
  2000. /* run */
  2001. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2002. __platform_printf("BEGIN\r\n");
  2003. obj_run(self, "'test'[:-2]\n");
  2004. /* assert */
  2005. EXPECT_STREQ(log_buff[0], "'te'\r\n");
  2006. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2007. /* deinit */
  2008. obj_deinit(self);
  2009. EXPECT_EQ(pikaMemNow(), 0);
  2010. }
  2011. #endif
  2012. #if PIKA_SYNTAX_SLICE_ENABLE
  2013. TEST(pikaMain, slice_a90) {
  2014. /* init */
  2015. g_PikaMemInfo.heapUsedMax = 0;
  2016. /* run */
  2017. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2018. __platform_printf("BEGIN\r\n");
  2019. obj_run(self, "'test'[1:]\n");
  2020. /* assert */
  2021. EXPECT_STREQ(log_buff[0], "'est'\r\n");
  2022. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2023. /* deinit */
  2024. obj_deinit(self);
  2025. EXPECT_EQ(pikaMemNow(), 0);
  2026. }
  2027. #endif
  2028. #if PIKA_SYNTAX_SLICE_ENABLE
  2029. TEST(pikaMain, slice_a12) {
  2030. /* init */
  2031. g_PikaMemInfo.heapUsedMax = 0;
  2032. /* run */
  2033. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2034. __platform_printf("BEGIN\r\n");
  2035. obj_run(self, "b'test'[1:]\n");
  2036. /* assert */
  2037. EXPECT_STREQ(log_buff[0], "b'\\x65\\x73\\x74'\r\n");
  2038. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2039. /* deinit */
  2040. obj_deinit(self);
  2041. EXPECT_EQ(pikaMemNow(), 0);
  2042. }
  2043. #endif
  2044. TEST(pikaMain, str_string) {
  2045. /* init */
  2046. g_PikaMemInfo.heapUsedMax = 0;
  2047. /* run */
  2048. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2049. __platform_printf("BEGIN\r\n");
  2050. obj_run(self,
  2051. "a = PikaStdData.String('test')\n"
  2052. "print(a)\n");
  2053. /* assert */
  2054. EXPECT_STREQ(log_buff[0], "test\r\n");
  2055. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2056. /* deinit */
  2057. obj_deinit(self);
  2058. EXPECT_EQ(pikaMemNow(), 0);
  2059. }
  2060. #if PIKA_SYNTAX_SLICE_ENABLE
  2061. TEST(pikaMain, string_index) {
  2062. /* init */
  2063. g_PikaMemInfo.heapUsedMax = 0;
  2064. /* run */
  2065. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2066. __platform_printf("BEGIN\r\n");
  2067. obj_run(self,
  2068. "a = PikaStdData.String('test')\n"
  2069. "a[1]\n"
  2070. "a[1] = 'q'\n"
  2071. "print(a)\n");
  2072. /* assert */
  2073. EXPECT_STREQ(log_buff[0], "tqst\r\n");
  2074. EXPECT_STREQ(log_buff[1], "'e'\r\n");
  2075. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  2076. /* deinit */
  2077. obj_deinit(self);
  2078. EXPECT_EQ(pikaMemNow(), 0);
  2079. }
  2080. #endif
  2081. TEST(pikaMain, list_for_append) {
  2082. /* init */
  2083. g_PikaMemInfo.heapUsedMax = 0;
  2084. /* run */
  2085. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2086. __platform_printf("BEGIN\r\n");
  2087. obj_run(self,
  2088. "rcv_buf = PikaStdData.List()\n"
  2089. "for i in range(0, 1024):\n"
  2090. " rcv_buf.append(0)\n"
  2091. "rcv_buf.len()\n");
  2092. /* assert */
  2093. EXPECT_STREQ(log_buff[0], "1024\r\n");
  2094. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2095. /* deinit */
  2096. obj_deinit(self);
  2097. EXPECT_EQ(pikaMemNow(), 0);
  2098. }
  2099. TEST(pikaMain, string_str) {
  2100. /* init */
  2101. g_PikaMemInfo.heapUsedMax = 0;
  2102. /* run */
  2103. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2104. __platform_printf("BEGIN\r\n");
  2105. obj_run(self, "str(PikaStdData.String('test'))\n");
  2106. /* assert */
  2107. EXPECT_STREQ(log_buff[0], "'test'\r\n");
  2108. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2109. /* deinit */
  2110. obj_deinit(self);
  2111. EXPECT_EQ(pikaMemNow(), 0);
  2112. }
  2113. TEST(pikaMain, module_import_from_module) {
  2114. g_PikaMemInfo.heapUsedMax = 0;
  2115. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2116. extern unsigned char pikaModules_py_a[];
  2117. obj_linkLibrary(self, pikaModules_py_a);
  2118. __platform_printf("BEGIN\r\n");
  2119. obj_run(self,
  2120. "import test_module1\n"
  2121. "test_module1.test_module_import()\n");
  2122. EXPECT_STREQ(log_buff[0], "test_module_2_hello\r\n");
  2123. EXPECT_STREQ(log_buff[1], "in test module 2\r\n");
  2124. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  2125. obj_deinit(self);
  2126. EXPECT_EQ(pikaMemNow(), 0);
  2127. }
  2128. TEST(pikaMain, module_1_module2_test) {
  2129. g_PikaMemInfo.heapUsedMax = 0;
  2130. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2131. extern unsigned char pikaModules_py_a[];
  2132. obj_linkLibrary(self, pikaModules_py_a);
  2133. __platform_printf("BEGIN\r\n");
  2134. obj_run(self,
  2135. "import test_module1\n"
  2136. "test_module1.test_module2.mytest()\n");
  2137. EXPECT_STREQ(log_buff[0], "test_module_2_hello\r\n");
  2138. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2139. obj_deinit(self);
  2140. EXPECT_EQ(pikaMemNow(), 0);
  2141. }
  2142. TEST(pikaMain, string_startswith) {
  2143. /* init */
  2144. g_PikaMemInfo.heapUsedMax = 0;
  2145. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2146. /* run */
  2147. obj_run(pikaMain,
  2148. "a = PikaStdData.String('test')\n"
  2149. "res1 = a.startswith('te')\n"
  2150. "res2 = a.startswith('st')\n");
  2151. /* collect */
  2152. int res1 = obj_getInt(pikaMain, "res1");
  2153. int res2 = obj_getInt(pikaMain, "res2");
  2154. /* assert */
  2155. EXPECT_EQ(res1, 1);
  2156. EXPECT_EQ(res2, 0);
  2157. /* deinit */
  2158. obj_deinit(pikaMain);
  2159. EXPECT_EQ(pikaMemNow(), 0);
  2160. }
  2161. TEST(pikaMain, string_endswith) {
  2162. /* init */
  2163. g_PikaMemInfo.heapUsedMax = 0;
  2164. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2165. /* run */
  2166. obj_run(pikaMain,
  2167. "a = PikaStdData.String('test')\n"
  2168. "res1 = a.endswith('te')\n"
  2169. "res2 = a.endswith('st')\n");
  2170. /* collect */
  2171. int res1 = obj_getInt(pikaMain, "res1");
  2172. int res2 = obj_getInt(pikaMain, "res2");
  2173. /* assert */
  2174. EXPECT_EQ(res1, 0);
  2175. EXPECT_EQ(res2, 1);
  2176. /* deinit */
  2177. obj_deinit(pikaMain);
  2178. EXPECT_EQ(pikaMemNow(), 0);
  2179. }
  2180. TEST(pikaMain, string_isdigit) {
  2181. /* init */
  2182. g_PikaMemInfo.heapUsedMax = 0;
  2183. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2184. /* run */
  2185. obj_run(pikaMain,
  2186. "a = PikaStdData.String('test')\n"
  2187. "b = PikaStdData.String('1234')\n"
  2188. "c = PikaStdData.String('test1234')\n"
  2189. "res1 = a.isdigit()\n"
  2190. "res2 = b.isdigit()\n"
  2191. "res3 = c.isdigit()\n");
  2192. /* collect */
  2193. int res1 = obj_getInt(pikaMain, "res1");
  2194. int res2 = obj_getInt(pikaMain, "res2");
  2195. int res3 = obj_getInt(pikaMain, "res3");
  2196. /* assert */
  2197. EXPECT_EQ(res1, 0);
  2198. EXPECT_EQ(res2, 1);
  2199. EXPECT_EQ(res3, 0);
  2200. /* deinit */
  2201. obj_deinit(pikaMain);
  2202. EXPECT_EQ(pikaMemNow(), 0);
  2203. }
  2204. TEST(pikaMain, string_islower) {
  2205. /* init */
  2206. g_PikaMemInfo.heapUsedMax = 0;
  2207. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2208. /* run */
  2209. obj_run(pikaMain,
  2210. "a = PikaStdData.String('test')\n"
  2211. "b = PikaStdData.String('1234')\n"
  2212. "c = PikaStdData.String('Test')\n"
  2213. "res1 = a.islower()\n"
  2214. "res2 = b.islower()\n"
  2215. "res3 = c.islower()\n");
  2216. /* collect */
  2217. int res1 = obj_getInt(pikaMain, "res1");
  2218. int res2 = obj_getInt(pikaMain, "res2");
  2219. int res3 = obj_getInt(pikaMain, "res3");
  2220. /* assert */
  2221. EXPECT_EQ(res1, 1);
  2222. EXPECT_EQ(res2, 0);
  2223. EXPECT_EQ(res3, 0);
  2224. /* deinit */
  2225. obj_deinit(pikaMain);
  2226. EXPECT_EQ(pikaMemNow(), 0);
  2227. }
  2228. TEST(pikaMain, string_isalnum) {
  2229. /* init */
  2230. g_PikaMemInfo.heapUsedMax = 0;
  2231. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2232. /* run */
  2233. obj_run(pikaMain,
  2234. "a = PikaStdData.String('test1234')\n"
  2235. "b = PikaStdData.String('1234')\n"
  2236. "c = PikaStdData.String(' ')\n"
  2237. "res1 = a.isalnum()\n"
  2238. "res2 = b.isalnum()\n"
  2239. "res3 = c.isalnum()\n");
  2240. /* collect */
  2241. int res1 = obj_getInt(pikaMain, "res1");
  2242. int res2 = obj_getInt(pikaMain, "res2");
  2243. int res3 = obj_getInt(pikaMain, "res3");
  2244. /* assert */
  2245. EXPECT_EQ(res1, 1);
  2246. EXPECT_EQ(res2, 1);
  2247. EXPECT_EQ(res3, 0);
  2248. /* deinit */
  2249. obj_deinit(pikaMain);
  2250. EXPECT_EQ(pikaMemNow(), 0);
  2251. }
  2252. TEST(pikaMain, string_isalpha) {
  2253. /* init */
  2254. g_PikaMemInfo.heapUsedMax = 0;
  2255. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2256. /* run */
  2257. obj_run(pikaMain,
  2258. "a = PikaStdData.String('test')\n"
  2259. "b = PikaStdData.String('1234')\n"
  2260. "c = PikaStdData.String(' ')\n"
  2261. "res1 = a.isalpha()\n"
  2262. "res2 = b.isalpha()\n"
  2263. "res3 = c.isalpha()\n");
  2264. /* collect */
  2265. int res1 = obj_getInt(pikaMain, "res1");
  2266. int res2 = obj_getInt(pikaMain, "res2");
  2267. int res3 = obj_getInt(pikaMain, "res3");
  2268. /* assert */
  2269. EXPECT_EQ(res1, 1);
  2270. EXPECT_EQ(res2, 0);
  2271. EXPECT_EQ(res3, 0);
  2272. /* deinit */
  2273. obj_deinit(pikaMain);
  2274. EXPECT_EQ(pikaMemNow(), 0);
  2275. }
  2276. TEST(pikaMain, string_isspace) {
  2277. /* init */
  2278. g_PikaMemInfo.heapUsedMax = 0;
  2279. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2280. /* run */
  2281. obj_run(pikaMain,
  2282. "a = PikaStdData.String('test')\n"
  2283. "b = PikaStdData.String(' ')\n"
  2284. "res1 = a.isspace()\n"
  2285. "res2 = b.isspace()\n");
  2286. /* collect */
  2287. int res1 = obj_getInt(pikaMain, "res1");
  2288. int res2 = obj_getInt(pikaMain, "res2");
  2289. /* assert */
  2290. EXPECT_EQ(res1, 0);
  2291. EXPECT_EQ(res2, 1);
  2292. /* deinit */
  2293. obj_deinit(pikaMain);
  2294. EXPECT_EQ(pikaMemNow(), 0);
  2295. }
  2296. TEST(pikaMain, print_obj) {
  2297. /* init */
  2298. g_PikaMemInfo.heapUsedMax = 0;
  2299. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2300. /* run */
  2301. __platform_printf("BEGIN\r\n");
  2302. obj_run(pikaMain,
  2303. "mem = PikaStdLib.MemChecker()\n"
  2304. "mem\n"
  2305. "print(mem)\n"
  2306. "res = str(mem)\n");
  2307. /* collect */
  2308. char* res = obj_getStr(pikaMain, "res");
  2309. /* assert */
  2310. EXPECT_EQ(strIsStartWith(log_buff[0], "<object at "), true);
  2311. EXPECT_EQ(strIsStartWith(log_buff[1], "<object at "), true);
  2312. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  2313. EXPECT_EQ(strIsStartWith(res, "<object at "), true);
  2314. /* deinit */
  2315. obj_deinit(pikaMain);
  2316. EXPECT_EQ(pikaMemNow(), 0);
  2317. }
  2318. TEST(pikaMain, num_issue1) {
  2319. /* init */
  2320. g_PikaMemInfo.heapUsedMax = 0;
  2321. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2322. /* run */
  2323. __platform_printf("BEGIN\r\n");
  2324. char* lines = "(((1 + (2 * 3)/(4 + 5))*(6 - 7) + (8 + 9) * 10)/11) - 12\n";
  2325. obj_run(pikaMain, lines);
  2326. /* collect */
  2327. /* assert */
  2328. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2329. EXPECT_STREQ(log_buff[0], "3.303030\r\n");
  2330. /* deinit */
  2331. obj_deinit(pikaMain);
  2332. EXPECT_EQ(pikaMemNow(), 0);
  2333. }
  2334. TEST(pikaMain, returnNullString) {
  2335. /* init */
  2336. g_PikaMemInfo.heapUsedMax = 0;
  2337. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2338. /* run */
  2339. __platform_printf("BEGIN\r\n");
  2340. char* lines =
  2341. "import GTestTask\n"
  2342. "task = GTestTask.Task()\n"
  2343. "def test_return_none():\n"
  2344. " s = task.returnNullString()\n"
  2345. " return s\n"
  2346. "s = test_return_none()\n"
  2347. "print(s)\n"
  2348. "if None == s:\n"
  2349. " print('s is None')\n";
  2350. obj_run(pikaMain, lines);
  2351. /* collect */
  2352. Arg* s = obj_getArg(pikaMain, "s");
  2353. /* assert */
  2354. EXPECT_EQ((uintptr_t)s, NULL);
  2355. /* deinit */
  2356. obj_deinit(pikaMain);
  2357. EXPECT_EQ(pikaMemNow(), 0);
  2358. }
  2359. TEST(pikaMain, for_loop_issue_1b2a3f1bdf) {
  2360. char* lines =
  2361. "res = 0\n"
  2362. "for i in range(0, 10):\n"
  2363. " if i < 3:\n"
  2364. " continue\n"
  2365. " if i > 7:\n"
  2366. " continue\n"
  2367. " for i in range(i, i+3):\n"
  2368. " res += i\n"
  2369. "\n";
  2370. Args* buffs = New_strBuff();
  2371. __platform_printf("%s\n", lines);
  2372. char* pikaAsm = pika_lines2Asm(buffs, lines);
  2373. __platform_printf("%s", pikaAsm);
  2374. args_deinit(buffs);
  2375. EXPECT_EQ(pikaMemNow(), 0);
  2376. /* init */
  2377. g_PikaMemInfo.heapUsedMax = 0;
  2378. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2379. /* run */
  2380. __platform_printf("BEGIN\r\n");
  2381. obj_run(pikaMain, lines);
  2382. /* collect */
  2383. int res = obj_getInt(pikaMain, "res");
  2384. /* assert */
  2385. EXPECT_EQ(res, 90);
  2386. /* deinit */
  2387. obj_deinit(pikaMain);
  2388. EXPECT_EQ(pikaMemNow(), 0);
  2389. }
  2390. #define pika_floatYPE_DOUBLE 0
  2391. TEST(pikaMain, syantex_issue123lkjxi) {
  2392. char* lines = "if i < 3\n";
  2393. /* init */
  2394. g_PikaMemInfo.heapUsedMax = 0;
  2395. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2396. /* run */
  2397. __platform_printf("BEGIN\r\n");
  2398. obj_run(pikaMain, lines);
  2399. /* collect */
  2400. /* assert */
  2401. EXPECT_STREQ(log_buff[0], PIKA_ERR_STRING_SYNTAX_ERROR "\n");
  2402. /* deinit */
  2403. obj_deinit(pikaMain);
  2404. EXPECT_EQ(pikaMemNow(), 0);
  2405. }
  2406. TEST(pikaMain, syantex_issue_fae13) {
  2407. char* lines = "for i < 3\n";
  2408. /* init */
  2409. g_PikaMemInfo.heapUsedMax = 0;
  2410. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2411. /* run */
  2412. __platform_printf("BEGIN\r\n");
  2413. obj_run(pikaMain, lines);
  2414. /* collect */
  2415. /* assert */
  2416. EXPECT_STREQ(log_buff[0], PIKA_ERR_STRING_SYNTAX_ERROR "\n");
  2417. /* deinit */
  2418. obj_deinit(pikaMain);
  2419. EXPECT_EQ(pikaMemNow(), 0);
  2420. }
  2421. TEST(pikaMain, syantex_issue_1289341) {
  2422. char* lines = "class lkj\n";
  2423. /* init */
  2424. g_PikaMemInfo.heapUsedMax = 0;
  2425. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2426. /* run */
  2427. __platform_printf("BEGIN\r\n");
  2428. obj_run(pikaMain, lines);
  2429. /* collect */
  2430. /* assert */
  2431. EXPECT_STREQ(log_buff[0], PIKA_ERR_STRING_SYNTAX_ERROR "\n");
  2432. /* deinit */
  2433. obj_deinit(pikaMain);
  2434. EXPECT_EQ(pikaMemNow(), 0);
  2435. }
  2436. TEST(pikaMain, syantex_issue_183571) {
  2437. char* lines = "def test\n";
  2438. /* init */
  2439. g_PikaMemInfo.heapUsedMax = 0;
  2440. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2441. /* run */
  2442. __platform_printf("BEGIN\r\n");
  2443. obj_run(pikaMain, lines);
  2444. /* collect */
  2445. /* assert */
  2446. EXPECT_STREQ(log_buff[0], PIKA_ERR_STRING_SYNTAX_ERROR "\n");
  2447. /* deinit */
  2448. obj_deinit(pikaMain);
  2449. EXPECT_EQ(pikaMemNow(), 0);
  2450. }
  2451. TEST(pikaMain, dump_issue_12l3kjioa) {
  2452. char* lines =
  2453. "if i = 1:\n"
  2454. " print('test')\n"
  2455. "\n";
  2456. /* init */
  2457. g_PikaMemInfo.heapUsedMax = 0;
  2458. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2459. /* run */
  2460. __platform_printf("BEGIN\r\n");
  2461. obj_run(pikaMain, lines);
  2462. /* collect */
  2463. /* assert */
  2464. /* deinit */
  2465. obj_deinit(pikaMain);
  2466. EXPECT_EQ(pikaMemNow(), 0);
  2467. }
  2468. #if PIKA_SYNTAX_SLICE_ENABLE
  2469. TEST(pikaMain, issue_fa13f4) {
  2470. char* lines =
  2471. "d = {'language':['c','python']}\n"
  2472. "d['language'].append('Java')\n"
  2473. "print(d)\n";
  2474. /* init */
  2475. g_PikaMemInfo.heapUsedMax = 0;
  2476. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2477. /* run */
  2478. __platform_printf("BEGIN\r\n");
  2479. obj_run(pikaMain, lines);
  2480. /* collect */
  2481. /* assert */
  2482. EXPECT_STREQ(log_buff[0], "{'language': ['c', 'python', 'Java']}\r\n");
  2483. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2484. /* deinit */
  2485. obj_deinit(pikaMain);
  2486. EXPECT_EQ(pikaMemNow(), 0);
  2487. }
  2488. #endif
  2489. TEST(pikaMain, callback_issue_I5L1MI) {
  2490. char* lines =
  2491. "def callback(func):\n"
  2492. " func()\n"
  2493. "\n"
  2494. "def test():\n"
  2495. " print('hello')\n"
  2496. "\n"
  2497. "f = test\n"
  2498. "callback(f)\n";
  2499. /* init */
  2500. g_PikaMemInfo.heapUsedMax = 0;
  2501. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2502. /* run */
  2503. __platform_printf("BEGIN\r\n");
  2504. obj_run(pikaMain, lines);
  2505. /* collect */
  2506. /* assert */
  2507. EXPECT_STREQ(log_buff[0], "hello\r\n");
  2508. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2509. /* deinit */
  2510. obj_deinit(pikaMain);
  2511. EXPECT_EQ(pikaMemNow(), 0);
  2512. }
  2513. TEST(pikaMain, runchar_multiline) {
  2514. char* lines =
  2515. "def test():\r\n"
  2516. " print('hello')\r\n"
  2517. "\r\n"
  2518. "test()\r\n";
  2519. /* init */
  2520. g_PikaMemInfo.heapUsedMax = 0;
  2521. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2522. /* run */
  2523. __platform_printf("BEGIN\r\n");
  2524. for (size_t i = 0; i < strGetSize(lines); i++) {
  2525. obj_runChar(pikaMain, lines[i]);
  2526. }
  2527. /* collect */
  2528. /* assert */
  2529. EXPECT_STREQ(log_buff[1], "hello\r\n");
  2530. /* deinit */
  2531. obj_deinit(pikaMain);
  2532. EXPECT_EQ(pikaMemNow(), 0);
  2533. }
  2534. TEST(pikaMain, callback_run_char) {
  2535. char* lines =
  2536. "def test2():\r\n"
  2537. " print('hello')\r\n"
  2538. "\r\n"
  2539. "def test():\r\n"
  2540. " print('hello')\r\n"
  2541. " print('in test')\r\n"
  2542. "\r\n"
  2543. "def test3():\r\n"
  2544. " print('hello3')\r\n"
  2545. "\r\n"
  2546. "test()\r\n";
  2547. /* init */
  2548. g_PikaMemInfo.heapUsedMax = 0;
  2549. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2550. /* run */
  2551. __platform_printf("BEGIN\r\n");
  2552. for (size_t i = 0; i < strGetSize(lines); i++) {
  2553. obj_runChar(pikaMain, lines[i]);
  2554. }
  2555. /* collect */
  2556. /* assert */
  2557. EXPECT_STREQ(log_buff[1], "in test\r\n");
  2558. /* deinit */
  2559. obj_deinit(pikaMain);
  2560. EXPECT_EQ(pikaMemNow(), 0);
  2561. }
  2562. TEST(pikaMain, REPL_backspace) {
  2563. char* lines = "print('test'\b\b')\r\n";
  2564. /* init */
  2565. g_PikaMemInfo.heapUsedMax = 0;
  2566. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2567. /* run */
  2568. __platform_printf("BEGIN\r\n");
  2569. for (size_t i = 0; i < strGetSize(lines); i++) {
  2570. obj_runChar(pikaMain, lines[i]);
  2571. }
  2572. /* collect */
  2573. /* assert */
  2574. EXPECT_STREQ(log_buff[1], "tes\r\n");
  2575. /* deinit */
  2576. obj_deinit(pikaMain);
  2577. EXPECT_EQ(pikaMemNow(), 0);
  2578. }
  2579. #if PIKA_TAB_ENABLE
  2580. TEST(pikaMain, REPL_tab_completion) {
  2581. char lines[] = {'p', 'r', 0x09, '(', '\"', 't', 'e',
  2582. 's', 't', '\"', ')', '\r', '\n', 0x00};
  2583. /* init */
  2584. g_PikaMemInfo.heapUsedMax = 0;
  2585. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2586. /* run */
  2587. __platform_printf("BEGIN\r\n");
  2588. for (size_t i = 0; i < strGetSize(lines); i++) {
  2589. obj_runChar(pikaMain, lines[i]);
  2590. }
  2591. /* collect */
  2592. /* assert */
  2593. EXPECT_STREQ(log_buff[1], "test\r\n");
  2594. /* deinit */
  2595. obj_deinit(pikaMain);
  2596. EXPECT_EQ(pikaMemNow(), 0);
  2597. }
  2598. #endif
  2599. TEST(pikaMain, REPL_backspace_issue_1) {
  2600. 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";
  2601. /* init */
  2602. g_PikaMemInfo.heapUsedMax = 0;
  2603. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2604. /* run */
  2605. __platform_printf("BEGIN\r\n");
  2606. for (size_t i = 0; i < strGetSize(lines); i++) {
  2607. obj_runChar(pikaMain, lines[i]);
  2608. }
  2609. /* collect */
  2610. /* assert */
  2611. EXPECT_STREQ(log_buff[1], "'a'\r\n");
  2612. /* deinit */
  2613. obj_deinit(pikaMain);
  2614. EXPECT_EQ(pikaMemNow(), 0);
  2615. }
  2616. TEST(pikaMain, REPL_backspace_issue_2) {
  2617. char* lines = "loop\bp\n";
  2618. /* init */
  2619. g_PikaMemInfo.heapUsedMax = 0;
  2620. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2621. /* run */
  2622. __platform_printf("BEGIN\r\n");
  2623. for (size_t i = 0; i < strGetSize(lines); i++) {
  2624. obj_runChar(pikaMain, lines[i]);
  2625. }
  2626. /* collect */
  2627. /* assert */
  2628. /* deinit */
  2629. obj_deinit(pikaMain);
  2630. EXPECT_EQ(pikaMemNow(), 0);
  2631. }
  2632. TEST(pikaMain, REPL_key_left) {
  2633. char lines[] = {'1', '2', '+', '3', '4', 0x1b,
  2634. 0x5b, PIKA_KEY_LEFT, '5', '\n', 0x00};
  2635. /* init */
  2636. g_PikaMemInfo.heapUsedMax = 0;
  2637. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2638. /* run */
  2639. __platform_printf("BEGIN\r\n");
  2640. for (size_t i = 0; i < strGetSize(lines); i++) {
  2641. obj_runChar(pikaMain, lines[i]);
  2642. }
  2643. /* collect */
  2644. /* assert */
  2645. EXPECT_STREQ(log_buff[1], "366\r\n");
  2646. /* deinit */
  2647. obj_deinit(pikaMain);
  2648. EXPECT_EQ(pikaMemNow(), 0);
  2649. }
  2650. TEST(pikaMain, REPL_key_left_del) {
  2651. char lines[] = {'1', '2', '+', '3', '4', 0x1b,
  2652. 0x5b, PIKA_KEY_LEFT, '\b', '\n', 0x00};
  2653. /* init */
  2654. g_PikaMemInfo.heapUsedMax = 0;
  2655. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2656. /* run */
  2657. __platform_printf("BEGIN\r\n");
  2658. for (size_t i = 0; i < strGetSize(lines); i++) {
  2659. obj_runChar(pikaMain, lines[i]);
  2660. }
  2661. /* collect */
  2662. /* assert */
  2663. EXPECT_STREQ(log_buff[1], "16\r\n");
  2664. /* deinit */
  2665. obj_deinit(pikaMain);
  2666. EXPECT_EQ(pikaMemNow(), 0);
  2667. }
  2668. #if PIKA_SHELL_HISTORY_ENABLE
  2669. TEST(pikaMain, REPL_key_up) {
  2670. char lines[] = {'1', '2', '+', '3', '4', '5', '\n',
  2671. 0x1b, 0x5b, PIKA_KEY_UP, '2', '\n', 0x00};
  2672. /* init */
  2673. g_PikaMemInfo.heapUsedMax = 0;
  2674. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2675. /* run */
  2676. __platform_printf("BEGIN\r\n");
  2677. for (size_t i = 0; i < strGetSize(lines); i++) {
  2678. obj_runChar(pikaMain, lines[i]);
  2679. }
  2680. /* collect */
  2681. /* assert */
  2682. EXPECT_STREQ(log_buff[1], "3464\r\n");
  2683. /* deinit */
  2684. obj_deinit(pikaMain);
  2685. EXPECT_EQ(pikaMemNow(), 0);
  2686. }
  2687. TEST(pikaMain, REPL_key_up_void) {
  2688. char lines[] = {0x1b, 0x5b, PIKA_KEY_UP, '2', '\n', 0x00};
  2689. /* init */
  2690. g_PikaMemInfo.heapUsedMax = 0;
  2691. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2692. /* run */
  2693. __platform_printf("BEGIN\r\n");
  2694. for (size_t i = 0; i < strGetSize(lines); i++) {
  2695. obj_runChar(pikaMain, lines[i]);
  2696. }
  2697. /* collect */
  2698. /* assert */
  2699. EXPECT_STREQ(log_buff[1], "2\r\n");
  2700. /* deinit */
  2701. obj_deinit(pikaMain);
  2702. EXPECT_EQ(pikaMemNow(), 0);
  2703. }
  2704. TEST(pikaMain, REPL_key_up_2) {
  2705. char lines[] = {'1', '2', '+', '3', '4', '5', '\n',
  2706. '1', '3', '*', '2', '\n', 0x1b, 0x5b,
  2707. PIKA_KEY_UP, 0x1b, 0x5b, PIKA_KEY_UP, '2', '\n', 0x00};
  2708. /* init */
  2709. g_PikaMemInfo.heapUsedMax = 0;
  2710. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2711. /* run */
  2712. __platform_printf("BEGIN\r\n");
  2713. for (size_t i = 0; i < strGetSize(lines); i++) {
  2714. obj_runChar(pikaMain, lines[i]);
  2715. }
  2716. /* collect */
  2717. /* assert */
  2718. EXPECT_STREQ(log_buff[1], "3464\r\n");
  2719. /* deinit */
  2720. obj_deinit(pikaMain);
  2721. EXPECT_EQ(pikaMemNow(), 0);
  2722. }
  2723. TEST(pikaMain, REPL_key_down) {
  2724. char lines[] = {'1', '2', '+', '3', '4', '5',
  2725. '\n', '1', '3', '*', '2', '\n',
  2726. 0x1b, 0x5b, PIKA_KEY_UP, 0x1b, 0x5b, PIKA_KEY_UP,
  2727. 0x1b, 0x5b, PIKA_KEY_DOWN, '2', '\n', 0x00};
  2728. /* init */
  2729. g_PikaMemInfo.heapUsedMax = 0;
  2730. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2731. /* run */
  2732. __platform_printf("BEGIN\r\n");
  2733. for (size_t i = 0; i < strGetSize(lines); i++) {
  2734. obj_runChar(pikaMain, lines[i]);
  2735. }
  2736. /* collect */
  2737. /* assert */
  2738. EXPECT_STREQ(log_buff[1], "286\r\n");
  2739. /* deinit */
  2740. obj_deinit(pikaMain);
  2741. EXPECT_EQ(pikaMemNow(), 0);
  2742. }
  2743. TEST(pikaMain, REPL_key_down_over) {
  2744. char lines[] = {'1', '2', '+', '3', '4', '5',
  2745. '\n', '1', '3', '*', '2', '\n',
  2746. 0x1b, 0x5b, PIKA_KEY_UP, 0x1b, 0x5b, PIKA_KEY_UP,
  2747. 0x1b, 0x5b, PIKA_KEY_DOWN, 0x1b, 0x5b, PIKA_KEY_DOWN,
  2748. 0x1b, 0x5b, PIKA_KEY_DOWN, '2', '\n', 0x00};
  2749. /* init */
  2750. g_PikaMemInfo.heapUsedMax = 0;
  2751. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2752. /* run */
  2753. __platform_printf("BEGIN\r\n");
  2754. for (size_t i = 0; i < strGetSize(lines); i++) {
  2755. obj_runChar(pikaMain, lines[i]);
  2756. }
  2757. /* collect */
  2758. /* assert */
  2759. EXPECT_STREQ(log_buff[1], "2\r\n");
  2760. /* deinit */
  2761. obj_deinit(pikaMain);
  2762. EXPECT_EQ(pikaMemNow(), 0);
  2763. }
  2764. TEST(pikaMain, REPL_pdb_set_break) {
  2765. char lines[] = "pdb.set_break('pdb_set_break', 36)\nn\nn\nn\n";
  2766. write_to_getchar_buffer(lines, strGetSize(lines));
  2767. /* init */
  2768. g_PikaMemInfo.heapUsedMax = 0;
  2769. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2770. extern unsigned char pikaModules_py_a[];
  2771. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2772. pikaVM_runSingleFile(pikaMain, "test/python/builtins/pdb_set_break.py");
  2773. /* run */
  2774. __platform_printf("BEGIN\r\n");
  2775. /* collect */
  2776. /* assert */
  2777. // EXPECT_STREQ(log_buff[1], "2\r\n");
  2778. /* deinit */
  2779. obj_deinit(pikaMain);
  2780. EXPECT_EQ(pikaMemNow(), 0);
  2781. }
  2782. #endif
  2783. #if PIKA_SHELL_FILTER_ENABLE
  2784. TEST(pikaMain, SHELL_filter_hi_pika) {
  2785. char lines[] = {"###Hi Pika"};
  2786. /* init */
  2787. g_PikaMemInfo.heapUsedMax = 0;
  2788. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2789. /* run */
  2790. __platform_printf("BEGIN\r\n");
  2791. for (size_t i = 0; i < strGetSize(lines); i++) {
  2792. obj_runChar(pikaMain, lines[i]);
  2793. }
  2794. /* collect */
  2795. /* assert */
  2796. EXPECT_STREQ(log_buff[0], "Yes, I am here\r\n");
  2797. /* deinit */
  2798. obj_deinit(pikaMain);
  2799. EXPECT_EQ(pikaMemNow(), 0);
  2800. }
  2801. TEST(pikaMain, SHELL_filter_hi_pika_sence) {
  2802. char lines[] = {"###hi pika"};
  2803. /* init */
  2804. g_PikaMemInfo.heapUsedMax = 0;
  2805. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2806. /* run */
  2807. __platform_printf("BEGIN\r\n");
  2808. for (size_t i = 0; i < strGetSize(lines); i++) {
  2809. obj_runChar(pikaMain, lines[i]);
  2810. }
  2811. /* collect */
  2812. /* assert */
  2813. EXPECT_STREQ(log_buff[0], ">>> ");
  2814. /* deinit */
  2815. obj_deinit(pikaMain);
  2816. EXPECT_EQ(pikaMemNow(), 0);
  2817. }
  2818. TEST(pikaMain, SHELL_filter_bye_pika_sence) {
  2819. char lines[] = {"###Bye piKa"};
  2820. /* init */
  2821. g_PikaMemInfo.heapUsedMax = 0;
  2822. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2823. /* run */
  2824. __platform_printf("BEGIN\r\n");
  2825. for (size_t i = 0; i < strGetSize(lines); i++) {
  2826. obj_runChar(pikaMain, lines[i]);
  2827. }
  2828. /* collect */
  2829. /* assert */
  2830. EXPECT_STREQ(log_buff[0], "OK, see you\r\n");
  2831. /* deinit */
  2832. obj_deinit(pikaMain);
  2833. EXPECT_EQ(pikaMemNow(), 0);
  2834. }
  2835. TEST(pikaMain, SHELL_filter_bye_pika_nomache) {
  2836. char lines[] = {"print('###Hi Pika')\nprint('###Hi_Pika')\n"};
  2837. /* init */
  2838. g_PikaMemInfo.heapUsedMax = 0;
  2839. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2840. /* run */
  2841. __platform_printf("BEGIN\r\n");
  2842. for (size_t i = 0; i < strGetSize(lines); i++) {
  2843. obj_runChar(pikaMain, lines[i]);
  2844. }
  2845. /* collect */
  2846. /* assert */
  2847. EXPECT_STREQ(log_buff[1], "###Hi_Pika\r\n");
  2848. EXPECT_STREQ(log_buff[4], "Yes, I am here\r\n");
  2849. /* deinit */
  2850. obj_deinit(pikaMain);
  2851. EXPECT_EQ(pikaMemNow(), 0);
  2852. }
  2853. TEST(pikaMain, mem_now) {
  2854. /* init */
  2855. g_PikaMemInfo.heapUsedMax = 0;
  2856. /* run */
  2857. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2858. __platform_printf("BEGIN\r\n");
  2859. obj_run(pikaMain,
  2860. "from PikaStdLib import MemChecker as mem\n"
  2861. "mem.now()\n");
  2862. /* collect */
  2863. /* assert */
  2864. /* deinit */
  2865. obj_deinit(pikaMain);
  2866. EXPECT_EQ(pikaMemNow(), 0);
  2867. }
  2868. #endif
  2869. TEST(pikaMain, obj_setNone) {
  2870. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2871. obj_setNone(pikaMain, "n");
  2872. obj_run(pikaMain, "print(n)");
  2873. EXPECT_STREQ(log_buff[0], "None\r\n");
  2874. obj_deinit(pikaMain);
  2875. EXPECT_EQ(pikaMemNow(), 0);
  2876. }
  2877. TEST(pikaMain, obj_setStr_NULL) {
  2878. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2879. obj_setStr(pikaMain, "n", NULL);
  2880. obj_run(pikaMain, "print(n)");
  2881. EXPECT_STREQ(log_buff[2], "NameError: name 'n' is not defined\n");
  2882. obj_deinit(pikaMain);
  2883. EXPECT_EQ(pikaMemNow(), 0);
  2884. }
  2885. TEST(pikaMain, _obj_set_str_null) {
  2886. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2887. extern unsigned char pikaModules_py_a[];
  2888. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2889. obj_run(pikaMain,
  2890. "class _test(GTestTask._test):\n"
  2891. " def __init__(self):\n"
  2892. " super().__init__()\n"
  2893. "t = _test()\n");
  2894. obj_deinit(pikaMain);
  2895. EXPECT_EQ(pikaMemNow(), 0);
  2896. }
  2897. #if !PIKA_NANO_ENABLE
  2898. TEST(pikaMain, shell_nolib_print_help) {
  2899. char lines[] = {"help(\"modules\")\n"};
  2900. /* init */
  2901. g_PikaMemInfo.heapUsedMax = 0;
  2902. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2903. /* run */
  2904. for (size_t i = 0; i < strGetSize(lines); i++) {
  2905. obj_runChar(pikaMain, lines[i]);
  2906. }
  2907. /* assert */
  2908. EXPECT_STREQ(
  2909. log_buff[1],
  2910. "Error: Not found LibObj, please execute obj_linkLibrary()\r\n");
  2911. /* deinit */
  2912. obj_deinit(pikaMain);
  2913. EXPECT_EQ(pikaMemNow(), 0);
  2914. }
  2915. #endif
  2916. TEST_END