pikaMain-test.cpp 76 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864
  1. #include "test_common.h"
  2. TEST_START
  3. TEST(pikaMain, init) {
  4. pikaMemInfo.heapUsedMax = 0;
  5. PikaObj* pikaMain = pikaScriptInit();
  6. obj_deinit(pikaMain);
  7. EXPECT_EQ(pikaMemNow(), 0);
  8. }
  9. TEST(pikaMain, new_mem) {
  10. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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, -999999999);
  244. /* deinit */
  245. obj_deinit(pikaMain);
  246. EXPECT_EQ(pikaMemNow(), 0);
  247. }
  248. TEST(pikaMain, list_method) {
  249. /* init */
  250. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. 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. pikaMemInfo.heapUsedMax = 0;
  645. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  646. /* run */
  647. __platform_printf("BEGIN\r\n");
  648. obj_run(pikaMain, "s = PikaStdData.String()\n");
  649. /* collect */
  650. /* assert */
  651. EXPECT_STREQ(log_buff[3], "BEGIN\r\n");
  652. EXPECT_STREQ(log_buff[2],
  653. "TypeError: __init__() takes 1 positional argument but 0 were "
  654. "given\r\n");
  655. /* deinit */
  656. obj_deinit(pikaMain);
  657. EXPECT_EQ(pikaMemNow(), 0);
  658. }
  659. TEST(pikaMain, obj_no_free) {
  660. /* init */
  661. 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. TEST(pikaMain, list__setitem__) {
  676. /* init */
  677. pikaMemInfo.heapUsedMax = 0;
  678. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  679. /* run */
  680. __platform_printf("BEGIN\n");
  681. obj_run(pikaMain,
  682. "list = PikaStdData.List()\n"
  683. "list = __setitem__(list, 0, 2)\n"
  684. "res = __getitem__(list, 0)\n"
  685. "\n");
  686. /* collect */
  687. int res = obj_getInt(pikaMain, "res");
  688. /* assert */
  689. EXPECT_EQ(res, 2);
  690. /* deinit */
  691. obj_deinit(pikaMain);
  692. EXPECT_EQ(pikaMemNow(), 0);
  693. }
  694. TEST(pikaMain, string__getitem__) {
  695. /* init */
  696. pikaMemInfo.heapUsedMax = 0;
  697. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  698. /* run */
  699. __platform_printf("BEGIN\n");
  700. obj_run(pikaMain,
  701. "s = PikaStdData.String('test')\n"
  702. "res = __getitem__(s, 2)\n"
  703. "\n");
  704. /* collect */
  705. char* res = obj_getStr(pikaMain, "res");
  706. /* assert */
  707. EXPECT_STREQ(res, "s");
  708. /* deinit */
  709. obj_deinit(pikaMain);
  710. EXPECT_EQ(pikaMemNow(), 0);
  711. }
  712. TEST(pikaMain, dict__setitem__get) {
  713. /* init */
  714. pikaMemInfo.heapUsedMax = 0;
  715. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  716. /* run */
  717. __platform_printf("BEGIN\n");
  718. obj_run(pikaMain,
  719. "list = PikaStdData.Dict()\n"
  720. "list = __setitem__(list, 'a', 2)\n"
  721. "res = __getitem__(list, 'a')\n"
  722. "\n");
  723. /* collect */
  724. int res = obj_getInt(pikaMain, "res");
  725. /* assert */
  726. EXPECT_EQ(res, 2);
  727. /* deinit */
  728. obj_deinit(pikaMain);
  729. EXPECT_EQ(pikaMemNow(), 0);
  730. }
  731. TEST(pikaMain, str___getitem____setitem__) {
  732. /* init */
  733. pikaMemInfo.heapUsedMax = 0;
  734. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  735. /* run */
  736. __platform_printf("BEGIN\n");
  737. obj_run(pikaMain,
  738. "s = 'test'\n"
  739. "res = __getitem__(s, 2)\n"
  740. "s = __setitem__(s, 2, 'q')\n"
  741. "\n");
  742. /* collect */
  743. char* res = obj_getStr(pikaMain, "res");
  744. char* s = obj_getStr(pikaMain, "s");
  745. /* assert */
  746. EXPECT_STREQ(res, "s");
  747. EXPECT_STREQ(s, "teqt");
  748. /* deinit */
  749. obj_deinit(pikaMain);
  750. EXPECT_EQ(pikaMemNow(), 0);
  751. }
  752. #if PIKA_SYNTAX_SLICE_ENABLE
  753. TEST(pikaMain, str__index__) {
  754. /* init */
  755. pikaMemInfo.heapUsedMax = 0;
  756. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  757. /* run */
  758. __platform_printf("BEGIN\n");
  759. obj_run(pikaMain,
  760. "s = 'test'\n"
  761. "res = s[2]\n"
  762. "res2 = 'eqrt'[2]\n"
  763. "s[2] = 'q'\n"
  764. "\n");
  765. /* collect */
  766. char* res = obj_getStr(pikaMain, "res");
  767. char* res2 = obj_getStr(pikaMain, "res2");
  768. char* s = obj_getStr(pikaMain, "s");
  769. /* assert */
  770. EXPECT_STREQ(res, "s");
  771. EXPECT_STREQ(res2, "r");
  772. EXPECT_STREQ(s, "teqt");
  773. // EXPECT_STREQ(s, "teqt");
  774. /* deinit */
  775. obj_deinit(pikaMain);
  776. EXPECT_EQ(pikaMemNow(), 0);
  777. }
  778. #endif
  779. #if PIKA_SYNTAX_SLICE_ENABLE
  780. TEST(pikaMain, bytes__index__) {
  781. /* init */
  782. pikaMemInfo.heapUsedMax = 0;
  783. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  784. /* run */
  785. __platform_printf("BEGIN\n");
  786. obj_run(pikaMain,
  787. "s = b'test'\n"
  788. "res = s[2]\n"
  789. "res2 = b'eqrt'[2]\n"
  790. "s[2] = b'q'\n"
  791. "\n");
  792. /* collect */
  793. uint8_t* res = obj_getBytes(pikaMain, "res");
  794. uint8_t* res2 = obj_getBytes(pikaMain, "res2");
  795. uint8_t* s = obj_getBytes(pikaMain, "s");
  796. /* assert */
  797. EXPECT_STREQ((char*)res, "s");
  798. EXPECT_STREQ((char*)res2, "r");
  799. EXPECT_EQ(s[2], 'q');
  800. // EXPECT_STREQ(s, "teqt");
  801. /* deinit */
  802. obj_deinit(pikaMain);
  803. EXPECT_EQ(pikaMemNow(), 0);
  804. }
  805. #endif
  806. #if PIKA_SYNTAX_SLICE_ENABLE
  807. TEST(pikaMain, list_index) {
  808. /* init */
  809. pikaMemInfo.heapUsedMax = 0;
  810. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  811. /* run */
  812. __platform_printf("BEGIN\n");
  813. obj_run(pikaMain,
  814. "list = PikaStdData.List()\n"
  815. "list[0] = 2\n"
  816. "res = list[0]\n"
  817. "\n");
  818. /* collect */
  819. int res = obj_getInt(pikaMain, "res");
  820. /* assert */
  821. EXPECT_EQ(res, 2);
  822. /* deinit */
  823. obj_deinit(pikaMain);
  824. EXPECT_EQ(pikaMemNow(), 0);
  825. }
  826. #endif
  827. #if PIKA_SYNTAX_SLICE_ENABLE
  828. TEST(pikaMain, dict_index) {
  829. /* init */
  830. pikaMemInfo.heapUsedMax = 0;
  831. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  832. /* run */
  833. __platform_printf("BEGIN\n");
  834. obj_run(pikaMain,
  835. "dict = PikaStdData.Dict()\n"
  836. "dict['a'] = 2\n"
  837. "res = dict['a']\n"
  838. "\n");
  839. /* collect */
  840. int res = obj_getInt(pikaMain, "res");
  841. /* assert */
  842. EXPECT_EQ(res, 2);
  843. /* deinit */
  844. obj_deinit(pikaMain);
  845. EXPECT_EQ(pikaMemNow(), 0);
  846. }
  847. #endif
  848. TEST(pikaMain, task_run_once) {
  849. /* init */
  850. pikaMemInfo.heapUsedMax = 0;
  851. /* run */
  852. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  853. obj_run(pikaMain,
  854. "def todo1():\n"
  855. " print('task 1 running...')\n"
  856. "def todo2():\n"
  857. " print('task 2 running...')\n"
  858. "task = GTestTask.Task()\n"
  859. "task.call_always(todo1)\n"
  860. "task.call_always(todo2)\n"
  861. "task.run_once()\n"
  862. "\n");
  863. /* collect */
  864. /* assert */
  865. EXPECT_STREQ(log_buff[0], "task 2 running...\r\n");
  866. EXPECT_STREQ(log_buff[1], "task 1 running...\r\n");
  867. /* deinit */
  868. obj_deinit(pikaMain);
  869. EXPECT_EQ(pikaMemNow(), 0);
  870. }
  871. TEST(pikaMain, task_run_when) {
  872. /* init */
  873. pikaMemInfo.heapUsedMax = 0;
  874. /* run */
  875. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  876. obj_run(pikaMain,
  877. "def todo1():\n"
  878. " print('task 1 running...')\n"
  879. "def todo2():\n"
  880. " print('task 2 running...')\n"
  881. "def todo3():\n"
  882. " print('task 3 running...')\n"
  883. "def when3():\n"
  884. " return True\n"
  885. "task = GTestTask.Task()\n"
  886. "task.call_always(todo1)\n"
  887. "task.call_always(todo2)\n"
  888. "task.call_when(todo3, when3)\n"
  889. "task.run_once()\n"
  890. "\n");
  891. /* collect */
  892. /* assert */
  893. EXPECT_STREQ(log_buff[0], "task 3 running...\r\n");
  894. EXPECT_STREQ(log_buff[1], "task 2 running...\r\n");
  895. EXPECT_STREQ(log_buff[2], "task 1 running...\r\n");
  896. /* deinit */
  897. obj_deinit(pikaMain);
  898. EXPECT_EQ(pikaMemNow(), 0);
  899. }
  900. #if !PIKA_NANO_ENABLE
  901. TEST(pikaMain, task_run_period_until) {
  902. /* init */
  903. pikaMemInfo.heapUsedMax = 0;
  904. /* run */
  905. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  906. __platform_printf("BEGIN\r\n");
  907. obj_run(pikaMain,
  908. "import PikaStdLib\n"
  909. "mem = PikaStdLib.MemChecker()\n"
  910. "mem_now = 0\n"
  911. "mem_max = 0\n"
  912. "mem_lack = False\n"
  913. "def todo1():\n"
  914. " global mem_now\n"
  915. " global mem_lack\n"
  916. " global mem_max\n"
  917. " mem_max = mem.getMax()\n"
  918. " print('task 1 running...')\n"
  919. " if mem_now == 0:\n"
  920. " mem_now = mem.getNow()\n"
  921. " else:\n"
  922. " if mem.getNow() > mem_now + 1:\n"
  923. " mem_lack = True\n"
  924. " print(cformat('%0.8f', mem.getNow()))\n"
  925. "def todo2():\n"
  926. " print('task 2 running...')\n"
  927. "def todo3():\n"
  928. " print('task 3 running...')\n"
  929. "def when3():\n"
  930. " return True\n"
  931. "task = GTestTask.Task()\n"
  932. "task.call_period_ms(todo1, 200)\n"
  933. "task.call_period_ms(todo2, 500)\n"
  934. "# task.call_always(todo3)\n"
  935. "task.run_until_ms(1000)\n"
  936. "\n");
  937. /* collect */
  938. int mem_lack = obj_getInt(pikaMain, "mem_lack");
  939. /* assert */
  940. EXPECT_STREQ(log_buff[0], "task 1 running...\r\n");
  941. EXPECT_STREQ(log_buff[1], "task 1 running...\r\n");
  942. EXPECT_STREQ(log_buff[2], "task 2 running...\r\n");
  943. EXPECT_STREQ(log_buff[3], "task 1 running...\r\n");
  944. EXPECT_STREQ(log_buff[4], "task 1 running...\r\n");
  945. EXPECT_STREQ(log_buff[5], "task 2 running...\r\n");
  946. EXPECT_STREQ(log_buff[6], "task 1 running...\r\n");
  947. EXPECT_STREQ(log_buff[7], "BEGIN\r\n");
  948. EXPECT_EQ(mem_lack, 0);
  949. /* deinit */
  950. obj_deinit(pikaMain);
  951. EXPECT_EQ(pikaMemNow(), 0);
  952. }
  953. #endif
  954. TEST(pikaMain, fun_call) {
  955. /* init */
  956. pikaMemInfo.heapUsedMax = 0;
  957. /* run */
  958. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  959. __platform_printf("BEGIN\r\n");
  960. obj_run(pikaMain,
  961. "def fun(a,b,c,d,e):\n"
  962. " print(a)\n"
  963. " print(b)\n"
  964. " print(c)\n"
  965. " print(d)\n"
  966. " print(e)\n"
  967. "fun(10,20,30,40, 'xxx')\n"
  968. "\n");
  969. /* assert */
  970. EXPECT_STREQ(log_buff[0], "xxx\r\n");
  971. EXPECT_STREQ(log_buff[1], "40\r\n");
  972. EXPECT_STREQ(log_buff[2], "30\r\n");
  973. EXPECT_STREQ(log_buff[3], "20\r\n");
  974. EXPECT_STREQ(log_buff[4], "10\r\n");
  975. EXPECT_STREQ(log_buff[5], "BEGIN\r\n");
  976. /* deinit */
  977. obj_deinit(pikaMain);
  978. EXPECT_EQ(pikaMemNow(), 0);
  979. }
  980. TEST(pikaMain, global) {
  981. /* init */
  982. pikaMemInfo.heapUsedMax = 0;
  983. /* run */
  984. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  985. __platform_printf("BEGIN\r\n");
  986. obj_run(pikaMain,
  987. "x = 0\n"
  988. "a = 0\n"
  989. "z = 0\n"
  990. "def test_global():\n"
  991. " global x\n"
  992. " global y, z\n"
  993. " x = 1\n"
  994. " a = 1\n"
  995. " z = 1\n"
  996. " print('test global')\n"
  997. "test_global()\n"
  998. "\n");
  999. /* assert */
  1000. int x = obj_getInt(pikaMain, "x");
  1001. int a = obj_getInt(pikaMain, "a");
  1002. int z = obj_getInt(pikaMain, "z");
  1003. EXPECT_EQ(x, 1);
  1004. EXPECT_EQ(a, 0);
  1005. EXPECT_EQ(z, 1);
  1006. /* deinit */
  1007. obj_deinit(pikaMain);
  1008. EXPECT_EQ(pikaMemNow(), 0);
  1009. }
  1010. TEST(pikaMain, complex_str) {
  1011. /* init */
  1012. pikaMemInfo.heapUsedMax = 0;
  1013. /* run */
  1014. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1015. __platform_printf("BEGIN\r\n");
  1016. obj_run(pikaMain, "print('test,test')\n");
  1017. /* assert */
  1018. EXPECT_STREQ(log_buff[0], "test,test\r\n");
  1019. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1020. /* deinit */
  1021. obj_deinit(pikaMain);
  1022. EXPECT_EQ(pikaMemNow(), 0);
  1023. }
  1024. TEST(pikaMain, syntax_err_1) {
  1025. /* init */
  1026. pikaMemInfo.heapUsedMax = 0;
  1027. /* run */
  1028. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1029. __platform_printf("BEGIN\r\n");
  1030. obj_run(pikaMain, "print('testtest)\n");
  1031. /* assert */
  1032. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  1033. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1034. /* deinit */
  1035. obj_deinit(pikaMain);
  1036. EXPECT_EQ(pikaMemNow(), 0);
  1037. }
  1038. TEST(pikaMain, class_arg) {
  1039. /* init */
  1040. pikaMemInfo.heapUsedMax = 0;
  1041. /* run */
  1042. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1043. __platform_printf("BEGIN\r\n");
  1044. Args* buffs = New_strBuff();
  1045. char* pikaAsm = Parser_linesToAsm(buffs,
  1046. "class Test(TinyObj):\n"
  1047. " x = 1\n"
  1048. "\n"
  1049. "test = Test()\n"
  1050. "print(test.x)\n");
  1051. printf("%s", pikaAsm);
  1052. pikaVM_runAsm(pikaMain, pikaAsm);
  1053. /* assert */
  1054. EXPECT_STREQ(log_buff[0], "1\r\n");
  1055. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1056. /* deinit */
  1057. obj_deinit(pikaMain);
  1058. args_deinit(buffs);
  1059. EXPECT_EQ(pikaMemNow(), 0);
  1060. }
  1061. TEST(pikaMain, class_def) {
  1062. /* init */
  1063. pikaMemInfo.heapUsedMax = 0;
  1064. /* run */
  1065. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1066. __platform_printf("BEGIN\r\n");
  1067. obj_run(self,
  1068. "class Test():\n"
  1069. " x = 1\n"
  1070. " def hello(self, x):\n"
  1071. " return x + 2\n"
  1072. "test = Test()\n"
  1073. "x = test.hello(test.x)\n"
  1074. "print(x)\n");
  1075. /* collect */
  1076. PikaObj* test = obj_getObj(self, "test");
  1077. EXPECT_EQ(obj_refcntNow(test), 1);
  1078. /* assert */
  1079. EXPECT_STREQ(log_buff[0], "3\r\n");
  1080. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1081. /* deinit */
  1082. obj_deinit(self);
  1083. EXPECT_EQ(pikaMemNow(), 0);
  1084. }
  1085. TEST(pikaMain, class_def_print) {
  1086. /* init */
  1087. pikaMemInfo.heapUsedMax = 0;
  1088. /* run */
  1089. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1090. __platform_printf("BEGIN\r\n");
  1091. obj_run(self,
  1092. "class Test():\n"
  1093. " x = 1\n"
  1094. " def hi(self):\n"
  1095. " print('hi' + str(self.x))\n"
  1096. "test = Test()\n"
  1097. "test.hi()\n");
  1098. /* assert */
  1099. EXPECT_STREQ(log_buff[0], "hi1\r\n");
  1100. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1101. /* deinit */
  1102. obj_deinit(self);
  1103. EXPECT_EQ(pikaMemNow(), 0);
  1104. }
  1105. TEST(pikaMain, class_demo_1) {
  1106. /* init */
  1107. pikaMemInfo.heapUsedMax = 0;
  1108. /* run */
  1109. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1110. __platform_printf("BEGIN\r\n");
  1111. obj_run(self,
  1112. "class MyClass:\n"
  1113. " i = 12345\n"
  1114. " def f(self):\n"
  1115. " return 'hello world'\n"
  1116. " \n"
  1117. "x = MyClass()\n"
  1118. " \n"
  1119. "print(x.i)\n"
  1120. "print(x.f())\n");
  1121. /* assert */
  1122. EXPECT_STREQ(log_buff[0], "hello world\r\n");
  1123. EXPECT_STREQ(log_buff[1], "12345\r\n");
  1124. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1125. /* deinit */
  1126. obj_deinit(self);
  1127. EXPECT_EQ(pikaMemNow(), 0);
  1128. }
  1129. TEST(pikaMain, class_demo_2) {
  1130. /* init */
  1131. pikaMemInfo.heapUsedMax = 0;
  1132. /* run */
  1133. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1134. __platform_printf("BEGIN\r\n");
  1135. obj_run(self,
  1136. "class Complex:\n"
  1137. " def __init__(self):\n"
  1138. " self.r = 3.0\n"
  1139. " self.i = -4.5\n"
  1140. "x = Complex()\n"
  1141. "print(x.r)\n"
  1142. "print(x.i)\n");
  1143. /* assert */
  1144. EXPECT_STREQ(log_buff[0], "-4.500000\r\n");
  1145. EXPECT_STREQ(log_buff[1], "3.000000\r\n");
  1146. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1147. /* deinit */
  1148. obj_deinit(self);
  1149. EXPECT_EQ(pikaMemNow(), 0);
  1150. }
  1151. TEST(pikaMain, class_demo_3) {
  1152. /* init */
  1153. pikaMemInfo.heapUsedMax = 0;
  1154. /* run */
  1155. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1156. __platform_printf("BEGIN\r\n");
  1157. obj_run(self,
  1158. "class people:\n"
  1159. " def speak(self):\n"
  1160. " print('i am a people')\n"
  1161. " \n"
  1162. "class student(people):\n"
  1163. " def speak(self):\n"
  1164. " print('i am a student')\n"
  1165. " \n"
  1166. "p = people()\n"
  1167. "s = student()\n"
  1168. "p.speak()\n"
  1169. "s.speak()\n");
  1170. /* assert */
  1171. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1172. EXPECT_STREQ(log_buff[1], "i am a people\r\n");
  1173. EXPECT_STREQ(log_buff[0], "i am a student\r\n");
  1174. /* deinit */
  1175. obj_deinit(self);
  1176. EXPECT_EQ(pikaMemNow(), 0);
  1177. }
  1178. TEST(pikaMain, class_demo_save_asm) {
  1179. /* init */
  1180. pikaMemInfo.heapUsedMax = 0;
  1181. /* run */
  1182. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1183. __platform_printf("BEGIN\r\n");
  1184. obj_run(self,
  1185. "class people:\n"
  1186. " def speak(self):\n"
  1187. " print('i am a people')\n"
  1188. " \n"
  1189. "class student(people):\n"
  1190. " def speak(self):\n"
  1191. " print('i am a student')\n"
  1192. " \n");
  1193. obj_run(self,
  1194. "p = people()\n"
  1195. "s = student()\n"
  1196. "p.speak()\n"
  1197. "s.speak()\n");
  1198. /* assert */
  1199. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1200. EXPECT_STREQ(log_buff[1], "i am a people\r\n");
  1201. EXPECT_STREQ(log_buff[0], "i am a student\r\n");
  1202. /* deinit */
  1203. obj_deinit(self);
  1204. EXPECT_EQ(pikaMemNow(), 0);
  1205. }
  1206. TEST(pikaMain, for_if_continue_byte_code) {
  1207. /* init */
  1208. pikaMemInfo.heapUsedMax = 0;
  1209. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1210. /* run */
  1211. pikaVM_run(pikaMain,
  1212. "a = 0\n"
  1213. "for i in range(0, 10):\n"
  1214. " if i == 5:\n"
  1215. " continue\n"
  1216. " a = a + i\n"
  1217. "\n");
  1218. /* collect */
  1219. int a = obj_getInt(pikaMain, "a");
  1220. /* assert */
  1221. EXPECT_EQ(a, 40);
  1222. /* deinit */
  1223. obj_deinit(pikaMain);
  1224. /* mem check */
  1225. EXPECT_EQ(pikaMemNow(), 0);
  1226. }
  1227. TEST(pikaMain, print_in_def_byte_code) {
  1228. /* init */
  1229. pikaMemInfo.heapUsedMax = 0;
  1230. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1231. /* run */
  1232. __platform_printf("BEGIN\r\n");
  1233. pikaVM_run(pikaMain,
  1234. "def test_print():\n"
  1235. " print('test')\n"
  1236. "test_print()\n");
  1237. /* collect */
  1238. /* assert */
  1239. /* should only print once, so the second log (log_buff[1]) shuold be '\0' */
  1240. EXPECT_STREQ(log_buff[0], "test\r\n");
  1241. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1242. /* deinit */
  1243. obj_deinit(pikaMain);
  1244. /* mem check */
  1245. EXPECT_EQ(pikaMemNow(), 0);
  1246. }
  1247. TEST(pikaMain, class_demo_1_file) {
  1248. /* init */
  1249. pikaMemInfo.heapUsedMax = 0;
  1250. /* run */
  1251. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1252. __platform_printf("BEGIN\r\n");
  1253. char lines[] =
  1254. "class MyClass:\n"
  1255. " i = 12345\n"
  1256. " def f(self):\n"
  1257. " return 'hello world'\n"
  1258. " \n"
  1259. "x = MyClass()\n"
  1260. " \n"
  1261. "print(x.i)\n"
  1262. "print(x.f())\n";
  1263. pikaCompile("pika_bytecode.bin", lines);
  1264. char bytecodebuff[4096] = {0};
  1265. FILE* f = __platform_fopen("pika_bytecode.bin", "rb");
  1266. printf("size: %ld\n", __platform_fread(bytecodebuff, 1, 4096, f));
  1267. __platform_printf("BEGIN\r\n");
  1268. pikaVM_runByteCode(self, (uint8_t*)bytecodebuff);
  1269. /* assert */
  1270. EXPECT_STREQ(log_buff[0], "hello world\r\n");
  1271. EXPECT_STREQ(log_buff[1], "12345\r\n");
  1272. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1273. /* deinit */
  1274. obj_deinit(self);
  1275. __platform_fclose(f);
  1276. EXPECT_EQ(pikaMemNow(), 0);
  1277. }
  1278. // #if !PIKA_NANO_ENABLE
  1279. // TEST(pikaMain, get_native_method) {
  1280. // /* init */
  1281. // pikaMemInfo.heapUsedMax = 0;
  1282. // PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1283. // __platform_printf("BEGIN\r\n");
  1284. // /* do some thing */
  1285. // Args args = {0};
  1286. // args_setStr(&args, "val", "t");
  1287. // obj_runNativeMethod(pikaMain, "ord", &args);
  1288. // args_deinit_stack(&args);
  1289. // /* assert */
  1290. // EXPECT_STREQ(log_buff[0], "test\r\n");
  1291. // EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1292. // /* deinit */
  1293. // obj_deinit(pikaMain);
  1294. // EXPECT_EQ(pikaMemNow(), 0);
  1295. // }
  1296. // #endif
  1297. TEST(pikaMain, hex_list) {
  1298. /* init */
  1299. pikaMemInfo.heapUsedMax = 0;
  1300. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1301. /* run */
  1302. __platform_printf("BEGIN\r\n");
  1303. pikaVM_run(pikaMain,
  1304. "bytes = PikaStdData.List()\n"
  1305. "bytes.append(0x0F)\n"
  1306. "bytes.append(0xFF)\n"
  1307. "sum = 0\n"
  1308. "for byte in bytes:\n"
  1309. " sum = sum + byte\n"
  1310. "\n"
  1311. "print(sum)\n");
  1312. /* collect */
  1313. /* assert */
  1314. EXPECT_STREQ(log_buff[0], "270\r\n");
  1315. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1316. /* deinit */
  1317. obj_deinit(pikaMain);
  1318. /* mem check */
  1319. EXPECT_EQ(pikaMemNow(), 0);
  1320. }
  1321. TEST(pikaMain, bytearray) {
  1322. /* init */
  1323. pikaMemInfo.heapUsedMax = 0;
  1324. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1325. /* run */
  1326. __platform_printf("BEGIN\r\n");
  1327. pikaVM_run(pikaMain,
  1328. "bytes = PikaStdData.ByteArray(b'test')\n"
  1329. "sum = 0\n"
  1330. "for byte in bytes:\n"
  1331. " sum += byte\n"
  1332. "print(bytes)\n"
  1333. "\n");
  1334. /* collect */
  1335. int sum = obj_getInt(pikaMain, "sum");
  1336. /* assert */
  1337. EXPECT_EQ(sum, 448);
  1338. EXPECT_STREQ(log_buff[0], "bytearray(b'\\x74\\x65\\x73\\x74')\r\n");
  1339. /* deinit */
  1340. obj_deinit(pikaMain);
  1341. /* mem check */
  1342. EXPECT_EQ(pikaMemNow(), 0);
  1343. }
  1344. TEST(pikaMain, not_4_space) {
  1345. /* init */
  1346. pikaMemInfo.heapUsedMax = 0;
  1347. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1348. /* run */
  1349. __platform_printf("BEGIN\r\n");
  1350. pikaVM_run(pikaMain, " print('test')\n");
  1351. /* collect */
  1352. /* assert */
  1353. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  1354. EXPECT_STREQ(
  1355. log_buff[1],
  1356. "IndentationError: unexpected indent, only support 4 spaces\r\n");
  1357. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1358. /* deinit */
  1359. obj_deinit(pikaMain);
  1360. /* mem check */
  1361. EXPECT_EQ(pikaMemNow(), 0);
  1362. }
  1363. TEST(pikaMain, self_operator) {
  1364. /* init */
  1365. pikaMemInfo.heapUsedMax = 0;
  1366. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1367. /* run */
  1368. __platform_printf("BEGIN\r\n");
  1369. pikaVM_run(pikaMain,
  1370. "a = 1\n"
  1371. "b = 2\n"
  1372. "c = 3\n"
  1373. "d = 4\n"
  1374. "a += 2\n"
  1375. "b -= 1+1-3\n"
  1376. "c *= 4\n"
  1377. "d /= 2\n");
  1378. /* collect */
  1379. int a = obj_getInt(pikaMain, "a");
  1380. int b = obj_getInt(pikaMain, "b");
  1381. int c = obj_getInt(pikaMain, "c");
  1382. int d = obj_getInt(pikaMain, "d");
  1383. /* assert */
  1384. EXPECT_EQ(a, 3);
  1385. EXPECT_EQ(b, 3);
  1386. EXPECT_EQ(c, 12);
  1387. EXPECT_EQ(d, 2);
  1388. /* deinit */
  1389. obj_deinit(pikaMain);
  1390. /* mem check */
  1391. EXPECT_EQ(pikaMemNow(), 0);
  1392. }
  1393. TEST(pikaMain, class_demo_2_initwitharg) {
  1394. /* init */
  1395. pikaMemInfo.heapUsedMax = 0;
  1396. /* run */
  1397. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1398. __platform_printf("BEGIN\r\n");
  1399. obj_run(self,
  1400. "class Complex:\n"
  1401. " def __init__(self, r, i):\n"
  1402. " self.r = r\n"
  1403. " self.i = i\n"
  1404. "x = Complex(3.0, -4.5)\n"
  1405. "print(x.r)\n"
  1406. "print(x.i)\n");
  1407. /* assert */
  1408. EXPECT_STREQ(log_buff[0], "-4.500000\r\n");
  1409. EXPECT_STREQ(log_buff[1], "3.000000\r\n");
  1410. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1411. /* deinit */
  1412. obj_deinit(self);
  1413. EXPECT_EQ(pikaMemNow(), 0);
  1414. }
  1415. TEST(pikaMain, def_args_err) {
  1416. /* init */
  1417. pikaMemInfo.heapUsedMax = 0;
  1418. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1419. /* run */
  1420. obj_run(pikaMain, "print()\n");
  1421. // obj_run(pikaMain, "printNoEnd()\n");
  1422. /* collect */
  1423. /* assert */
  1424. /* deinit */
  1425. obj_deinit(pikaMain);
  1426. EXPECT_EQ(pikaMemNow(), 0);
  1427. }
  1428. // TEST(pikaMain, class_args_err) {
  1429. // /* init */
  1430. // pikaMemInfo.heapUsedMax = 0;
  1431. // PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1432. // /* run */
  1433. // obj_run(pikaMain, "mem = PikaStdLib.MemChecker(0)\n");
  1434. // /* collect */
  1435. // /* assert */
  1436. // EXPECT_STREQ(log_buff[4],
  1437. // "TypeError: PikaStdLib.MemChecker() takes no
  1438. // arguments\r\n");
  1439. // /* deinit */
  1440. // obj_deinit(pikaMain);
  1441. // EXPECT_EQ(pikaMemNow(), 0);
  1442. // }
  1443. TEST(pikaMain, int_) {
  1444. /* init */
  1445. pikaMemInfo.heapUsedMax = 0;
  1446. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1447. /* run */
  1448. obj_run(pikaMain, "a = int('3')\n");
  1449. /* collect */
  1450. int a = obj_getInt(pikaMain, "a");
  1451. /* assert */
  1452. EXPECT_EQ(a, 3);
  1453. /* deinit */
  1454. obj_deinit(pikaMain);
  1455. EXPECT_EQ(pikaMemNow(), 0);
  1456. }
  1457. TEST(pikaMain, len_) {
  1458. /* init */
  1459. pikaMemInfo.heapUsedMax = 0;
  1460. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1461. /* run */
  1462. obj_run(pikaMain, "a = len('331dd')\n");
  1463. /* collect */
  1464. int a = obj_getInt(pikaMain, "a");
  1465. /* assert */
  1466. EXPECT_EQ(a, 5);
  1467. /* deinit */
  1468. obj_deinit(pikaMain);
  1469. EXPECT_EQ(pikaMemNow(), 0);
  1470. }
  1471. TEST(pikaMain, def_in_shell) {
  1472. /* init */
  1473. pikaMemInfo.heapUsedMax = 0;
  1474. /* run */
  1475. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1476. /* skip the first obj_run */
  1477. __platform_printf("BEGIN\r\n");
  1478. /* as run in shell */
  1479. obj_run(pikaMain,
  1480. "def test():\n"
  1481. " print('test')\n"
  1482. "\n");
  1483. obj_run(pikaMain,
  1484. "def test2():\n"
  1485. " print('test2')\n"
  1486. "\n");
  1487. obj_run(pikaMain, "test()\n");
  1488. obj_run(pikaMain, "test2()\n");
  1489. /* collect */
  1490. /* assert */
  1491. EXPECT_STREQ("test2\r\n", log_buff[0]);
  1492. EXPECT_STREQ("test\r\n", log_buff[1]);
  1493. EXPECT_STREQ("BEGIN\r\n", log_buff[2]);
  1494. /* deinit */
  1495. obj_deinit(pikaMain);
  1496. EXPECT_EQ(pikaMemNow(), 0);
  1497. }
  1498. TEST(pikaMain, def_in_shell_override) {
  1499. /* init */
  1500. pikaMemInfo.heapUsedMax = 0;
  1501. /* run */
  1502. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1503. /* skip the first obj_run */
  1504. __platform_printf("BEGIN\r\n");
  1505. /* as run in shell */
  1506. obj_run(pikaMain,
  1507. "def test():\n"
  1508. " print('test')\n"
  1509. "\n");
  1510. obj_run(pikaMain, "test()\n");
  1511. obj_run(pikaMain,
  1512. "def test():\n"
  1513. " print('test2')\n"
  1514. "\n");
  1515. obj_run(pikaMain, "test()\n");
  1516. /* collect */
  1517. /* assert */
  1518. EXPECT_STREQ("test2\r\n", log_buff[0]);
  1519. EXPECT_STREQ("test\r\n", log_buff[1]);
  1520. EXPECT_STREQ("BEGIN\r\n", log_buff[2]);
  1521. /* deinit */
  1522. obj_deinit(pikaMain);
  1523. EXPECT_EQ(pikaMemNow(), 0);
  1524. }
  1525. TEST(pikaMain, class_in_shell) {
  1526. /* init */
  1527. pikaMemInfo.heapUsedMax = 0;
  1528. /* run */
  1529. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1530. /* skip the first obj_run */
  1531. __platform_printf("BEGIN\r\n");
  1532. /* as run in shell */
  1533. obj_run(pikaMain,
  1534. "class Test():\n"
  1535. " def hi(self):\n"
  1536. " print('hi')\n"
  1537. "\n");
  1538. obj_run(pikaMain,
  1539. "t = Test()\n"
  1540. "t.hi()\n");
  1541. /* collect */
  1542. /* assert */
  1543. EXPECT_STREQ("hi\r\n", log_buff[0]);
  1544. EXPECT_STREQ("BEGIN\r\n", log_buff[1]);
  1545. /* deinit */
  1546. obj_deinit(pikaMain);
  1547. EXPECT_EQ(pikaMemNow(), 0);
  1548. }
  1549. TEST(pikaMain, class_TinyObj) {
  1550. /* init */
  1551. pikaMemInfo.heapUsedMax = 0;
  1552. /* run */
  1553. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1554. /* skip the first obj_run */
  1555. __platform_printf("BEGIN\r\n");
  1556. /* as run in shell */
  1557. obj_run(pikaMain,
  1558. "class MyClass( TinyObj ):\n"
  1559. " def __init__(self):\n"
  1560. " self.val = 123\n"
  1561. "\n");
  1562. obj_run(pikaMain,
  1563. "x = MyClass()\n"
  1564. "x.val\n");
  1565. /* collect */
  1566. /* assert */
  1567. EXPECT_STREQ("123\r\n", log_buff[0]);
  1568. EXPECT_STREQ("BEGIN\r\n", log_buff[1]);
  1569. /* deinit */
  1570. obj_deinit(pikaMain);
  1571. EXPECT_EQ(pikaMemNow(), 0);
  1572. }
  1573. TEST(pikaMain, list_sysobj) {
  1574. /* init */
  1575. pikaMemInfo.heapUsedMax = 0;
  1576. /* run */
  1577. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1578. /* skip the first obj_run */
  1579. obj_run(pikaMain,
  1580. "a = list()\n"
  1581. "a.len()\n");
  1582. /* as run in shell */
  1583. /* collect */
  1584. /* assert */
  1585. /* deinit */
  1586. obj_deinit(pikaMain);
  1587. EXPECT_EQ(pikaMemNow(), 0);
  1588. }
  1589. #if PIKA_BUILTIN_STRUCT_ENABLE
  1590. TEST(pikaMain, list_init) {
  1591. /* init */
  1592. pikaMemInfo.heapUsedMax = 0;
  1593. /* run */
  1594. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1595. __platform_printf("BEGIN\r\n");
  1596. obj_run(pikaMain,
  1597. "a = [1, 2, 3]\n"
  1598. "a.len()\n");
  1599. /* as run in shell */
  1600. /* collect */
  1601. /* assert */
  1602. EXPECT_STREQ("3\r\n", log_buff[0]);
  1603. EXPECT_STREQ("BEGIN\r\n", log_buff[1]);
  1604. /* deinit */
  1605. obj_deinit(pikaMain);
  1606. EXPECT_EQ(pikaMemNow(), 0);
  1607. }
  1608. #endif
  1609. TEST(pikaMain, ctypes) {
  1610. /* init */
  1611. pikaMemInfo.heapUsedMax = 0;
  1612. /* run */
  1613. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1614. char* lines =
  1615. "t = ctypes.Test()\n"
  1616. "sendbuf = b'\\x03\\x04\\x33\\x00\\x05'\n"
  1617. "rlen = ctypes.c_uint(0)\n"
  1618. "rcvbuf = ctypes.c_char_p('')\n"
  1619. "res = t.dc_cpuapdu_hex(5, sendbuf, rlen, rcvbuf)\n"
  1620. "t.print_rcv(rcvbuf)\n";
  1621. __platform_printf("%s\n", lines);
  1622. __platform_printf("BEGIN\r\n");
  1623. obj_run(pikaMain, lines);
  1624. /* as run in shell */
  1625. /* collect */
  1626. Arg* sendbuf = obj_getArg(pikaMain, "sendbuf");
  1627. Arg* rcvbuf = obj_getArg(pikaMain, "rcvbuf.value");
  1628. /* assert */
  1629. EXPECT_EQ(arg_getBytesSize(sendbuf), 5);
  1630. EXPECT_EQ(arg_getBytesSize(rcvbuf), 6);
  1631. /* deinit */
  1632. obj_deinit(pikaMain);
  1633. EXPECT_EQ(pikaMemNow(), 0);
  1634. }
  1635. TEST(pikaMain, import_err) {
  1636. /* init */
  1637. pikaMemInfo.heapUsedMax = 0;
  1638. /* run */
  1639. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  1640. char* lines = "import qqpe\n";
  1641. __platform_printf("%s\n", lines);
  1642. __platform_printf("BEGIN\r\n");
  1643. obj_run(pikaMain, lines);
  1644. /* as run in shell */
  1645. /* collect */
  1646. /* assert */
  1647. EXPECT_STREQ(log_buff[2],
  1648. "ModuleNotFoundError: No module named 'qqpe'\r\n");
  1649. /* deinit */
  1650. obj_deinit(pikaMain);
  1651. EXPECT_EQ(pikaMemNow(), 0);
  1652. }
  1653. TEST(pikaMain, cmodule_in_py) {
  1654. /* init */
  1655. pikaMemInfo.heapUsedMax = 0;
  1656. /* run */
  1657. PikaObj* pikaMain = pikaScriptInit();
  1658. char* lines =
  1659. "import test_cmodule\n"
  1660. "test_cmodule.test_mem()\n";
  1661. __platform_printf("%s\n", lines);
  1662. __platform_printf("BEGIN\r\n");
  1663. obj_run(pikaMain, lines);
  1664. /* as run in shell */
  1665. /* collect */
  1666. /* assert */
  1667. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1668. EXPECT_STREQ(log_buff[1], "mem used max:\r\n");
  1669. /* deinit */
  1670. obj_deinit(pikaMain);
  1671. EXPECT_EQ(pikaMemNow(), 0);
  1672. }
  1673. TEST(pikaMain, class___str___print) {
  1674. /* init */
  1675. pikaMemInfo.heapUsedMax = 0;
  1676. /* run */
  1677. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1678. __platform_printf("BEGIN\r\n");
  1679. obj_run(self,
  1680. "class Test():\n"
  1681. " def __str__(self):\n"
  1682. " return 'test'\n"
  1683. "test = Test()\n"
  1684. "print(test)\n");
  1685. /* assert */
  1686. EXPECT_STREQ(log_buff[0], "test\r\n");
  1687. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1688. /* deinit */
  1689. obj_deinit(self);
  1690. EXPECT_EQ(pikaMemNow(), 0);
  1691. }
  1692. TEST(pikaMain, CModule__str__) {
  1693. /* init */
  1694. pikaMemInfo.heapUsedMax = 0;
  1695. /* run */
  1696. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1697. __platform_printf("BEGIN\r\n");
  1698. obj_run(self,
  1699. "op = PikaMath.Operator()\n"
  1700. "print(op)\n"
  1701. "print('obj: ' + str(op))\n");
  1702. /* assert */
  1703. EXPECT_STREQ(log_buff[0], "obj: test\r\n");
  1704. EXPECT_STREQ(log_buff[1], "test\r\n");
  1705. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1706. /* deinit */
  1707. obj_deinit(self);
  1708. EXPECT_STREQ(log_buff[0], "del operator...\r\n");
  1709. EXPECT_EQ(pikaMemNow(), 0);
  1710. }
  1711. #if !PIKA_NANO_ENABLE
  1712. TEST(pikaMain, builtin_hex) {
  1713. /* init */
  1714. pikaMemInfo.heapUsedMax = 0;
  1715. /* run */
  1716. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1717. __platform_printf("BEGIN\r\n");
  1718. obj_run(self,
  1719. "hex(3)\n"
  1720. "hex(-3)\n"
  1721. "hex(1234)\n"
  1722. "hex(-12039)\n");
  1723. /* assert */
  1724. EXPECT_STREQ(log_buff[4], "BEGIN\r\n");
  1725. EXPECT_STREQ(log_buff[3], "'0x03'\r\n");
  1726. EXPECT_STREQ(log_buff[2], "'-0x03'\r\n");
  1727. EXPECT_STREQ(log_buff[1], "'0x4d2'\r\n");
  1728. EXPECT_STREQ(log_buff[0], "'-0x2f07'\r\n");
  1729. /* deinit */
  1730. obj_deinit(self);
  1731. EXPECT_EQ(pikaMemNow(), 0);
  1732. }
  1733. #endif
  1734. #if !PIKA_NANO_ENABLE
  1735. TEST(pikaMain, builtin_ord) {
  1736. /* init */
  1737. pikaMemInfo.heapUsedMax = 0;
  1738. /* run */
  1739. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1740. __platform_printf("BEGIN\r\n");
  1741. obj_run(self, "ord('a')\n");
  1742. /* assert */
  1743. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1744. EXPECT_STREQ(log_buff[0], "97\r\n");
  1745. /* deinit */
  1746. obj_deinit(self);
  1747. EXPECT_EQ(pikaMemNow(), 0);
  1748. }
  1749. #endif
  1750. #if !PIKA_NANO_ENABLE
  1751. TEST(pikaMain, builtin_chr) {
  1752. /* init */
  1753. pikaMemInfo.heapUsedMax = 0;
  1754. /* run */
  1755. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1756. __platform_printf("BEGIN\r\n");
  1757. obj_run(self, "chr(97)\n");
  1758. /* assert */
  1759. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1760. EXPECT_STREQ(log_buff[0], "'a'\r\n");
  1761. /* deinit */
  1762. obj_deinit(self);
  1763. EXPECT_EQ(pikaMemNow(), 0);
  1764. }
  1765. #endif
  1766. TEST(pikaMain, iteral_oct) {
  1767. /* init */
  1768. pikaMemInfo.heapUsedMax = 0;
  1769. /* run */
  1770. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1771. __platform_printf("BEGIN\r\n");
  1772. obj_run(self, "0o100\n");
  1773. /* assert */
  1774. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1775. EXPECT_STREQ(log_buff[0], "64\r\n");
  1776. /* deinit */
  1777. obj_deinit(self);
  1778. EXPECT_EQ(pikaMemNow(), 0);
  1779. }
  1780. TEST(pikaMain, iteral_bin) {
  1781. /* init */
  1782. pikaMemInfo.heapUsedMax = 0;
  1783. /* run */
  1784. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1785. __platform_printf("BEGIN\r\n");
  1786. obj_run(self, "0b100\n");
  1787. /* assert */
  1788. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1789. EXPECT_STREQ(log_buff[0], "4\r\n");
  1790. /* deinit */
  1791. obj_deinit(self);
  1792. EXPECT_EQ(pikaMemNow(), 0);
  1793. }
  1794. TEST(pikaMain, REPL_push_mode) {
  1795. /* init */
  1796. pikaMemInfo.heapUsedMax = 0;
  1797. /* run */
  1798. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1799. __platform_printf("BEGIN\r\n");
  1800. char lines[] =
  1801. "print('test')\n"
  1802. "for i in range(0, 10):\n"
  1803. " i\n"
  1804. "\n";
  1805. for (size_t i = 0; i < strGetSize(lines); i++) {
  1806. obj_runChar(self, lines[i]);
  1807. }
  1808. /* assert */
  1809. // EXPECT_STREQ(log_buff[18], "BEGIN\r\n");
  1810. EXPECT_STREQ(log_buff[14], "test\r\n");
  1811. EXPECT_STREQ(log_buff[11], "... ");
  1812. EXPECT_STREQ(log_buff[1], "9\r\n");
  1813. EXPECT_STREQ(log_buff[0], ">>> ");
  1814. /* deinit */
  1815. obj_deinit(self);
  1816. EXPECT_EQ(pikaMemNow(), 0);
  1817. }
  1818. TEST(pikaMain, REPL_push_mode_err) {
  1819. /* init */
  1820. pikaMemInfo.heapUsedMax = 0;
  1821. /* run */
  1822. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1823. __platform_printf("BEGIN\r\n");
  1824. char lines[] =
  1825. "def test:\n"
  1826. " print('test')\n"
  1827. "\n";
  1828. for (size_t i = 0; i < strGetSize(lines); i++) {
  1829. obj_runChar(self, lines[i]);
  1830. }
  1831. /* assert */
  1832. EXPECT_STREQ(log_buff[1], "Error: Syntax error.\r\n");
  1833. /* deinit */
  1834. obj_deinit(self);
  1835. EXPECT_EQ(pikaMemNow(), 0);
  1836. }
  1837. #if 0
  1838. TEST(pikaMain, REPL_input) {
  1839. /* init */
  1840. pikaMemInfo.heapUsedMax = 0;
  1841. /* run */
  1842. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1843. __platform_printf("BEGIN\r\n");
  1844. char lines[] =
  1845. "res = input('input:\\n')\n"
  1846. "1234\n"
  1847. "print(res)\n";
  1848. for (size_t i = 0; i < strGetSize(lines); i++) {
  1849. obj_runChar(self, lines[i]);
  1850. }
  1851. /* assert */
  1852. /* deinit */
  1853. obj_deinit(self);
  1854. EXPECT_EQ(pikaMemNow(), 0);
  1855. }
  1856. #endif
  1857. #if PIKA_SYNTAX_SLICE_ENABLE
  1858. TEST(pikaMain, int_from_bytes) {
  1859. /* init */
  1860. pikaMemInfo.heapUsedMax = 0;
  1861. /* run */
  1862. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1863. __platform_printf("BEGIN\r\n");
  1864. obj_run(self, "int(b'test'[0])\n");
  1865. /* assert */
  1866. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1867. EXPECT_STREQ(log_buff[0], "116\r\n");
  1868. /* deinit */
  1869. obj_deinit(self);
  1870. EXPECT_EQ(pikaMemNow(), 0);
  1871. }
  1872. #endif
  1873. // TEST(pikaMain, big_bytes) {
  1874. // /* init */
  1875. // pikaMemInfo.heapUsedMax = 0;
  1876. // /* run */
  1877. // PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1878. // __platform_printf("BEGIN\r\n");
  1879. // obj_run(self, "b = bytes(8192)\n");
  1880. // /* assert */
  1881. // /* deinit */
  1882. // obj_deinit(self);
  1883. // EXPECT_EQ(pikaMemNow(), 0);
  1884. // }
  1885. #if PIKA_SYNTAX_SLICE_ENABLE
  1886. TEST(pikaMain, utils_int_to_bytes) {
  1887. /* init */
  1888. pikaMemInfo.heapUsedMax = 0;
  1889. /* run */
  1890. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1891. __platform_printf("BEGIN\r\n");
  1892. obj_run(self,
  1893. "ut = PikaStdData.Utils()\n"
  1894. "b = b'test'\n"
  1895. "b[0] = ut.int_to_bytes(0x35)\n"
  1896. "b\n");
  1897. /* assert */
  1898. EXPECT_STREQ(log_buff[0], "b'\\x35\\x65\\x73\\x74'\r\n");
  1899. /* deinit */
  1900. obj_deinit(self);
  1901. EXPECT_EQ(pikaMemNow(), 0);
  1902. }
  1903. #endif
  1904. #if PIKA_SYNTAX_SLICE_ENABLE
  1905. TEST(pikaMain, neg_index) {
  1906. /* init */
  1907. pikaMemInfo.heapUsedMax = 0;
  1908. /* run */
  1909. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1910. __platform_printf("BEGIN\r\n");
  1911. obj_run(self,
  1912. "'test'[-2]\n"
  1913. "b'test'[-2]\n");
  1914. /* assert */
  1915. EXPECT_STREQ(log_buff[0], "b'\\x73'\r\n");
  1916. EXPECT_STREQ(log_buff[1], "'s'\r\n");
  1917. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  1918. /* deinit */
  1919. obj_deinit(self);
  1920. EXPECT_EQ(pikaMemNow(), 0);
  1921. }
  1922. #endif
  1923. #if PIKA_SYNTAX_SLICE_ENABLE
  1924. TEST(pikaMain, slice1) {
  1925. /* init */
  1926. pikaMemInfo.heapUsedMax = 0;
  1927. /* run */
  1928. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1929. __platform_printf("BEGIN\r\n");
  1930. obj_run(self, "'test'[1:3]\n");
  1931. /* assert */
  1932. EXPECT_STREQ(log_buff[0], "'es'\r\n");
  1933. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1934. /* deinit */
  1935. obj_deinit(self);
  1936. EXPECT_EQ(pikaMemNow(), 0);
  1937. }
  1938. #endif
  1939. #if PIKA_SYNTAX_SLICE_ENABLE
  1940. TEST(pikaMain, slice2) {
  1941. /* init */
  1942. pikaMemInfo.heapUsedMax = 0;
  1943. /* run */
  1944. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1945. __platform_printf("BEGIN\r\n");
  1946. obj_run(self, "b'test'[1:3]\n");
  1947. /* assert */
  1948. EXPECT_STREQ(log_buff[0], "b'\\x65\\x73'\r\n");
  1949. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1950. /* deinit */
  1951. obj_deinit(self);
  1952. EXPECT_EQ(pikaMemNow(), 0);
  1953. }
  1954. #endif
  1955. TEST(pikaMain, c_buffer) {
  1956. /* init */
  1957. pikaMemInfo.heapUsedMax = 0;
  1958. /* run */
  1959. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1960. __platform_printf("BEGIN\r\n");
  1961. obj_run(self, "version = ctypes.c_buffer(b\"\", 128)\n");
  1962. /* assert */
  1963. /* deinit */
  1964. obj_deinit(self);
  1965. EXPECT_EQ(pikaMemNow(), 0);
  1966. }
  1967. TEST(pikaMain, str_add1) {
  1968. /* init */
  1969. pikaMemInfo.heapUsedMax = 0;
  1970. /* run */
  1971. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1972. __platform_printf("BEGIN\r\n");
  1973. obj_run(self,
  1974. "i = 32\n"
  1975. "msg = \"device_names[\" + str(i) + \"]:\"\n"
  1976. "msg\n");
  1977. /* assert */
  1978. EXPECT_STREQ(log_buff[0], "'device_names[32]:'\r\n");
  1979. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1980. /* deinit */
  1981. obj_deinit(self);
  1982. EXPECT_EQ(pikaMemNow(), 0);
  1983. }
  1984. #if PIKA_SYNTAX_SLICE_ENABLE
  1985. TEST(pikaMain, slice_eei) {
  1986. /* init */
  1987. pikaMemInfo.heapUsedMax = 0;
  1988. /* run */
  1989. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  1990. __platform_printf("BEGIN\r\n");
  1991. obj_run(self, "'test'[1:-1]\n");
  1992. /* assert */
  1993. EXPECT_STREQ(log_buff[0], "'es'\r\n");
  1994. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  1995. /* deinit */
  1996. obj_deinit(self);
  1997. EXPECT_EQ(pikaMemNow(), 0);
  1998. }
  1999. #endif
  2000. #if PIKA_SYNTAX_SLICE_ENABLE
  2001. TEST(pikaMain, slice_a97) {
  2002. /* init */
  2003. pikaMemInfo.heapUsedMax = 0;
  2004. /* run */
  2005. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2006. __platform_printf("BEGIN\r\n");
  2007. obj_run(self, "'test'[:-2]\n");
  2008. /* assert */
  2009. EXPECT_STREQ(log_buff[0], "'te'\r\n");
  2010. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2011. /* deinit */
  2012. obj_deinit(self);
  2013. EXPECT_EQ(pikaMemNow(), 0);
  2014. }
  2015. #endif
  2016. #if PIKA_SYNTAX_SLICE_ENABLE
  2017. TEST(pikaMain, slice_a90) {
  2018. /* init */
  2019. pikaMemInfo.heapUsedMax = 0;
  2020. /* run */
  2021. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2022. __platform_printf("BEGIN\r\n");
  2023. obj_run(self, "'test'[1:]\n");
  2024. /* assert */
  2025. EXPECT_STREQ(log_buff[0], "'est'\r\n");
  2026. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2027. /* deinit */
  2028. obj_deinit(self);
  2029. EXPECT_EQ(pikaMemNow(), 0);
  2030. }
  2031. #endif
  2032. #if PIKA_SYNTAX_SLICE_ENABLE
  2033. TEST(pikaMain, slice_a12) {
  2034. /* init */
  2035. pikaMemInfo.heapUsedMax = 0;
  2036. /* run */
  2037. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2038. __platform_printf("BEGIN\r\n");
  2039. obj_run(self, "b'test'[1:]\n");
  2040. /* assert */
  2041. EXPECT_STREQ(log_buff[0], "b'\\x65\\x73\\x74'\r\n");
  2042. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2043. /* deinit */
  2044. obj_deinit(self);
  2045. EXPECT_EQ(pikaMemNow(), 0);
  2046. }
  2047. #endif
  2048. TEST(pikaMain, str_string) {
  2049. /* init */
  2050. pikaMemInfo.heapUsedMax = 0;
  2051. /* run */
  2052. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2053. __platform_printf("BEGIN\r\n");
  2054. obj_run(self,
  2055. "a = PikaStdData.String('test')\n"
  2056. "print(a)\n");
  2057. /* assert */
  2058. EXPECT_STREQ(log_buff[0], "test\r\n");
  2059. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2060. /* deinit */
  2061. obj_deinit(self);
  2062. EXPECT_EQ(pikaMemNow(), 0);
  2063. }
  2064. #if PIKA_SYNTAX_SLICE_ENABLE
  2065. TEST(pikaMain, string_index) {
  2066. /* init */
  2067. pikaMemInfo.heapUsedMax = 0;
  2068. /* run */
  2069. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2070. __platform_printf("BEGIN\r\n");
  2071. obj_run(self,
  2072. "a = PikaStdData.String('test')\n"
  2073. "a[1]\n"
  2074. "a[1] = 'q'\n"
  2075. "print(a)\n");
  2076. /* assert */
  2077. EXPECT_STREQ(log_buff[0], "tqst\r\n");
  2078. EXPECT_STREQ(log_buff[1], "'e'\r\n");
  2079. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  2080. /* deinit */
  2081. obj_deinit(self);
  2082. EXPECT_EQ(pikaMemNow(), 0);
  2083. }
  2084. #endif
  2085. TEST(pikaMain, list_for_append) {
  2086. /* init */
  2087. pikaMemInfo.heapUsedMax = 0;
  2088. /* run */
  2089. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2090. __platform_printf("BEGIN\r\n");
  2091. obj_run(self,
  2092. "rcv_buf = PikaStdData.List()\n"
  2093. "for i in range(0, 1024):\n"
  2094. " rcv_buf.append(0)\n"
  2095. "rcv_buf.len()\n");
  2096. /* assert */
  2097. EXPECT_STREQ(log_buff[0], "1024\r\n");
  2098. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2099. /* deinit */
  2100. obj_deinit(self);
  2101. EXPECT_EQ(pikaMemNow(), 0);
  2102. }
  2103. TEST(pikaMain, string_str) {
  2104. /* init */
  2105. pikaMemInfo.heapUsedMax = 0;
  2106. /* run */
  2107. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2108. __platform_printf("BEGIN\r\n");
  2109. obj_run(self, "str(PikaStdData.String('test'))\n");
  2110. /* assert */
  2111. EXPECT_STREQ(log_buff[0], "'test'\r\n");
  2112. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2113. /* deinit */
  2114. obj_deinit(self);
  2115. EXPECT_EQ(pikaMemNow(), 0);
  2116. }
  2117. TEST(pikaMain, module_import_from_module) {
  2118. pikaMemInfo.heapUsedMax = 0;
  2119. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2120. extern unsigned char pikaModules_py_a[];
  2121. obj_linkLibrary(self, pikaModules_py_a);
  2122. __platform_printf("BEGIN\r\n");
  2123. obj_run(self,
  2124. "import test_module1\n"
  2125. "test_module1.test_module_import()\n");
  2126. EXPECT_STREQ(log_buff[0], "test_module_2_hello\r\n");
  2127. EXPECT_STREQ(log_buff[1], "in test module 2\r\n");
  2128. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  2129. obj_deinit(self);
  2130. EXPECT_EQ(pikaMemNow(), 0);
  2131. }
  2132. TEST(pikaMain, module_1_module2_test) {
  2133. pikaMemInfo.heapUsedMax = 0;
  2134. PikaObj* self = newRootObj("pikaMain", New_PikaMain);
  2135. extern unsigned char pikaModules_py_a[];
  2136. obj_linkLibrary(self, pikaModules_py_a);
  2137. __platform_printf("BEGIN\r\n");
  2138. obj_run(self,
  2139. "import test_module1\n"
  2140. "test_module1.test_module2.mytest()\n");
  2141. EXPECT_STREQ(log_buff[0], "test_module_2_hello\r\n");
  2142. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2143. obj_deinit(self);
  2144. EXPECT_EQ(pikaMemNow(), 0);
  2145. }
  2146. TEST(pikaMain, string_startswith) {
  2147. /* init */
  2148. pikaMemInfo.heapUsedMax = 0;
  2149. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2150. /* run */
  2151. obj_run(pikaMain,
  2152. "a = PikaStdData.String('test')\n"
  2153. "res1 = a.startswith('te')\n"
  2154. "res2 = a.startswith('st')\n");
  2155. /* collect */
  2156. int res1 = obj_getInt(pikaMain, "res1");
  2157. int res2 = obj_getInt(pikaMain, "res2");
  2158. /* assert */
  2159. EXPECT_EQ(res1, 1);
  2160. EXPECT_EQ(res2, 0);
  2161. /* deinit */
  2162. obj_deinit(pikaMain);
  2163. EXPECT_EQ(pikaMemNow(), 0);
  2164. }
  2165. TEST(pikaMain, string_endswith) {
  2166. /* init */
  2167. pikaMemInfo.heapUsedMax = 0;
  2168. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2169. /* run */
  2170. obj_run(pikaMain,
  2171. "a = PikaStdData.String('test')\n"
  2172. "res1 = a.endswith('te')\n"
  2173. "res2 = a.endswith('st')\n");
  2174. /* collect */
  2175. int res1 = obj_getInt(pikaMain, "res1");
  2176. int res2 = obj_getInt(pikaMain, "res2");
  2177. /* assert */
  2178. EXPECT_EQ(res1, 0);
  2179. EXPECT_EQ(res2, 1);
  2180. /* deinit */
  2181. obj_deinit(pikaMain);
  2182. EXPECT_EQ(pikaMemNow(), 0);
  2183. }
  2184. TEST(pikaMain, string_isdigit) {
  2185. /* init */
  2186. pikaMemInfo.heapUsedMax = 0;
  2187. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2188. /* run */
  2189. obj_run(pikaMain,
  2190. "a = PikaStdData.String('test')\n"
  2191. "b = PikaStdData.String('1234')\n"
  2192. "c = PikaStdData.String('test1234')\n"
  2193. "res1 = a.isdigit()\n"
  2194. "res2 = b.isdigit()\n"
  2195. "res3 = c.isdigit()\n");
  2196. /* collect */
  2197. int res1 = obj_getInt(pikaMain, "res1");
  2198. int res2 = obj_getInt(pikaMain, "res2");
  2199. int res3 = obj_getInt(pikaMain, "res3");
  2200. /* assert */
  2201. EXPECT_EQ(res1, 0);
  2202. EXPECT_EQ(res2, 1);
  2203. EXPECT_EQ(res3, 0);
  2204. /* deinit */
  2205. obj_deinit(pikaMain);
  2206. EXPECT_EQ(pikaMemNow(), 0);
  2207. }
  2208. TEST(pikaMain, string_islower) {
  2209. /* init */
  2210. pikaMemInfo.heapUsedMax = 0;
  2211. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2212. /* run */
  2213. obj_run(pikaMain,
  2214. "a = PikaStdData.String('test')\n"
  2215. "b = PikaStdData.String('1234')\n"
  2216. "c = PikaStdData.String('Test')\n"
  2217. "res1 = a.islower()\n"
  2218. "res2 = b.islower()\n"
  2219. "res3 = c.islower()\n");
  2220. /* collect */
  2221. int res1 = obj_getInt(pikaMain, "res1");
  2222. int res2 = obj_getInt(pikaMain, "res2");
  2223. int res3 = obj_getInt(pikaMain, "res3");
  2224. /* assert */
  2225. EXPECT_EQ(res1, 1);
  2226. EXPECT_EQ(res2, 0);
  2227. EXPECT_EQ(res3, 0);
  2228. /* deinit */
  2229. obj_deinit(pikaMain);
  2230. EXPECT_EQ(pikaMemNow(), 0);
  2231. }
  2232. TEST(pikaMain, string_isalnum) {
  2233. /* init */
  2234. pikaMemInfo.heapUsedMax = 0;
  2235. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2236. /* run */
  2237. obj_run(pikaMain,
  2238. "a = PikaStdData.String('test1234')\n"
  2239. "b = PikaStdData.String('1234')\n"
  2240. "c = PikaStdData.String(' ')\n"
  2241. "res1 = a.isalnum()\n"
  2242. "res2 = b.isalnum()\n"
  2243. "res3 = c.isalnum()\n");
  2244. /* collect */
  2245. int res1 = obj_getInt(pikaMain, "res1");
  2246. int res2 = obj_getInt(pikaMain, "res2");
  2247. int res3 = obj_getInt(pikaMain, "res3");
  2248. /* assert */
  2249. EXPECT_EQ(res1, 1);
  2250. EXPECT_EQ(res2, 1);
  2251. EXPECT_EQ(res3, 0);
  2252. /* deinit */
  2253. obj_deinit(pikaMain);
  2254. EXPECT_EQ(pikaMemNow(), 0);
  2255. }
  2256. TEST(pikaMain, string_isalpha) {
  2257. /* init */
  2258. pikaMemInfo.heapUsedMax = 0;
  2259. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2260. /* run */
  2261. obj_run(pikaMain,
  2262. "a = PikaStdData.String('test')\n"
  2263. "b = PikaStdData.String('1234')\n"
  2264. "c = PikaStdData.String(' ')\n"
  2265. "res1 = a.isalpha()\n"
  2266. "res2 = b.isalpha()\n"
  2267. "res3 = c.isalpha()\n");
  2268. /* collect */
  2269. int res1 = obj_getInt(pikaMain, "res1");
  2270. int res2 = obj_getInt(pikaMain, "res2");
  2271. int res3 = obj_getInt(pikaMain, "res3");
  2272. /* assert */
  2273. EXPECT_EQ(res1, 1);
  2274. EXPECT_EQ(res2, 0);
  2275. EXPECT_EQ(res3, 0);
  2276. /* deinit */
  2277. obj_deinit(pikaMain);
  2278. EXPECT_EQ(pikaMemNow(), 0);
  2279. }
  2280. TEST(pikaMain, string_isspace) {
  2281. /* init */
  2282. pikaMemInfo.heapUsedMax = 0;
  2283. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2284. /* run */
  2285. obj_run(pikaMain,
  2286. "a = PikaStdData.String('test')\n"
  2287. "b = PikaStdData.String(' ')\n"
  2288. "res1 = a.isspace()\n"
  2289. "res2 = b.isspace()\n");
  2290. /* collect */
  2291. int res1 = obj_getInt(pikaMain, "res1");
  2292. int res2 = obj_getInt(pikaMain, "res2");
  2293. /* assert */
  2294. EXPECT_EQ(res1, 0);
  2295. EXPECT_EQ(res2, 1);
  2296. /* deinit */
  2297. obj_deinit(pikaMain);
  2298. EXPECT_EQ(pikaMemNow(), 0);
  2299. }
  2300. TEST(pikaMain, print_obj) {
  2301. /* init */
  2302. pikaMemInfo.heapUsedMax = 0;
  2303. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2304. /* run */
  2305. __platform_printf("BEGIN\r\n");
  2306. obj_run(pikaMain,
  2307. "mem = PikaStdLib.MemChecker()\n"
  2308. "mem\n"
  2309. "print(mem)\n"
  2310. "res = str(mem)\n");
  2311. /* collect */
  2312. char* res = obj_getStr(pikaMain, "res");
  2313. /* assert */
  2314. EXPECT_EQ(strIsStartWith(log_buff[0], "<object at "), true);
  2315. EXPECT_EQ(strIsStartWith(log_buff[1], "<object at "), true);
  2316. EXPECT_STREQ(log_buff[2], "BEGIN\r\n");
  2317. EXPECT_EQ(strIsStartWith(res, "<object at "), true);
  2318. /* deinit */
  2319. obj_deinit(pikaMain);
  2320. EXPECT_EQ(pikaMemNow(), 0);
  2321. }
  2322. TEST(pikaMain, num_issue1) {
  2323. /* init */
  2324. pikaMemInfo.heapUsedMax = 0;
  2325. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2326. /* run */
  2327. __platform_printf("BEGIN\r\n");
  2328. char* lines = "(((1 + (2 * 3)/(4 + 5))*(6 - 7) + (8 + 9) * 10)/11) - 12\n";
  2329. obj_run(pikaMain, lines);
  2330. /* collect */
  2331. /* assert */
  2332. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2333. EXPECT_STREQ(log_buff[0], "3.303030\r\n");
  2334. /* deinit */
  2335. obj_deinit(pikaMain);
  2336. EXPECT_EQ(pikaMemNow(), 0);
  2337. }
  2338. TEST(pikaMain, returnNullString) {
  2339. /* init */
  2340. pikaMemInfo.heapUsedMax = 0;
  2341. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2342. /* run */
  2343. __platform_printf("BEGIN\r\n");
  2344. char* lines =
  2345. "import GTestTask\n"
  2346. "task = GTestTask.Task()\n"
  2347. "def test_return_none():\n"
  2348. " s = task.returnNullString()\n"
  2349. " return s\n"
  2350. "s = test_return_none()\n"
  2351. "print(s)\n"
  2352. "if None == s:\n"
  2353. " print('s is None')\n";
  2354. obj_run(pikaMain, lines);
  2355. /* collect */
  2356. Arg* s = obj_getArg(pikaMain, "s");
  2357. /* assert */
  2358. EXPECT_EQ((uintptr_t)s, NULL);
  2359. /* deinit */
  2360. obj_deinit(pikaMain);
  2361. EXPECT_EQ(pikaMemNow(), 0);
  2362. }
  2363. TEST(pikaMain, for_loop_issue_1b2a3f1bdf) {
  2364. char* lines =
  2365. "res = 0\n"
  2366. "for i in range(0, 10):\n"
  2367. " if i < 3:\n"
  2368. " continue\n"
  2369. " if i > 7:\n"
  2370. " continue\n"
  2371. " for i in range(i, i+3):\n"
  2372. " res += i\n"
  2373. "\n";
  2374. Args* buffs = New_strBuff();
  2375. __platform_printf("%s\n", lines);
  2376. char* pikaAsm = Parser_linesToAsm(buffs, lines);
  2377. __platform_printf("%s", pikaAsm);
  2378. args_deinit(buffs);
  2379. EXPECT_EQ(pikaMemNow(), 0);
  2380. /* init */
  2381. pikaMemInfo.heapUsedMax = 0;
  2382. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2383. /* run */
  2384. __platform_printf("BEGIN\r\n");
  2385. obj_run(pikaMain, lines);
  2386. /* collect */
  2387. int res = obj_getInt(pikaMain, "res");
  2388. /* assert */
  2389. EXPECT_EQ(res, 90);
  2390. /* deinit */
  2391. obj_deinit(pikaMain);
  2392. EXPECT_EQ(pikaMemNow(), 0);
  2393. }
  2394. #define pika_floatYPE_DOUBLE 0
  2395. TEST(pikaMain, syantex_issue123lkjxi) {
  2396. char* lines = "if i < 3\n";
  2397. /* init */
  2398. pikaMemInfo.heapUsedMax = 0;
  2399. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2400. /* run */
  2401. __platform_printf("BEGIN\r\n");
  2402. obj_run(pikaMain, lines);
  2403. /* collect */
  2404. /* assert */
  2405. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  2406. /* deinit */
  2407. obj_deinit(pikaMain);
  2408. EXPECT_EQ(pikaMemNow(), 0);
  2409. }
  2410. TEST(pikaMain, syantex_issue_fae13) {
  2411. char* lines = "for i < 3\n";
  2412. /* init */
  2413. pikaMemInfo.heapUsedMax = 0;
  2414. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2415. /* run */
  2416. __platform_printf("BEGIN\r\n");
  2417. obj_run(pikaMain, lines);
  2418. /* collect */
  2419. /* assert */
  2420. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  2421. /* deinit */
  2422. obj_deinit(pikaMain);
  2423. EXPECT_EQ(pikaMemNow(), 0);
  2424. }
  2425. TEST(pikaMain, syantex_issue_1289341) {
  2426. char* lines = "class lkj\n";
  2427. /* init */
  2428. pikaMemInfo.heapUsedMax = 0;
  2429. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2430. /* run */
  2431. __platform_printf("BEGIN\r\n");
  2432. obj_run(pikaMain, lines);
  2433. /* collect */
  2434. /* assert */
  2435. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  2436. /* deinit */
  2437. obj_deinit(pikaMain);
  2438. EXPECT_EQ(pikaMemNow(), 0);
  2439. }
  2440. TEST(pikaMain, syantex_issue_183571) {
  2441. char* lines = "def test\n";
  2442. /* init */
  2443. pikaMemInfo.heapUsedMax = 0;
  2444. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2445. /* run */
  2446. __platform_printf("BEGIN\r\n");
  2447. obj_run(pikaMain, lines);
  2448. /* collect */
  2449. /* assert */
  2450. EXPECT_STREQ(log_buff[0], "Error: Syntax error.\r\n");
  2451. /* deinit */
  2452. obj_deinit(pikaMain);
  2453. EXPECT_EQ(pikaMemNow(), 0);
  2454. }
  2455. TEST(pikaMain, dump_issue_12l3kjioa) {
  2456. char* lines =
  2457. "if i = 1:\n"
  2458. " print('test')\n"
  2459. "\n";
  2460. /* init */
  2461. pikaMemInfo.heapUsedMax = 0;
  2462. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2463. /* run */
  2464. __platform_printf("BEGIN\r\n");
  2465. obj_run(pikaMain, lines);
  2466. /* collect */
  2467. /* assert */
  2468. /* deinit */
  2469. obj_deinit(pikaMain);
  2470. EXPECT_EQ(pikaMemNow(), 0);
  2471. }
  2472. #if PIKA_SYNTAX_SLICE_ENABLE
  2473. TEST(pikaMain, issue_fa13f4) {
  2474. char* lines =
  2475. "d = {'language':['c','python']}\n"
  2476. "d['language'].append('Java')\n"
  2477. "print(d)\n";
  2478. /* init */
  2479. pikaMemInfo.heapUsedMax = 0;
  2480. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2481. /* run */
  2482. __platform_printf("BEGIN\r\n");
  2483. obj_run(pikaMain, lines);
  2484. /* collect */
  2485. /* assert */
  2486. EXPECT_STREQ(log_buff[0], "{'language': ['c', 'python', 'Java']}\r\n");
  2487. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2488. /* deinit */
  2489. obj_deinit(pikaMain);
  2490. EXPECT_EQ(pikaMemNow(), 0);
  2491. }
  2492. #endif
  2493. TEST(pikaMain, callback_issue_I5L1MI) {
  2494. char* lines =
  2495. "def callback(func):\n"
  2496. " func()\n"
  2497. "\n"
  2498. "def test():\n"
  2499. " print('hello')\n"
  2500. "\n"
  2501. "f = test\n"
  2502. "callback(f)\n";
  2503. /* init */
  2504. pikaMemInfo.heapUsedMax = 0;
  2505. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2506. /* run */
  2507. __platform_printf("BEGIN\r\n");
  2508. obj_run(pikaMain, lines);
  2509. /* collect */
  2510. /* assert */
  2511. EXPECT_STREQ(log_buff[0], "hello\r\n");
  2512. EXPECT_STREQ(log_buff[1], "BEGIN\r\n");
  2513. /* deinit */
  2514. obj_deinit(pikaMain);
  2515. EXPECT_EQ(pikaMemNow(), 0);
  2516. }
  2517. TEST(pikaMain, runchar_multiline) {
  2518. char* lines =
  2519. "def test():\r\n"
  2520. " print('hello')\r\n"
  2521. "\r\n"
  2522. "test()\r\n";
  2523. /* init */
  2524. pikaMemInfo.heapUsedMax = 0;
  2525. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2526. /* run */
  2527. __platform_printf("BEGIN\r\n");
  2528. for (size_t i = 0; i < strGetSize(lines); i++) {
  2529. obj_runChar(pikaMain, lines[i]);
  2530. }
  2531. /* collect */
  2532. /* assert */
  2533. EXPECT_STREQ(log_buff[1], "hello\r\n");
  2534. /* deinit */
  2535. obj_deinit(pikaMain);
  2536. EXPECT_EQ(pikaMemNow(), 0);
  2537. }
  2538. TEST(pikaMain, callback_run_char) {
  2539. char* lines =
  2540. "def test2():\r\n"
  2541. " print('hello')\r\n"
  2542. "\r\n"
  2543. "def test():\r\n"
  2544. " print('hello')\r\n"
  2545. " print('in test')\r\n"
  2546. "\r\n"
  2547. "def test3():\r\n"
  2548. " print('hello3')\r\n"
  2549. "\r\n"
  2550. "test()\r\n";
  2551. /* init */
  2552. pikaMemInfo.heapUsedMax = 0;
  2553. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2554. /* run */
  2555. __platform_printf("BEGIN\r\n");
  2556. for (size_t i = 0; i < strGetSize(lines); i++) {
  2557. obj_runChar(pikaMain, lines[i]);
  2558. }
  2559. /* collect */
  2560. /* assert */
  2561. EXPECT_STREQ(log_buff[1], "in test\r\n");
  2562. /* deinit */
  2563. obj_deinit(pikaMain);
  2564. EXPECT_EQ(pikaMemNow(), 0);
  2565. }
  2566. TEST(pikaMain, REPL_backspace) {
  2567. char* lines = "print('test'\b\b')\r\n";
  2568. /* init */
  2569. pikaMemInfo.heapUsedMax = 0;
  2570. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2571. /* run */
  2572. __platform_printf("BEGIN\r\n");
  2573. for (size_t i = 0; i < strGetSize(lines); i++) {
  2574. obj_runChar(pikaMain, lines[i]);
  2575. }
  2576. /* collect */
  2577. /* assert */
  2578. EXPECT_STREQ(log_buff[1], "tes\r\n");
  2579. /* deinit */
  2580. obj_deinit(pikaMain);
  2581. EXPECT_EQ(pikaMemNow(), 0);
  2582. }
  2583. TEST(pikaMain, REPL_backspace_issue_1) {
  2584. 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";
  2585. /* init */
  2586. pikaMemInfo.heapUsedMax = 0;
  2587. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2588. /* run */
  2589. __platform_printf("BEGIN\r\n");
  2590. for (size_t i = 0; i < strGetSize(lines); i++) {
  2591. obj_runChar(pikaMain, lines[i]);
  2592. }
  2593. /* collect */
  2594. /* assert */
  2595. EXPECT_STREQ(log_buff[1], "'a'\r\n");
  2596. /* deinit */
  2597. obj_deinit(pikaMain);
  2598. EXPECT_EQ(pikaMemNow(), 0);
  2599. }
  2600. TEST(pikaMain, REPL_backspace_issue_2) {
  2601. char* lines = "loop\bp\n";
  2602. /* init */
  2603. pikaMemInfo.heapUsedMax = 0;
  2604. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2605. /* run */
  2606. __platform_printf("BEGIN\r\n");
  2607. for (size_t i = 0; i < strGetSize(lines); i++) {
  2608. obj_runChar(pikaMain, lines[i]);
  2609. }
  2610. /* collect */
  2611. /* assert */
  2612. /* deinit */
  2613. obj_deinit(pikaMain);
  2614. EXPECT_EQ(pikaMemNow(), 0);
  2615. }
  2616. TEST(pikaMain, REPL_key_left) {
  2617. char lines[] = {'1', '2', '+', '3', '4', 0x1b,
  2618. 0x5b, KEY_LEFT, '5', '\n', 0x00};
  2619. /* init */
  2620. pikaMemInfo.heapUsedMax = 0;
  2621. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2622. /* run */
  2623. __platform_printf("BEGIN\r\n");
  2624. for (size_t i = 0; i < strGetSize(lines); i++) {
  2625. obj_runChar(pikaMain, lines[i]);
  2626. }
  2627. /* collect */
  2628. /* assert */
  2629. EXPECT_STREQ(log_buff[1], "366\r\n");
  2630. /* deinit */
  2631. obj_deinit(pikaMain);
  2632. EXPECT_EQ(pikaMemNow(), 0);
  2633. }
  2634. TEST(pikaMain, REPL_key_left_del) {
  2635. char lines[] = {'1', '2', '+', '3', '4', 0x1b,
  2636. 0x5b, KEY_LEFT, '\b', '\n', 0x00};
  2637. /* init */
  2638. pikaMemInfo.heapUsedMax = 0;
  2639. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2640. /* run */
  2641. __platform_printf("BEGIN\r\n");
  2642. for (size_t i = 0; i < strGetSize(lines); i++) {
  2643. obj_runChar(pikaMain, lines[i]);
  2644. }
  2645. /* collect */
  2646. /* assert */
  2647. EXPECT_STREQ(log_buff[1], "16\r\n");
  2648. /* deinit */
  2649. obj_deinit(pikaMain);
  2650. EXPECT_EQ(pikaMemNow(), 0);
  2651. }
  2652. TEST(pikaMain, obj_setNone) {
  2653. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2654. obj_setNone(pikaMain, "n");
  2655. obj_run(pikaMain, "print(n)");
  2656. EXPECT_STREQ(log_buff[0], "None\r\n");
  2657. obj_deinit(pikaMain);
  2658. EXPECT_EQ(pikaMemNow(), 0);
  2659. }
  2660. TEST(pikaMain, obj_setStr_NULL) {
  2661. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2662. obj_setStr(pikaMain, "n", NULL);
  2663. obj_run(pikaMain, "print(n)");
  2664. EXPECT_STREQ(log_buff[2], "NameError: name 'n' is not defined\r\n");
  2665. obj_deinit(pikaMain);
  2666. EXPECT_EQ(pikaMemNow(), 0);
  2667. }
  2668. TEST(pikaMain, _obj_set_str_null) {
  2669. PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
  2670. extern unsigned char pikaModules_py_a[];
  2671. obj_linkLibrary(pikaMain, pikaModules_py_a);
  2672. obj_run(pikaMain,
  2673. "class _test(GTestTask._test):\n"
  2674. " def __init__(self):\n"
  2675. " super().__init__()\n"
  2676. "t = _test()\n");
  2677. obj_deinit(pikaMain);
  2678. EXPECT_EQ(pikaMemNow(), 0);
  2679. }
  2680. TEST_END