pikaMain-test.cpp 79 KB

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