pikaMain-test.cpp 83 KB

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