pikaMain-test.cpp 72 KB

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