parse-test.cpp 55 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145
  1. #include "gtest/gtest.h"
  2. extern "C" {
  3. #include "BaseObj.h"
  4. #include "PikaParser.h"
  5. #include "dataMemory.h"
  6. #include "dataStrs.h"
  7. /* head infomation */
  8. typedef QueueObj AST;
  9. char* Parser_multiLineToAsm(Args* outBuffs, char* multiLine);
  10. char* AST_toPikaASM(AST* ast, Args* outBuffs);
  11. AST* AST_parseLine(char* line, Stack* blockStack);
  12. char* Parser_LineToAsm(Args* buffs, char* line, Stack* blockStack);
  13. int32_t AST_deinit(AST* ast);
  14. char* Lexer_getTokens(Args* outBuffs, char* stmt);
  15. char* Lexer_printTokens(Args* outBuffs, char* tokens);
  16. char* strsPopTokenWithSkip_byStr(Args* buffs,
  17. char* stmts,
  18. char* str,
  19. char skipStart,
  20. char skipEnd);
  21. char* strsGetCleanCmd(Args* outBuffs, char* cmd);
  22. }
  23. TEST(parser, NEW) {
  24. AST* ast = AST_parseLine((char*)"add(a,b)", NULL);
  25. Args* buffs = New_strBuff();
  26. char* pikaAsm = AST_toPikaASM(ast, buffs);
  27. printf("%s", pikaAsm);
  28. args_deinit(buffs);
  29. AST_deinit(ast);
  30. EXPECT_EQ(pikaMemNow(), 0);
  31. }
  32. TEST(parser, add_a_b) {
  33. AST* ast = AST_parseLine((char*)"add( a , b)", NULL);
  34. Args* buffs = New_strBuff();
  35. char* pikaAsm = AST_toPikaASM(ast, buffs);
  36. printf("%s", pikaAsm);
  37. EXPECT_STREQ(pikaAsm,
  38. "B0\n"
  39. "1 REF a\n"
  40. "1 REF b\n"
  41. "0 RUN add\n");
  42. args_deinit(buffs);
  43. AST_deinit(ast);
  44. EXPECT_EQ(pikaMemNow(), 0);
  45. }
  46. TEST(parser, add_a_b_c) {
  47. AST* ast = AST_parseLine((char*)"d = add(add(a,b) , c)", NULL);
  48. Args* buffs = New_strBuff();
  49. char* pikaAsm = AST_toPikaASM(ast, buffs);
  50. printf("%s", pikaAsm);
  51. EXPECT_STREQ(pikaAsm,
  52. "B0\n"
  53. "2 REF a\n"
  54. "2 REF b\n"
  55. "1 RUN add\n"
  56. "1 REF c\n"
  57. "0 RUN add\n"
  58. "0 OUT d\n");
  59. args_deinit(buffs);
  60. AST_deinit(ast);
  61. EXPECT_EQ(pikaMemNow(), 0);
  62. }
  63. TEST(parser, method1) {
  64. AST* ast =
  65. AST_parseLine((char*)"d.p = a.add(b.add(a,se.b) , pmw.c)", NULL);
  66. Args* buffs = New_strBuff();
  67. char* pikaAsm = AST_toPikaASM(ast, buffs);
  68. printf("%s", pikaAsm);
  69. EXPECT_STREQ(pikaAsm,
  70. "B0\n"
  71. "2 REF a\n"
  72. "2 REF se.b\n"
  73. "1 RUN b.add\n"
  74. "1 REF pmw.c\n"
  75. "0 RUN "
  76. "a.add\n"
  77. "0 OUT d.p\n");
  78. args_deinit(buffs);
  79. AST_deinit(ast);
  80. EXPECT_EQ(pikaMemNow(), 0);
  81. }
  82. TEST(parser, method2) {
  83. char* line = (char*)"d.p = a.add(b.add(a,se.b,diek(pp)) , pmw.c())";
  84. Args* buffs = New_strBuff();
  85. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  86. printf("%s", pikaAsm);
  87. EXPECT_STREQ(pikaAsm,
  88. "B0\n"
  89. "2 REF a\n"
  90. "2 REF se.b\n"
  91. "3 REF pp\n"
  92. "2 RUN diek\n"
  93. "1 RUN b.add\n"
  94. "1 RUN pmw.c\n"
  95. "0 RUN a.add\n"
  96. "0 OUT d.p\n");
  97. args_deinit(buffs);
  98. EXPECT_EQ(pikaMemNow(), 0);
  99. }
  100. TEST(parser, method_void) {
  101. char* line = (char*)"a(c())";
  102. Args* buffs = New_strBuff();
  103. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  104. printf("%s", pikaAsm);
  105. EXPECT_STREQ(pikaAsm,(char *)
  106. "B0\n"
  107. "1 RUN c\n"
  108. "0 RUN a\n"
  109. );
  110. args_deinit(buffs);
  111. EXPECT_EQ(pikaMemNow(), 0);
  112. }
  113. TEST(parser, str1) {
  114. char* line = (char*)"literal('2.322')";
  115. Args* buffs = New_strBuff();
  116. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  117. printf("%s", pikaAsm);
  118. EXPECT_STREQ(pikaAsm,
  119. "B0\n"
  120. "1 STR 2.322\n"
  121. "0 RUN literal\n");
  122. args_deinit(buffs);
  123. EXPECT_EQ(pikaMemNow(), 0);
  124. }
  125. TEST(parser, str2) {
  126. char* line = (char*)"b=add(a,literal('1'))";
  127. Args* buffs = New_strBuff();
  128. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  129. printf("%s", pikaAsm);
  130. EXPECT_STREQ(pikaAsm,
  131. "B0\n"
  132. "1 REF a\n"
  133. "2 STR 1\n"
  134. "1 RUN literal\n"
  135. "0 RUN add\n"
  136. "0 OUT b\n");
  137. args_deinit(buffs);
  138. EXPECT_EQ(pikaMemNow(), 0);
  139. }
  140. TEST(parser, num1) {
  141. char* line = (char*)"b=add(a,1)";
  142. Args* buffs = New_strBuff();
  143. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  144. printf("%s", pikaAsm);
  145. EXPECT_STREQ(pikaAsm,
  146. "B0\n"
  147. "1 REF a\n"
  148. "1 NUM 1\n"
  149. "0 RUN add\n"
  150. "0 OUT b\n");
  151. args_deinit(buffs);
  152. EXPECT_EQ(pikaMemNow(), 0);
  153. }
  154. TEST(parser, add_str) {
  155. char* line = (char*)"b=add(a,'1')";
  156. Args* buffs = New_strBuff();
  157. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  158. printf("%s", pikaAsm);
  159. EXPECT_STREQ(pikaAsm,
  160. "B0\n"
  161. "1 REF a\n"
  162. "1 STR 1\n"
  163. "0 RUN add\n"
  164. "0 OUT b\n");
  165. args_deinit(buffs);
  166. EXPECT_EQ(pikaMemNow(), 0);
  167. }
  168. TEST(parser, deep4) {
  169. char* line = (char*)"b = add(add(add(add(1, 2), 3), 4), 5)";
  170. Args* buffs = New_strBuff();
  171. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  172. printf("%s", pikaAsm);
  173. EXPECT_STREQ(pikaAsm,
  174. "B0\n"
  175. "4 NUM 1\n"
  176. "4 NUM 2\n"
  177. "3 RUN add\n"
  178. "3 NUM 3\n"
  179. "2 RUN add\n"
  180. "2 NUM 4\n"
  181. "1 RUN add\n"
  182. "1 NUM 5\n"
  183. "0 RUN add\n"
  184. "0 OUT b\n");
  185. args_deinit(buffs);
  186. EXPECT_EQ(pikaMemNow(), 0);
  187. }
  188. TEST(parser, a_1) {
  189. char* line = (char*)"a = 1";
  190. Args* buffs = New_strBuff();
  191. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  192. printf("%s", pikaAsm);
  193. EXPECT_STREQ(pikaAsm,
  194. "B0\n"
  195. "0 NUM 1\n"
  196. "0 OUT a\n");
  197. args_deinit(buffs);
  198. EXPECT_EQ(pikaMemNow(), 0);
  199. }
  200. TEST(parser, while_true) {
  201. char* line = (char*)"while true:";
  202. Args* buffs = New_strBuff();
  203. char* pikaAsm = Parser_LineToAsm(buffs, line, NULL);
  204. printf("%s", pikaAsm);
  205. EXPECT_STREQ(pikaAsm,
  206. "B0\n"
  207. "0 REF true\n"
  208. "0 JEZ 2\n");
  209. args_deinit(buffs);
  210. EXPECT_EQ(pikaMemNow(), 0);
  211. }
  212. static char* parse(const char* line,
  213. Args* outBuffs,
  214. char* pikaAsm,
  215. Stack* blockStack) {
  216. printf("%s\n", line);
  217. Args* runBuffs = New_strBuff();
  218. pikaAsm = strsAppend(runBuffs, pikaAsm,
  219. Parser_LineToAsm(runBuffs, (char*)line, blockStack));
  220. pikaAsm = strsCopy(outBuffs, pikaAsm);
  221. args_deinit(runBuffs);
  222. return pikaAsm;
  223. }
  224. TEST(parser, while_true_block) {
  225. Args* bf = New_strBuff();
  226. Stack* bs = New_Stack();
  227. char* s = strsCopy(bf, (char*)"");
  228. s = parse("while true:", bf, s, bs);
  229. s = parse(" rgb.flow()", bf, s, bs);
  230. s = parse("", bf, s, bs);
  231. printf("%s", s);
  232. EXPECT_STREQ(s,
  233. "B0\n"
  234. "0 REF true\n"
  235. "0 JEZ 2\n"
  236. "B1\n"
  237. "0 RUN rgb.flow\n"
  238. "B0\n"
  239. "0 JMP -1\n"
  240. "B0\n");
  241. stack_deinit(bs);
  242. args_deinit(bf);
  243. EXPECT_EQ(pikaMemNow(), 0);
  244. }
  245. TEST(parser, while_true_false) {
  246. Args* bf = New_strBuff();
  247. Stack* bs = New_Stack();
  248. char* s = strsCopy(bf, (char*)"");
  249. s = parse("while true:", bf, s, bs);
  250. s = parse(" rgb.flow()", bf, s, bs);
  251. s = parse(" while false:", bf, s, bs);
  252. s = parse(" a=3", bf, s, bs);
  253. s = parse(" test.on(add(2,3))", bf, s, bs);
  254. s = parse(" print('flowing')", bf, s, bs);
  255. s = parse("", bf, s, bs);
  256. Arg* buffArg = arg_setStr(NULL, (char*)"", s);
  257. stack_deinit(bs);
  258. args_deinit(bf);
  259. s = arg_getStr(buffArg);
  260. printf("%s", s);
  261. EXPECT_STREQ(s,
  262. "B0\n"
  263. "0 REF true\n"
  264. "0 JEZ 2\n"
  265. "B1\n"
  266. "0 RUN rgb.flow\n"
  267. "B1\n"
  268. "0 REF false\n"
  269. "0 JEZ 2\n"
  270. "B2\n"
  271. "0 NUM 3\n"
  272. "0 OUT a\n"
  273. "B2\n"
  274. "2 NUM 2\n"
  275. "2 NUM 3\n"
  276. "1 RUN add\n"
  277. "0 RUN test.on\n"
  278. "B1\n"
  279. "0 JMP -1\n"
  280. "B1\n"
  281. "1 STR flowing\n"
  282. "0 RUN print\n"
  283. "B0\n"
  284. "0 JMP -1\n"
  285. "B0\n");
  286. arg_deinit(buffArg);
  287. EXPECT_EQ(pikaMemNow(), 0);
  288. }
  289. TEST(parser, while_true_false_both_exit) {
  290. Args* bf = New_strBuff();
  291. Stack* bs = New_Stack();
  292. char* s = strsCopy(bf, (char*)"");
  293. s = parse("while true:", bf, s, bs);
  294. s = parse(" rgb.flow()", bf, s, bs);
  295. s = parse(" while false:", bf, s, bs);
  296. s = parse(" a=3", bf, s, bs);
  297. s = parse(" test.on(add(2,3))", bf, s, bs);
  298. s = parse("", bf, s, bs);
  299. printf("%s", s);
  300. EXPECT_STREQ(s,
  301. "B0\n"
  302. "0 REF true\n"
  303. "0 JEZ 2\n"
  304. "B1\n"
  305. "0 RUN rgb.flow\n"
  306. "B1\n"
  307. "0 REF false\n"
  308. "0 JEZ 2\n"
  309. "B2\n"
  310. "0 NUM 3\n"
  311. "0 OUT a\n"
  312. "B2\n"
  313. "2 NUM 2\n"
  314. "2 NUM 3\n"
  315. "1 RUN add\n"
  316. "0 RUN test.on\n"
  317. "B1\n"
  318. "0 JMP -1\n"
  319. "B0\n"
  320. "0 JMP -1\n"
  321. "B0\n");
  322. stack_deinit(bs);
  323. args_deinit(bf);
  324. EXPECT_EQ(pikaMemNow(), 0);
  325. }
  326. TEST(parser, if_) {
  327. Args* bf = New_strBuff();
  328. Stack* bs = New_Stack();
  329. char* s = strsCopy(bf, (char*)"");
  330. s = parse("if true:", bf, s, bs);
  331. s = parse(" rgb.flow()", bf, s, bs);
  332. s = parse("", bf, s, bs);
  333. printf("%s", s);
  334. EXPECT_STREQ(s,
  335. "B0\n"
  336. "0 REF true\n"
  337. "0 JEZ 1\n"
  338. "B1\n"
  339. "0 RUN rgb.flow\n"
  340. "B0\n");
  341. stack_deinit(bs);
  342. args_deinit(bf);
  343. EXPECT_EQ(pikaMemNow(), 0);
  344. }
  345. extern PikaMemInfo pikaMemInfo;
  346. TEST(parser, while_true_if_false_both_exit) {
  347. pikaMemInfo.heapUsedMax = 0;
  348. Args* bf = New_strBuff();
  349. Stack* bs = New_Stack();
  350. char* s = strsCopy(bf, (char*)"");
  351. s = parse("while true:", bf, s, bs);
  352. s = parse(" rgb.flow()", bf, s, bs);
  353. s = parse(" if false:", bf, s, bs);
  354. s = parse(" a=3", bf, s, bs);
  355. s = parse(" test.on(add(2,3))", bf, s, bs);
  356. s = parse("", bf, s, bs);
  357. printf("%s", s);
  358. EXPECT_STREQ(s,
  359. "B0\n"
  360. "0 REF true\n"
  361. "0 JEZ 2\n"
  362. "B1\n"
  363. "0 RUN rgb.flow\n"
  364. "B1\n"
  365. "0 REF false\n"
  366. "0 JEZ 1\n"
  367. "B2\n"
  368. "0 NUM 3\n"
  369. "0 OUT a\n"
  370. "B2\n"
  371. "2 NUM 2\n"
  372. "2 NUM 3\n"
  373. "1 RUN add\n"
  374. "0 RUN test.on\n"
  375. "B0\n"
  376. "0 JMP -1\n"
  377. "B0\n");
  378. stack_deinit(bs);
  379. args_deinit(bf);
  380. EXPECT_EQ(pikaMemNow(), 0);
  381. }
  382. TEST(parser, multiLine) {
  383. pikaMemInfo.heapUsedMax = 0;
  384. Args* buffs = New_strBuff();
  385. char* lines =(char *)
  386. "while true:\n"
  387. " rgb.flow()\n"
  388. " if false:\n"
  389. " a=3\n"
  390. " test.on(add(2,3))\n"
  391. "\n";
  392. printf("%s", lines);
  393. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  394. printf("%s", pikaAsm);
  395. EXPECT_STREQ(pikaAsm,
  396. "B0\n"
  397. "0 REF true\n"
  398. "0 JEZ 2\n"
  399. "B1\n"
  400. "0 RUN rgb.flow\n"
  401. "B1\n"
  402. "0 REF false\n"
  403. "0 JEZ 1\n"
  404. "B2\n"
  405. "0 NUM 3\n"
  406. "0 OUT a\n"
  407. "B2\n"
  408. "2 NUM 2\n"
  409. "2 NUM 3\n"
  410. "1 RUN add\n"
  411. "0 RUN test.on\n"
  412. "B0\n"
  413. "0 JMP -1\n"
  414. "B0\n");
  415. args_deinit(buffs);
  416. EXPECT_EQ(pikaMemNow(), 0);
  417. }
  418. TEST(parser, pikaPi) {
  419. pikaMemInfo.heapUsedMax = 0;
  420. Args* buffs = New_strBuff();
  421. const char lines[] =
  422. "\n"
  423. "time = STM32.Time()\n"
  424. "uart = STM32.UART()\n"
  425. "adc = STM32.ADC()\n"
  426. "pin = STM32.GPIO()\n"
  427. "pwm = STM32.PWM()\n"
  428. "uart = STM32.UART()\n"
  429. "rgb = PikaPiZero.RGB()\n"
  430. "mem = PikaStdLib.MemChecker()\n"
  431. "op = PikaMath.Operator()\n"
  432. "\n"
  433. "uart.init()\n"
  434. "uart.setId(1)\n"
  435. "uart.setBaudRate(115200)\n"
  436. "uart.enable()\n"
  437. "\n"
  438. "rgb.init()\n"
  439. "rgb.enable()\n"
  440. "\n"
  441. "print('hello 2')\n"
  442. "print('mem used max:')\n"
  443. "mem.max() \n"
  444. "\n"
  445. "while True:\n"
  446. " time.sleep_ms(10)\n"
  447. " rgb.flow()\n"
  448. " print('flowing')\n"
  449. "\n"
  450. "\n"
  451. "\n"
  452. "\n";
  453. printf("%s", lines);
  454. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  455. printf("mem max in parse: %0.2f Kb\n", pikaMemMax() / 1024.0);
  456. printf("%s", pikaAsm);
  457. EXPECT_STREQ(pikaAsm,
  458. "B0\n"
  459. "B0\n"
  460. "0 RUN STM32.Time\n"
  461. "0 OUT time\n"
  462. "B0\n"
  463. "0 RUN STM32.UART\n"
  464. "0 OUT uart\n"
  465. "B0\n"
  466. "0 RUN STM32.ADC\n"
  467. "0 OUT adc\n"
  468. "B0\n"
  469. "0 RUN STM32.GPIO\n"
  470. "0 OUT pin\n"
  471. "B0\n"
  472. "0 RUN STM32.PWM\n"
  473. "0 OUT pwm\n"
  474. "B0\n"
  475. "0 RUN STM32.UART\n"
  476. "0 OUT uart\n"
  477. "B0\n"
  478. "0 RUN PikaPiZero.RGB\n"
  479. "0 OUT rgb\n"
  480. "B0\n"
  481. "0 RUN PikaStdLib.MemChecker\n"
  482. "0 OUT mem\n"
  483. "B0\n"
  484. "0 RUN PikaMath.Operator\n"
  485. "0 OUT op\n"
  486. "B0\n"
  487. "B0\n"
  488. "0 RUN uart.init\n"
  489. "B0\n"
  490. "1 NUM 1\n"
  491. "0 RUN uart.setId\n"
  492. "B0\n"
  493. "1 NUM 115200\n"
  494. "0 RUN uart.setBaudRate\n"
  495. "B0\n"
  496. "0 RUN uart.enable\n"
  497. "B0\n"
  498. "B0\n"
  499. "0 RUN rgb.init\n"
  500. "B0\n"
  501. "0 RUN rgb.enable\n"
  502. "B0\n"
  503. "B0\n"
  504. "1 STR hello 2\n"
  505. "0 RUN print\n"
  506. "B0\n"
  507. "1 STR mem used max:\n"
  508. "0 RUN print\n"
  509. "B0\n"
  510. "0 RUN mem.max\n"
  511. "B0\n"
  512. "B0\n"
  513. "0 REF True\n"
  514. "0 JEZ 2\n"
  515. "B1\n"
  516. "1 NUM 10\n"
  517. "0 RUN time.sleep_ms\n"
  518. "B1\n"
  519. "0 RUN rgb.flow\n"
  520. "B1\n"
  521. "1 STR flowing\n"
  522. "0 RUN print\n"
  523. "B0\n"
  524. "0 JMP -1\n"
  525. "B0\n"
  526. "B0\n"
  527. "B0\n"
  528. "B0\n");
  529. args_deinit(buffs);
  530. EXPECT_EQ(pikaMemNow(), 0);
  531. }
  532. TEST(parser, add) {
  533. pikaMemInfo.heapUsedMax = 0;
  534. Args* buffs = New_strBuff();
  535. char* lines = (char*)"a = 1 + 1\n";
  536. printf("%s", lines);
  537. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  538. printf("%s", pikaAsm);
  539. EXPECT_STREQ(pikaAsm,
  540. "B0\n"
  541. "1 NUM 1\n"
  542. "1 NUM 1\n"
  543. "0 OPT +\n"
  544. "0 OUT a\n");
  545. args_deinit(buffs);
  546. EXPECT_EQ(pikaMemNow(), 0);
  547. }
  548. TEST(parser, add_3) {
  549. pikaMemInfo.heapUsedMax = 0;
  550. Args* buffs = New_strBuff();
  551. char* lines = (char*)"a = 1 + 2 + 3\n";
  552. printf("%s", lines);
  553. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  554. printf("%s", pikaAsm);
  555. EXPECT_STREQ(pikaAsm,
  556. "B0\n"
  557. "1 NUM 1\n"
  558. "2 NUM 2\n"
  559. "2 NUM 3\n"
  560. "1 OPT +\n"
  561. "0 OPT +\n"
  562. "0 OUT a\n");
  563. args_deinit(buffs);
  564. EXPECT_EQ(pikaMemNow(), 0);
  565. }
  566. TEST(parser, add_a_pp) {
  567. pikaMemInfo.heapUsedMax = 0;
  568. Args* buffs = New_strBuff();
  569. char* lines = (char*)"a = a + 1\n";
  570. printf("%s", lines);
  571. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  572. printf("%s", pikaAsm);
  573. EXPECT_STREQ(pikaAsm,
  574. "B0\n"
  575. "1 REF a\n"
  576. "1 NUM 1\n"
  577. "0 OPT +\n"
  578. "0 OUT a\n");
  579. args_deinit(buffs);
  580. EXPECT_EQ(pikaMemNow(), 0);
  581. }
  582. TEST(parser, while_a_pp) {
  583. pikaMemInfo.heapUsedMax = 0;
  584. Args* buffs = New_strBuff();
  585. char* lines = (char*)
  586. "while a < 10:\n"
  587. " print(a)\n"
  588. " a = a + 1\n"
  589. "\n";
  590. printf("%s", lines);
  591. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  592. printf("%s", pikaAsm);
  593. EXPECT_STREQ(pikaAsm,
  594. "B0\n"
  595. "1 REF a\n"
  596. "1 NUM 10\n"
  597. "0 OPT <\n"
  598. "0 JEZ 2\n"
  599. "B1\n"
  600. "1 REF a\n"
  601. "0 RUN print\n"
  602. "B1\n"
  603. "1 REF a\n"
  604. "1 NUM 1\n"
  605. "0 OPT +\n"
  606. "0 OUT a\n"
  607. "B0\n"
  608. "0 JMP -1\n"
  609. "B0\n");
  610. args_deinit(buffs);
  611. EXPECT_EQ(pikaMemNow(), 0);
  612. }
  613. TEST(parser, add_m2p3) {
  614. pikaMemInfo.heapUsedMax = 0;
  615. Args* buffs = New_strBuff();
  616. char* lines = (char*)"a = 1 * 2 + 3\n";
  617. printf("%s", lines);
  618. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  619. printf("%s", pikaAsm);
  620. EXPECT_STREQ(pikaAsm,
  621. "B0\n"
  622. "2 NUM 1\n"
  623. "2 NUM 2\n"
  624. "1 OPT *\n"
  625. "1 NUM 3\n"
  626. "0 OPT +\n"
  627. "0 OUT a\n");
  628. args_deinit(buffs);
  629. EXPECT_EQ(pikaMemNow(), 0);
  630. }
  631. TEST(parser, add_m2p3_) {
  632. pikaMemInfo.heapUsedMax = 0;
  633. Args* buffs = New_strBuff();
  634. char* lines = (char*)"a = 1 * (2 + 3)\n";
  635. printf("%s", lines);
  636. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  637. printf("%s", pikaAsm);
  638. EXPECT_STREQ(pikaAsm,
  639. "B0\n"
  640. "1 NUM 1\n"
  641. "3 NUM 2\n"
  642. "3 NUM 3\n"
  643. "2 OPT +\n"
  644. "1 RUN \n"
  645. "0 OPT *\n"
  646. "0 OUT a\n");
  647. args_deinit(buffs);
  648. EXPECT_EQ(pikaMemNow(), 0);
  649. }
  650. TEST(parser, add_m12p3_) {
  651. pikaMemInfo.heapUsedMax = 0;
  652. Args* buffs = New_strBuff();
  653. char* lines = (char*)"a = (1 + 2) * 3\n";
  654. printf("%s", lines);
  655. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  656. printf("%s", pikaAsm);
  657. EXPECT_STREQ(pikaAsm,
  658. "B0\n"
  659. "3 NUM 1\n"
  660. "3 NUM 2\n"
  661. "2 OPT +\n"
  662. "1 RUN \n"
  663. "1 NUM 3\n"
  664. "0 OPT *\n"
  665. "0 OUT a\n");
  666. args_deinit(buffs);
  667. EXPECT_EQ(pikaMemNow(), 0);
  668. }
  669. TEST(parser, method_equ) {
  670. pikaMemInfo.heapUsedMax = 0;
  671. Args* buffs = New_strBuff();
  672. char* lines = (char*)"if right.read() == 1:\n";
  673. printf("%s", lines);
  674. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  675. printf("%s", pikaAsm);
  676. EXPECT_STREQ(pikaAsm,
  677. "B0\n"
  678. "1 RUN right.read\n"
  679. "1 NUM 1\n"
  680. "0 OPT ==\n"
  681. "0 JEZ 1\n");
  682. args_deinit(buffs);
  683. EXPECT_EQ(pikaMemNow(), 0);
  684. }
  685. TEST(parser, equ_method) {
  686. pikaMemInfo.heapUsedMax = 0;
  687. Args* buffs = New_strBuff();
  688. char* lines = (char*)"if 1 == right.read() :\n";
  689. printf("%s", lines);
  690. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  691. printf("%s", pikaAsm);
  692. EXPECT_STREQ(pikaAsm,
  693. "B0\n"
  694. "1 NUM 1\n"
  695. "1 RUN right.read\n"
  696. "0 OPT ==\n"
  697. "0 JEZ 1\n");
  698. args_deinit(buffs);
  699. EXPECT_EQ(pikaMemNow(), 0);
  700. }
  701. TEST(parser, def_add) {
  702. pikaMemInfo.heapUsedMax = 0;
  703. Args* buffs = New_strBuff();
  704. char* lines = (char*)
  705. "def add(a, b):\n"
  706. " a + b\n"
  707. "\n"
  708. ;
  709. printf("%s", lines);
  710. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  711. printf("%s", pikaAsm);
  712. EXPECT_STREQ(pikaAsm,(char *)
  713. "B0\n"
  714. "0 DEF add(a,b)\n"
  715. "0 JMP 1\n"
  716. "B1\n"
  717. "1 REF a\n"
  718. "1 REF b\n"
  719. "0 OPT +\n"
  720. "B1\n"
  721. "0 RET\n"
  722. "B0\n"
  723. );
  724. args_deinit(buffs);
  725. EXPECT_EQ(pikaMemNow(), 0);
  726. }
  727. TEST(parser, def_add_return) {
  728. pikaMemInfo.heapUsedMax = 0;
  729. Args* buffs = New_strBuff();
  730. char* lines = (char*)
  731. "def add(a, b):\n"
  732. " return a + b\n"
  733. "\n"
  734. ;
  735. printf("%s", lines);
  736. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  737. printf("%s", pikaAsm);
  738. EXPECT_STREQ(pikaAsm,(char *)
  739. "B0\n"
  740. "0 DEF add(a,b)\n"
  741. "0 JMP 1\n"
  742. "B1\n"
  743. "1 REF a\n"
  744. "1 REF b\n"
  745. "0 OPT +\n"
  746. "0 RET\n"
  747. "B1\n"
  748. "0 RET\n"
  749. "B0\n"
  750. );
  751. args_deinit(buffs);
  752. EXPECT_EQ(pikaMemNow(), 0);
  753. }
  754. TEST(parser, def_while_return) {
  755. pikaMemInfo.heapUsedMax = 0;
  756. Args* buffs = New_strBuff();
  757. char* lines = (char*)
  758. "def add(a, b):\n"
  759. " while True:\n"
  760. " return a + b\n"
  761. "\n"
  762. ;
  763. printf("%s", lines);
  764. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  765. printf("%s", pikaAsm);
  766. EXPECT_STREQ(pikaAsm,(char *)
  767. "B0\n"
  768. "0 DEF add(a,b)\n"
  769. "0 JMP 1\n"
  770. "B1\n"
  771. "0 REF True\n"
  772. "0 JEZ 2\n"
  773. "B2\n"
  774. "1 REF a\n"
  775. "1 REF b\n"
  776. "0 OPT +\n"
  777. "0 RET\n"
  778. "B1\n"
  779. "0 JMP -1\n"
  780. "B1\n"
  781. "0 RET\n"
  782. "B0\n"
  783. );
  784. args_deinit(buffs);
  785. EXPECT_EQ(pikaMemNow(), 0);
  786. }
  787. TEST(parser, def_while_return_void) {
  788. pikaMemInfo.heapUsedMax = 0;
  789. Args* buffs = New_strBuff();
  790. char* lines = (char*)
  791. "def add(a, b):\n"
  792. " while True:\n"
  793. " return\n"
  794. "\n"
  795. ;
  796. printf("%s", lines);
  797. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  798. printf("%s", pikaAsm);
  799. EXPECT_STREQ(pikaAsm,(char *)
  800. "B0\n"
  801. "0 DEF add(a,b)\n"
  802. "0 JMP 1\n"
  803. "B1\n"
  804. "0 REF True\n"
  805. "0 JEZ 2\n"
  806. "B2\n"
  807. "0 RET\n"
  808. "B1\n"
  809. "0 JMP -1\n"
  810. "B1\n"
  811. "0 RET\n"
  812. "B0\n"
  813. );
  814. args_deinit(buffs);
  815. EXPECT_EQ(pikaMemNow(), 0);
  816. }
  817. TEST(parser, signed_num) {
  818. pikaMemInfo.heapUsedMax = 0;
  819. Args* buffs = New_strBuff();
  820. char* lines = (char*)"a = -1\n";
  821. printf("%s", lines);
  822. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  823. printf("%s", pikaAsm);
  824. EXPECT_STREQ(pikaAsm,(char *)
  825. "B0\n"
  826. "0 NUM -1\n"
  827. "0 OUT a\n"
  828. );
  829. args_deinit(buffs);
  830. EXPECT_EQ(pikaMemNow(), 0);
  831. }
  832. TEST(parser, comp_signed_num) {
  833. pikaMemInfo.heapUsedMax = 0;
  834. Args* buffs = New_strBuff();
  835. char* lines = (char*)"if a > -1:\n";
  836. printf("%s", lines);
  837. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  838. printf("%s", pikaAsm);
  839. EXPECT_STREQ(pikaAsm,(char *)
  840. "B0\n"
  841. "1 REF a\n"
  842. "1 NUM -1\n"
  843. "0 OPT >\n"
  844. "0 JEZ 1\n"
  845. );
  846. args_deinit(buffs);
  847. EXPECT_EQ(pikaMemNow(), 0);
  848. }
  849. TEST(lexser, symbol_add) {
  850. /* init */
  851. pikaMemInfo.heapUsedMax = 0;
  852. Args* buffs = New_strBuff();
  853. /* run */
  854. char* tokens = Lexer_getTokens(buffs, (char*)" res = add(1,2)");
  855. char* printTokens = Lexer_printTokens(buffs, tokens);
  856. printf((char*)"%s\n", printTokens);
  857. /* assert */
  858. EXPECT_STREQ(printTokens,
  859. "{sym}res{opt}={sym}add{dvd}({lit}1{dvd},{lit}2{dvd})");
  860. /* deinit */
  861. args_deinit(buffs);
  862. EXPECT_EQ(pikaMemNow(), 0);
  863. }
  864. TEST(lexser, symbol_1) {
  865. /* init */
  866. pikaMemInfo.heapUsedMax = 0;
  867. Args* buffs = New_strBuff();
  868. /* run */
  869. char* tokens = Lexer_getTokens(buffs, (char*)"a(");
  870. /* assert */
  871. EXPECT_EQ(tokens, (char*)NULL);
  872. /* deinit */
  873. args_deinit(buffs);
  874. EXPECT_EQ(pikaMemNow(), 0);
  875. }
  876. TEST(lexser, operator_not) {
  877. /* init */
  878. pikaMemInfo.heapUsedMax = 0;
  879. Args* buffs = New_strBuff();
  880. /* run */
  881. char* tokens = Lexer_getTokens(buffs, (char*)"not not not ");
  882. char* printTokens = Lexer_printTokens(buffs, tokens);
  883. printf((char*)"%s\n", printTokens);
  884. /* assert */
  885. EXPECT_STREQ(printTokens, "{opt} not {opt} not {opt} not ");
  886. /* deinit */
  887. args_deinit(buffs);
  888. EXPECT_EQ(pikaMemNow(), 0);
  889. }
  890. TEST(lexser, symbol_Nag) {
  891. /* init */
  892. pikaMemInfo.heapUsedMax = 0;
  893. Args* buffs = New_strBuff();
  894. /* run */
  895. char* tokens = Lexer_getTokens(buffs, (char*)"-10-20");
  896. char* printTokens = Lexer_printTokens(buffs, tokens);
  897. printf((char*)"%s\n", printTokens);
  898. /* assert */
  899. EXPECT_STREQ(printTokens, "{lit}-10{opt}-{lit}20");
  900. /* deinit */
  901. args_deinit(buffs);
  902. EXPECT_EQ(pikaMemNow(), 0);
  903. }
  904. TEST(lexser, operator_all) {
  905. /* init */
  906. pikaMemInfo.heapUsedMax = 0;
  907. Args* buffs = New_strBuff();
  908. /* run */
  909. char* tokens = Lexer_getTokens(buffs, (char*)
  910. "not or and "
  911. "+ += - -="
  912. "* ** *= **="
  913. "/ // /= //="
  914. "% %= = == !="
  915. "> >= >>"
  916. "< <= <<"
  917. "&|^~"
  918. );
  919. char* printTokens = Lexer_printTokens(buffs, tokens);
  920. printf((char*)"%s\n", printTokens);
  921. /* assert */
  922. EXPECT_STREQ(
  923. printTokens,
  924. "{opt} not {opt} or {opt} and {opt}+{opt}+={opt}-{opt}-={opt}*{opt}*"
  925. "*{opt}*={opt}**={opt}/{opt}//{opt}/={opt}//"
  926. "={opt}%{opt}%={opt}={opt}=={opt}!={opt}>{opt}>={opt}>>{opt}<{"
  927. "opt}<={opt}<<{opt}&{opt}|{opt}^{opt}~");
  928. /* deinit */
  929. args_deinit(buffs);
  930. EXPECT_EQ(pikaMemNow(), 0);
  931. }
  932. TEST(lexser, symbol_2) {
  933. /* init */
  934. pikaMemInfo.heapUsedMax = 0;
  935. Args* buffs = New_strBuff();
  936. /* run */
  937. char* tokens = Lexer_getTokens(buffs, (char*)"a+b-c(25**=ek)!=-28");
  938. char* printTokens = Lexer_printTokens(buffs, tokens);
  939. printf((char*)"%s\n", printTokens);
  940. /* assert */
  941. EXPECT_STREQ(printTokens,
  942. "{sym}a{opt}+{sym}b{opt}-{sym}c{dvd}({lit}25{opt}**={sym}ek{"
  943. "dvd}){opt}!={lit}-28");
  944. /* deinit */
  945. args_deinit(buffs);
  946. EXPECT_EQ(pikaMemNow(), 0);
  947. }
  948. TEST(lexser, symbol_and) {
  949. /* init */
  950. pikaMemInfo.heapUsedMax = 0;
  951. Args* buffs = New_strBuff();
  952. /* run */
  953. char* tokens = Lexer_getTokens(buffs, (char*)" res = add(1 and lkj,2)");
  954. char* printTokens = Lexer_printTokens(buffs, tokens);
  955. printf((char*)"%s\n", printTokens);
  956. /* assert */
  957. EXPECT_STREQ(printTokens,
  958. "{sym}res{opt}={sym}add{dvd}({lit}1{opt} and "
  959. "{sym}lkj{dvd},{lit}2{dvd})");
  960. /* deinit */
  961. args_deinit(buffs);
  962. EXPECT_EQ(pikaMemNow(), 0);
  963. }
  964. TEST(lexser, sting) {
  965. /* init */
  966. pikaMemInfo.heapUsedMax = 0;
  967. Args* buffs = New_strBuff();
  968. /* run */
  969. char* tokens = Lexer_getTokens(buffs, (char*)" a= 'elk 2'");
  970. char* printTokens = Lexer_printTokens(buffs, tokens);
  971. printf((char*)"%s\n", printTokens);
  972. /* assert */
  973. EXPECT_STREQ(printTokens, "{sym}a{opt}={lit}'elk 2'");
  974. /* deinit */
  975. args_deinit(buffs);
  976. EXPECT_EQ(pikaMemNow(), 0);
  977. }
  978. TEST(lexser, num_1) {
  979. /* init */
  980. pikaMemInfo.heapUsedMax = 0;
  981. Args* buffs = New_strBuff();
  982. /* run */
  983. char* tokens = Lexer_getTokens(buffs, (char*)"1");
  984. char* printTokens = Lexer_printTokens(buffs, tokens);
  985. printf((char*)"%s\n", printTokens);
  986. /* assert */
  987. EXPECT_STREQ(printTokens, "{lit}1");
  988. /* deinit */
  989. args_deinit(buffs);
  990. EXPECT_EQ(pikaMemNow(), 0);
  991. }
  992. TEST(lexser, jjcc) {
  993. /* init */
  994. pikaMemInfo.heapUsedMax = 0;
  995. Args* buffs = New_strBuff();
  996. /* run */
  997. char* tokens = Lexer_getTokens(buffs, (char*)"a = (1 + 1.1) * 3 - 2 /4.0");
  998. char* printTokens = Lexer_printTokens(buffs, tokens);
  999. printf((char*)"%s\n", printTokens);
  1000. /* assert */
  1001. EXPECT_STREQ(printTokens,
  1002. "{sym}a{opt}={dvd}({lit}1{opt}+{lit}1.1{dvd}){opt}*{lit}3{opt}"
  1003. "-{lit}2{opt}/{lit}4.0");
  1004. /* deinit */
  1005. args_deinit(buffs);
  1006. EXPECT_EQ(pikaMemNow(), 0);
  1007. }
  1008. TEST(parser, pop_by_str) {
  1009. Args* buffs = New_strBuff();
  1010. char* tokens = strsCopy(buffs, (char*)"3(>=)2>=29");
  1011. char* token1 =
  1012. strsPopTokenWithSkip_byStr(buffs, tokens, (char*)">=", '(', ')');
  1013. char* token2 = tokens;
  1014. /* assert */
  1015. EXPECT_STREQ((char*)"3(>=)2", token1);
  1016. EXPECT_STREQ((char*)"29", token2);
  1017. args_deinit(buffs);
  1018. }
  1019. TEST(parser, mm) {
  1020. pikaMemInfo.heapUsedMax = 0;
  1021. Args* buffs = New_strBuff();
  1022. char* lines = (char*)"a = a ** -1\n";
  1023. printf("%s", lines);
  1024. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1025. printf("%s", pikaAsm);
  1026. EXPECT_STREQ(pikaAsm,(char *)
  1027. "B0\n"
  1028. "1 REF a\n"
  1029. "1 NUM -1\n"
  1030. "0 OPT **\n"
  1031. "0 OUT a\n"
  1032. );
  1033. args_deinit(buffs);
  1034. EXPECT_EQ(pikaMemNow(), 0);
  1035. }
  1036. TEST(parser, self_inc) {
  1037. pikaMemInfo.heapUsedMax = 0;
  1038. Args* buffs = New_strBuff();
  1039. char* lines = (char*)
  1040. "a += -1\n"
  1041. "a -= -1\n"
  1042. "a *= -1\n"
  1043. "a /= -1\n"
  1044. "a **= -1\n"
  1045. "a //= -1\n"
  1046. "a >= -1\n"
  1047. "a <= -1\n"
  1048. "a != -1\n"
  1049. "a %= -1\n"
  1050. ;
  1051. printf("%s", lines);
  1052. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1053. printf("%s", pikaAsm);
  1054. EXPECT_STREQ(pikaAsm,(char *)
  1055. "B0\n"
  1056. "1 REF a\n"
  1057. "1 NUM -1\n"
  1058. "0 OPT +=\n"
  1059. "B0\n"
  1060. "1 REF a\n"
  1061. "1 NUM -1\n"
  1062. "0 OPT -=\n"
  1063. "B0\n"
  1064. "1 REF a\n"
  1065. "1 NUM -1\n"
  1066. "0 OPT *=\n"
  1067. "B0\n"
  1068. "1 REF a\n"
  1069. "1 NUM -1\n"
  1070. "0 OPT /=\n"
  1071. "B0\n"
  1072. "1 REF a\n"
  1073. "1 NUM -1\n"
  1074. "0 OPT **=\n"
  1075. "B0\n"
  1076. "1 REF a\n"
  1077. "1 NUM -1\n"
  1078. "0 OPT //=\n"
  1079. "B0\n"
  1080. "1 REF a\n"
  1081. "1 NUM -1\n"
  1082. "0 OPT >=\n"
  1083. "B0\n"
  1084. "1 REF a\n"
  1085. "1 NUM -1\n"
  1086. "0 OPT <=\n"
  1087. "B0\n"
  1088. "1 REF a\n"
  1089. "1 NUM -1\n"
  1090. "0 OPT !=\n"
  1091. "B0\n"
  1092. "1 REF a\n"
  1093. "1 NUM -1\n"
  1094. "0 OPT %=\n"
  1095. );
  1096. args_deinit(buffs);
  1097. EXPECT_EQ(pikaMemNow(), 0);
  1098. }
  1099. TEST(parser, n_n1) {
  1100. pikaMemInfo.heapUsedMax = 0;
  1101. Args* buffs = New_strBuff();
  1102. char* lines = (char*)"a = ~-1\n";
  1103. printf("%s", lines);
  1104. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1105. printf("%s", pikaAsm);
  1106. EXPECT_STREQ(pikaAsm,(char *)
  1107. "B0\n"
  1108. "1 NUM -1\n"
  1109. "0 OPT ~\n"
  1110. "0 OUT a\n"
  1111. );
  1112. args_deinit(buffs);
  1113. EXPECT_EQ(pikaMemNow(), 0);
  1114. }
  1115. TEST(parser, or_) {
  1116. pikaMemInfo.heapUsedMax = 0;
  1117. Args* buffs = New_strBuff();
  1118. char* lines = (char*)"( a>1) or (b<= 3)\n";
  1119. printf("%s", lines);
  1120. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1121. printf("%s", pikaAsm);
  1122. EXPECT_STREQ(pikaAsm,(char *)
  1123. "B0\n"
  1124. "3 REF a\n"
  1125. "3 NUM 1\n"
  1126. "2 OPT >\n"
  1127. "1 RUN \n"
  1128. "3 REF b\n"
  1129. "3 NUM 3\n"
  1130. "2 OPT <=\n"
  1131. "1 RUN \n"
  1132. "0 OPT or \n"
  1133. );
  1134. args_deinit(buffs);
  1135. EXPECT_EQ(pikaMemNow(), 0);
  1136. }
  1137. TEST(parser, _or_) {
  1138. pikaMemInfo.heapUsedMax = 0;
  1139. Args* buffs = New_strBuff();
  1140. char* lines = (char*)
  1141. "if (1>2) and (2>1):\n"
  1142. " b = 1\n"
  1143. "\n"
  1144. ;
  1145. printf("%s", lines);
  1146. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1147. printf("%s", pikaAsm);
  1148. EXPECT_STREQ(pikaAsm,(char *)
  1149. "B0\n"
  1150. "3 NUM 1\n"
  1151. "3 NUM 2\n"
  1152. "2 OPT >\n"
  1153. "1 RUN \n"
  1154. "3 NUM 2\n"
  1155. "3 NUM 1\n"
  1156. "2 OPT >\n"
  1157. "1 RUN \n"
  1158. "0 OPT and \n"
  1159. "0 JEZ 1\n"
  1160. "B1\n"
  1161. "0 NUM 1\n"
  1162. "0 OUT b\n"
  1163. "B0\n"
  1164. );
  1165. args_deinit(buffs);
  1166. EXPECT_EQ(pikaMemNow(), 0);
  1167. }
  1168. TEST(parser, annotation) {
  1169. pikaMemInfo.heapUsedMax = 0;
  1170. Args* buffs = New_strBuff();
  1171. char* lines = (char*)"a = t#test\n";
  1172. printf("%s", lines);
  1173. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1174. printf("%s", pikaAsm);
  1175. EXPECT_STREQ(pikaAsm,(char *)
  1176. "B0\n"
  1177. "0 REF t\n"
  1178. "0 OUT a\n"
  1179. );
  1180. args_deinit(buffs);
  1181. EXPECT_EQ(pikaMemNow(), 0);
  1182. }
  1183. TEST(parser, annotation_block) {
  1184. pikaMemInfo.heapUsedMax = 0;
  1185. Args* buffs = New_strBuff();
  1186. char* lines = (char*)
  1187. "while True:\n"
  1188. " a = 1\n"
  1189. "\n"
  1190. ;
  1191. printf("%s", lines);
  1192. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1193. printf("%s", pikaAsm);
  1194. EXPECT_STREQ(pikaAsm,(char *)
  1195. "B0\n"
  1196. "0 REF True\n"
  1197. "0 JEZ 2\n"
  1198. "B1\n"
  1199. "0 NUM 1\n"
  1200. "0 OUT a\n"
  1201. "B0\n"
  1202. "0 JMP -1\n"
  1203. "B0\n"
  1204. );
  1205. lines = (char*)
  1206. "while True:\n"
  1207. " a = 1\n"
  1208. "#\n"
  1209. ;
  1210. printf("%s", lines);
  1211. pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1212. printf("%s", pikaAsm);
  1213. EXPECT_STREQ(pikaAsm,(char *)
  1214. "B0\n"
  1215. "0 REF True\n"
  1216. "0 JEZ 2\n"
  1217. "B1\n"
  1218. "0 NUM 1\n"
  1219. "0 OUT a\n"
  1220. );
  1221. lines = (char*)
  1222. "while True:\n"
  1223. " a = 1\n"
  1224. "#test\n"
  1225. " b = 2\n"
  1226. "\n"
  1227. ;
  1228. printf("%s", lines);
  1229. pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1230. printf("%s", pikaAsm);
  1231. EXPECT_STREQ(pikaAsm,(char *)
  1232. "B0\n"
  1233. "0 REF True\n"
  1234. "0 JEZ 2\n"
  1235. "B1\n"
  1236. "0 NUM 1\n"
  1237. "0 OUT a\n"
  1238. "B1\n"
  1239. "0 NUM 2\n"
  1240. "0 OUT b\n"
  1241. "B0\n"
  1242. "0 JMP -1\n"
  1243. "B0\n"
  1244. );
  1245. args_deinit(buffs);
  1246. EXPECT_EQ(pikaMemNow(), 0);
  1247. }
  1248. TEST(parser, if_elif_else) {
  1249. pikaMemInfo.heapUsedMax = 0;
  1250. Args* buffs = New_strBuff();
  1251. char* lines = (char*)
  1252. "if a > 1:\n"
  1253. " b = 1\n"
  1254. "elif a > 2:\n"
  1255. " b = 2\n"
  1256. " if a > 1:\n"
  1257. " b = 1\n"
  1258. " elif a > 2:\n"
  1259. " b = 2\n"
  1260. " else:\n"
  1261. " b = 3\n"
  1262. "else:\n"
  1263. " b = 3\n"
  1264. "\n"
  1265. ;
  1266. printf("%s", lines);
  1267. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1268. printf("%s", pikaAsm);
  1269. EXPECT_STREQ(pikaAsm,(char *)
  1270. "B0\n"
  1271. "1 REF a\n"
  1272. "1 NUM 1\n"
  1273. "0 OPT >\n"
  1274. "0 JEZ 1\n"
  1275. "B1\n"
  1276. "0 NUM 1\n"
  1277. "0 OUT b\n"
  1278. "B0\n"
  1279. "0 NEL 1\n"
  1280. "1 REF a\n"
  1281. "1 NUM 2\n"
  1282. "0 OPT >\n"
  1283. "0 JEZ 1\n"
  1284. "B1\n"
  1285. "0 NUM 2\n"
  1286. "0 OUT b\n"
  1287. "B1\n"
  1288. "1 REF a\n"
  1289. "1 NUM 1\n"
  1290. "0 OPT >\n"
  1291. "0 JEZ 1\n"
  1292. "B2\n"
  1293. "0 NUM 1\n"
  1294. "0 OUT b\n"
  1295. "B1\n"
  1296. "0 NEL 1\n"
  1297. "1 REF a\n"
  1298. "1 NUM 2\n"
  1299. "0 OPT >\n"
  1300. "0 JEZ 1\n"
  1301. "B2\n"
  1302. "0 NUM 2\n"
  1303. "0 OUT b\n"
  1304. "B1\n"
  1305. "0 NEL 1\n"
  1306. "B2\n"
  1307. "0 NUM 3\n"
  1308. "0 OUT b\n"
  1309. "B0\n"
  1310. "0 NEL 1\n"
  1311. "B1\n"
  1312. "0 NUM 3\n"
  1313. "0 OUT b\n"
  1314. "B0\n"
  1315. );
  1316. args_deinit(buffs);
  1317. EXPECT_EQ(pikaMemNow(), 0);
  1318. }
  1319. TEST(parser, for_range) {
  1320. pikaMemInfo.heapUsedMax = 0;
  1321. Args* buffs = New_strBuff();
  1322. char* lines = (char*)
  1323. "for i in range(0,10):\n"
  1324. " print(i)\n"
  1325. "\n";
  1326. printf("%s", lines);
  1327. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1328. EXPECT_STREQ(pikaAsm,
  1329. "B0\n"
  1330. "2 NUM 0\n"
  1331. "2 NUM 10\n"
  1332. "1 RUN range\n"
  1333. "0 RUN iter\n"
  1334. "0 OUT _l0\n"
  1335. "0 REF _r1\n"
  1336. "0 REF _r2\n"
  1337. "0 REF _r3\n"
  1338. "0 OUT _l0.a1\n"
  1339. "0 OUT _l0.a2\n"
  1340. "0 OUT _l0.a3\n"
  1341. "B0\n"
  1342. "0 RUN _l0.__next__\n"
  1343. "0 OUT i\n"
  1344. "0 EST i\n"
  1345. "0 JEZ 2\n"
  1346. "B1\n"
  1347. "1 REF i\n"
  1348. "0 RUN print\n"
  1349. "B0\n"
  1350. "0 JMP -1\n"
  1351. "B0\n"
  1352. "0 DEL _l0\n"
  1353. "B0\n");
  1354. args_deinit(buffs);
  1355. EXPECT_EQ(pikaMemNow(), 0);
  1356. }
  1357. TEST(parser, for_range_rtt) {
  1358. pikaMemInfo.heapUsedMax = 0;
  1359. Args* buffs = New_strBuff();
  1360. char* lines = (char*)
  1361. "for i in range(0, 3):\n"
  1362. " print('hello PikaScript!')\n"
  1363. " thread.mdelay(500)\n"
  1364. "\n";
  1365. printf("%s", lines);
  1366. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1367. printf("%s", pikaAsm);
  1368. EXPECT_STREQ(pikaAsm,
  1369. "B0\n"
  1370. "2 NUM 0\n"
  1371. "2 NUM 3\n"
  1372. "1 RUN range\n"
  1373. "0 RUN iter\n"
  1374. "0 OUT _l0\n"
  1375. "0 REF _r1\n"
  1376. "0 REF _r2\n"
  1377. "0 REF _r3\n"
  1378. "0 OUT _l0.a1\n"
  1379. "0 OUT _l0.a2\n"
  1380. "0 OUT _l0.a3\n"
  1381. "B0\n"
  1382. "0 RUN _l0.__next__\n"
  1383. "0 OUT i\n"
  1384. "0 EST i\n"
  1385. "0 JEZ 2\n"
  1386. "B1\n"
  1387. "1 STR hello PikaScript!\n"
  1388. "0 RUN print\n"
  1389. "B1\n"
  1390. "1 NUM 500\n"
  1391. "0 RUN thread.mdelay\n"
  1392. "B0\n"
  1393. "0 JMP -1\n"
  1394. "B0\n"
  1395. "0 DEL _l0\n"
  1396. "B0\n");
  1397. args_deinit(buffs);
  1398. EXPECT_EQ(pikaMemNow(), 0);
  1399. }
  1400. TEST(parser, for_list) {
  1401. Args* bf = New_strBuff();
  1402. Stack* bs = New_Stack();
  1403. char* s = strsCopy(bf, (char*)"");
  1404. s = parse("for arg in xrange(0,10):", bf, s, bs);
  1405. s = parse(" print(arg)", bf, s, bs);
  1406. s = parse("", bf, s, bs);
  1407. printf("%s", s);
  1408. EXPECT_STREQ(s,
  1409. "B0\n"
  1410. "2 NUM 0\n"
  1411. "2 NUM 10\n"
  1412. "1 RUN xrange\n"
  1413. "0 RUN iter\n"
  1414. "0 OUT _l0\n"
  1415. "B0\n"
  1416. "0 RUN _l0.__next__\n"
  1417. "0 OUT arg\n"
  1418. "0 EST arg\n"
  1419. "0 JEZ 2\n"
  1420. "B1\n"
  1421. "1 REF arg\n"
  1422. "0 RUN print\n"
  1423. "B0\n"
  1424. "0 JMP -1\n"
  1425. "B0\n"
  1426. "0 DEL _l0\n"
  1427. "B0\n");
  1428. stack_deinit(bs);
  1429. args_deinit(bf);
  1430. EXPECT_EQ(pikaMemNow(), 0);
  1431. }
  1432. TEST(parser, for_for_range) {
  1433. pikaMemInfo.heapUsedMax = 0;
  1434. Args* buffs = New_strBuff();
  1435. char* lines = (char*)
  1436. "a = 0\n"
  1437. "for i in range(0, 10):\n"
  1438. " for k in range(0, 3):\n"
  1439. " print(k)\n"
  1440. " a = a + k\n"
  1441. "\n"
  1442. ;
  1443. printf("%s", lines);
  1444. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1445. printf("%s", pikaAsm);
  1446. EXPECT_STREQ(pikaAsm,(char *)
  1447. "B0\n"
  1448. "0 NUM 0\n"
  1449. "0 OUT a\n"
  1450. "B0\n"
  1451. "2 NUM 0\n"
  1452. "2 NUM 10\n"
  1453. "1 RUN range\n"
  1454. "0 RUN iter\n"
  1455. "0 OUT _l0\n"
  1456. "0 REF _r1\n"
  1457. "0 REF _r2\n"
  1458. "0 REF _r3\n"
  1459. "0 OUT _l0.a1\n"
  1460. "0 OUT _l0.a2\n"
  1461. "0 OUT _l0.a3\n"
  1462. "B0\n"
  1463. "0 RUN _l0.__next__\n"
  1464. "0 OUT i\n"
  1465. "0 EST i\n"
  1466. "0 JEZ 2\n"
  1467. "B1\n"
  1468. "2 NUM 0\n"
  1469. "2 NUM 3\n"
  1470. "1 RUN range\n"
  1471. "0 RUN iter\n"
  1472. "0 OUT _l1\n"
  1473. "0 REF _r1\n"
  1474. "0 REF _r2\n"
  1475. "0 REF _r3\n"
  1476. "0 OUT _l1.a1\n"
  1477. "0 OUT _l1.a2\n"
  1478. "0 OUT _l1.a3\n"
  1479. "B1\n"
  1480. "0 RUN _l1.__next__\n"
  1481. "0 OUT k\n"
  1482. "0 EST k\n"
  1483. "0 JEZ 2\n"
  1484. "B2\n"
  1485. "1 REF k\n"
  1486. "0 RUN print\n"
  1487. "B2\n"
  1488. "1 REF a\n"
  1489. "1 REF k\n"
  1490. "0 OPT +\n"
  1491. "0 OUT a\n"
  1492. "B1\n"
  1493. "0 JMP -1\n"
  1494. "B1\n"
  1495. "0 DEL _l1\n"
  1496. "B0\n"
  1497. "0 JMP -1\n"
  1498. "B0\n"
  1499. "0 DEL _l0\n"
  1500. "B0\n"
  1501. );
  1502. args_deinit(buffs);
  1503. EXPECT_EQ(pikaMemNow(), 0);
  1504. }
  1505. TEST(parser, break_) {
  1506. pikaMemInfo.heapUsedMax = 0;
  1507. Args* buffs = New_strBuff();
  1508. char* lines = (char*)
  1509. "for i in range(0, 3):\n"
  1510. " if i = 1:\n"
  1511. " break\n"
  1512. " break \n"
  1513. " continue\n"
  1514. " continue \n"
  1515. "\n";
  1516. printf("%s", lines);
  1517. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1518. printf("%s", pikaAsm);
  1519. EXPECT_STREQ(pikaAsm,
  1520. "B0\n"
  1521. "2 NUM 0\n"
  1522. "2 NUM 3\n"
  1523. "1 RUN range\n"
  1524. "0 RUN iter\n"
  1525. "0 OUT _l0\n"
  1526. "0 REF _r1\n"
  1527. "0 REF _r2\n"
  1528. "0 REF _r3\n"
  1529. "0 OUT _l0.a1\n"
  1530. "0 OUT _l0.a2\n"
  1531. "0 OUT _l0.a3\n"
  1532. "B0\n"
  1533. "0 RUN _l0.__next__\n"
  1534. "0 OUT i\n"
  1535. "0 EST i\n"
  1536. "0 JEZ 2\n"
  1537. "B1\n"
  1538. "0 NUM 1\n"
  1539. "0 OUT i\n"
  1540. "0 JEZ 1\n"
  1541. "B2\n"
  1542. "0 BRK\n"
  1543. "B2\n"
  1544. "0 BRK\n"
  1545. "B2\n"
  1546. "0 CTN\n"
  1547. "B2\n"
  1548. "0 CTN\n"
  1549. "B0\n"
  1550. "0 JMP -1\n"
  1551. "B0\n"
  1552. "0 DEL _l0\n"
  1553. "B0\n");
  1554. args_deinit(buffs);
  1555. EXPECT_EQ(pikaMemNow(), 0);
  1556. }
  1557. TEST(parser, prime_100) {
  1558. pikaMemInfo.heapUsedMax = 0;
  1559. Args* buffs = New_strBuff();
  1560. char* lines = (char*)
  1561. "num = 0\n"
  1562. "i = 2\n"
  1563. "for i in range(2,100):\n"
  1564. " j=2\n"
  1565. " is_prime = 1\n"
  1566. " for j in range(2 , i):\n"
  1567. " if i%j==0:\n"
  1568. " is_prime = 0\n"
  1569. " break\n"
  1570. " if is_prime:\n"
  1571. " num = num + i\n"
  1572. "\n";
  1573. printf("%s", lines);
  1574. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1575. printf("%s", pikaAsm);
  1576. EXPECT_STREQ(pikaAsm,
  1577. "B0\n"
  1578. "0 NUM 0\n"
  1579. "0 OUT num\n"
  1580. "B0\n"
  1581. "0 NUM 2\n"
  1582. "0 OUT i\n"
  1583. "B0\n"
  1584. "2 NUM 2\n"
  1585. "2 NUM 100\n"
  1586. "1 RUN range\n"
  1587. "0 RUN iter\n"
  1588. "0 OUT _l0\n"
  1589. "0 REF _r1\n"
  1590. "0 REF _r2\n"
  1591. "0 REF _r3\n"
  1592. "0 OUT _l0.a1\n"
  1593. "0 OUT _l0.a2\n"
  1594. "0 OUT _l0.a3\n"
  1595. "B0\n"
  1596. "0 RUN _l0.__next__\n"
  1597. "0 OUT i\n"
  1598. "0 EST i\n"
  1599. "0 JEZ 2\n"
  1600. "B1\n"
  1601. "0 NUM 2\n"
  1602. "0 OUT j\n"
  1603. "B1\n"
  1604. "0 NUM 1\n"
  1605. "0 OUT is_prime\n"
  1606. "B1\n"
  1607. "2 NUM 2\n"
  1608. "2 REF i\n"
  1609. "1 RUN range\n"
  1610. "0 RUN iter\n"
  1611. "0 OUT _l1\n"
  1612. "0 REF _r1\n"
  1613. "0 REF _r2\n"
  1614. "0 REF _r3\n"
  1615. "0 OUT _l1.a1\n"
  1616. "0 OUT _l1.a2\n"
  1617. "0 OUT _l1.a3\n"
  1618. "B1\n"
  1619. "0 RUN _l1.__next__\n"
  1620. "0 OUT j\n"
  1621. "0 EST j\n"
  1622. "0 JEZ 2\n"
  1623. "B2\n"
  1624. "2 REF i\n"
  1625. "2 REF j\n"
  1626. "1 OPT %\n"
  1627. "1 NUM 0\n"
  1628. "0 OPT ==\n"
  1629. "0 JEZ 1\n"
  1630. "B3\n"
  1631. "0 NUM 0\n"
  1632. "0 OUT is_prime\n"
  1633. "B3\n"
  1634. "0 BRK\n"
  1635. "B1\n"
  1636. "0 JMP -1\n"
  1637. "B1\n"
  1638. "0 DEL _l1\n"
  1639. "B1\n"
  1640. "0 REF is_prime\n"
  1641. "0 JEZ 1\n"
  1642. "B2\n"
  1643. "1 REF num\n"
  1644. "1 REF i\n"
  1645. "0 OPT +\n"
  1646. "0 OUT num\n"
  1647. "B0\n"
  1648. "0 JMP -1\n"
  1649. "B0\n"
  1650. "0 DEL _l0\n"
  1651. "B0\n");
  1652. args_deinit(buffs);
  1653. EXPECT_EQ(pikaMemNow(), 0);
  1654. }
  1655. TEST(parser, __iter__) {
  1656. pikaMemInfo.heapUsedMax = 0;
  1657. Args* buffs = New_strBuff();
  1658. char* lines = (char*)"__res = __iter__()\n";
  1659. printf("%s", lines);
  1660. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1661. printf("%s", pikaAsm);
  1662. EXPECT_STREQ(pikaAsm,
  1663. "B0\n"
  1664. "0 RUN __iter__\n"
  1665. "0 OUT __res\n");
  1666. args_deinit(buffs);
  1667. EXPECT_EQ(pikaMemNow(), 0);
  1668. }
  1669. /* not solve yet */
  1670. // TEST(parser, strEqu) {
  1671. // pikaMemInfo.heapUsedMax = 0;
  1672. // Args* buffs = New_strBuff();
  1673. // char* lines = (char*)
  1674. // "a = '1ee' == '1ee'\n"
  1675. // "b = '1e' == '1ee'\n"
  1676. // "\n";
  1677. // printf("%s", lines);
  1678. // char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1679. // printf("%s", pikaAsm);
  1680. // EXPECT_STREQ(pikaAsm,
  1681. // "a = ('1ee' == '1ee')\n"
  1682. // "b = ('1e' == '1ee')\n"
  1683. // "\n");
  1684. // args_deinit(buffs);
  1685. // EXPECT_EQ(pikaMemNow(), 0);
  1686. // }
  1687. TEST(parser, for_in_string) {
  1688. pikaMemInfo.heapUsedMax = 0;
  1689. Args* buffs = New_strBuff();
  1690. char* lines = (char*)
  1691. "for b in 'test':\n"
  1692. " a = 1\n"
  1693. "\n";
  1694. printf("%s", lines);
  1695. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1696. printf("%s", pikaAsm);
  1697. EXPECT_STREQ(pikaAsm,
  1698. "B0\n"
  1699. "1 STR test\n"
  1700. "0 RUN iter\n"
  1701. "0 OUT _l0\n"
  1702. "B0\n"
  1703. "0 RUN _l0.__next__\n"
  1704. "0 OUT b\n"
  1705. "0 EST b\n"
  1706. "0 JEZ 2\n"
  1707. "B1\n"
  1708. "0 NUM 1\n"
  1709. "0 OUT a\n"
  1710. "B0\n"
  1711. "0 JMP -1\n"
  1712. "B0\n"
  1713. "0 DEL _l0\n"
  1714. "B0\n");
  1715. args_deinit(buffs);
  1716. EXPECT_EQ(pikaMemNow(), 0);
  1717. }
  1718. TEST(parser, print_ddd) {
  1719. pikaMemInfo.heapUsedMax = 0;
  1720. Args* buffs = New_strBuff();
  1721. char* lines = (char*)"print(\"[Info]: in Python config...\")\n";
  1722. printf("%s", lines);
  1723. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1724. printf("%s", pikaAsm);
  1725. EXPECT_STREQ(pikaAsm,
  1726. "B0\n"
  1727. "1 STR [Info]: in Python config...\n"
  1728. "0 RUN print\n");
  1729. args_deinit(buffs);
  1730. EXPECT_EQ(pikaMemNow(), 0);
  1731. }
  1732. TEST(parser, __get__3) {
  1733. pikaMemInfo.heapUsedMax = 0;
  1734. Args* buffs = New_strBuff();
  1735. char* lines = (char*)"a = b[c+d] + e[f*j]\n";
  1736. printf("%s", lines);
  1737. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1738. printf("%s", pikaAsm);
  1739. EXPECT_STREQ(pikaAsm,
  1740. "B0\n"
  1741. "2 REF b\n"
  1742. "3 REF c\n"
  1743. "3 REF d\n"
  1744. "2 OPT +\n"
  1745. "1 RUN __get__\n"
  1746. "2 REF e\n"
  1747. "3 REF f\n"
  1748. "3 REF j\n"
  1749. "2 OPT *\n"
  1750. "1 RUN __get__\n"
  1751. "0 OPT +\n"
  1752. "0 OUT a\n");
  1753. args_deinit(buffs);
  1754. EXPECT_EQ(pikaMemNow(), 0);
  1755. }
  1756. TEST(parser, __get__) {
  1757. pikaMemInfo.heapUsedMax = 0;
  1758. Args* buffs = New_strBuff();
  1759. char* lines = (char*)"a = b[c]\n";
  1760. printf("%s", lines);
  1761. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1762. printf("%s", pikaAsm);
  1763. EXPECT_STREQ(pikaAsm,
  1764. "B0\n"
  1765. "1 REF b\n"
  1766. "1 REF c\n"
  1767. "0 RUN __get__\n"
  1768. "0 OUT a\n");
  1769. args_deinit(buffs);
  1770. EXPECT_EQ(pikaMemNow(), 0);
  1771. }
  1772. TEST(parser, __get__2) {
  1773. pikaMemInfo.heapUsedMax = 0;
  1774. Args* buffs = New_strBuff();
  1775. char* lines = (char*)"a = b[c+d]\n";
  1776. printf("%s", lines);
  1777. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1778. printf("%s", pikaAsm);
  1779. EXPECT_STREQ(pikaAsm,
  1780. "B0\n"
  1781. "1 REF b\n"
  1782. "2 REF c\n"
  1783. "2 REF d\n"
  1784. "1 OPT +\n"
  1785. "0 RUN __get__\n"
  1786. "0 OUT a\n");
  1787. args_deinit(buffs);
  1788. EXPECT_EQ(pikaMemNow(), 0);
  1789. }
  1790. TEST(parser, __set__) {
  1791. pikaMemInfo.heapUsedMax = 0;
  1792. Args* buffs = New_strBuff();
  1793. char* lines = (char*)"a[b] = c\n";
  1794. printf("%s", lines);
  1795. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1796. printf("%s", pikaAsm);
  1797. EXPECT_STREQ(pikaAsm,
  1798. "B0\n"
  1799. "1 REF a\n"
  1800. "1 REF b\n"
  1801. "1 REF c\n"
  1802. "1 STR a\n"
  1803. "0 RUN __set__\n");
  1804. args_deinit(buffs);
  1805. EXPECT_EQ(pikaMemNow(), 0);
  1806. }
  1807. TEST(parser, str_p_str) {
  1808. pikaMemInfo.heapUsedMax = 0;
  1809. Args* buffs = New_strBuff();
  1810. char* lines = (char*)"a = str(1) + str(1)\n";
  1811. printf("%s", lines);
  1812. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1813. printf("%s", pikaAsm);
  1814. EXPECT_STREQ(pikaAsm,(char*)
  1815. "B0\n"
  1816. "2 NUM 1\n"
  1817. "1 RUN str\n"
  1818. "2 NUM 1\n"
  1819. "1 RUN str\n"
  1820. "0 OPT +\n"
  1821. "0 OUT a\n");
  1822. args_deinit(buffs);
  1823. EXPECT_EQ(pikaMemNow(), 0);
  1824. }
  1825. TEST(parser, test__) {
  1826. pikaMemInfo.heapUsedMax = 0;
  1827. Args* buffs = New_strBuff();
  1828. char* lines = (char*)
  1829. "len = __calls.len()\n"
  1830. "mode = 'none'\n"
  1831. "info_index = 0\n"
  1832. "for i in range(0, len):\n"
  1833. " if len == 0:\n"
  1834. " break\n"
  1835. " if info_index == 0:\n"
  1836. " mode = __calls[i]\n"
  1837. " info_index = 1\n"
  1838. " elif info_index == 1:\n"
  1839. " if mode == 'always':\n"
  1840. " todo = __calls[i]\n"
  1841. " todo()\n"
  1842. " info_index = 0\n"
  1843. " elif mode == 'when':\n"
  1844. " when = __calls[i]\n"
  1845. " info_index = 2\n"
  1846. " elif mode == 'period_ms':\n"
  1847. " period_ms = __calls[i]\n"
  1848. " info_index = 2\n"
  1849. " elif info_index == 2:\n"
  1850. " if mode == 'when':\n"
  1851. " if when():\n"
  1852. " todo = __calls[i]\n"
  1853. " todo()\n"
  1854. " info_index = 0\n"
  1855. " elif mode == 'period_ms':\n"
  1856. " todo = __calls[i]\n"
  1857. " info_index = 3\n"
  1858. " elif info_index == 3:\n"
  1859. " if mode == 'period_ms':\n"
  1860. " if __tick > __calls[i]:\n"
  1861. " todo()\n"
  1862. " __calls[i] = __tick + period_ms\n"
  1863. " info_index = 0\n"
  1864. "\n";
  1865. printf("%s", lines);
  1866. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1867. printf("%s", pikaAsm);
  1868. args_deinit(buffs);
  1869. EXPECT_EQ(pikaMemNow(), 0);
  1870. }
  1871. TEST(parser, global) {
  1872. pikaMemInfo.heapUsedMax = 0;
  1873. Args* buffs = New_strBuff();
  1874. char* lines = (char*)
  1875. "def test_global():\n"
  1876. " global x\n"
  1877. " global y, z\n"
  1878. "\n"
  1879. ;
  1880. printf("%s", lines);
  1881. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1882. printf("%s", pikaAsm);
  1883. EXPECT_STREQ(pikaAsm, (char *)
  1884. "B0\n"
  1885. "0 DEF test_global()\n"
  1886. "0 JMP 1\n"
  1887. "B1\n"
  1888. "0 GLB x\n"
  1889. "B1\n"
  1890. "0 GLB y,z\n"
  1891. "B1\n"
  1892. "0 RET\n"
  1893. "B0\n"
  1894. );
  1895. args_deinit(buffs);
  1896. EXPECT_EQ(pikaMemNow(), 0);
  1897. }
  1898. TEST(parser, mpy_demo_1) {
  1899. pikaMemInfo.heapUsedMax = 0;
  1900. Args* buffs = New_strBuff();
  1901. char* lines = (char*)"chars = ' .,-:;i+hHM$*#@ '\n";
  1902. printf("%s", lines);
  1903. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1904. printf("%s", pikaAsm);
  1905. EXPECT_STREQ(pikaAsm,
  1906. "B0\n"
  1907. "0 STR .,-:;i+hHM$*#@ \n"
  1908. "0 OUT chars\n");
  1909. args_deinit(buffs);
  1910. EXPECT_EQ(pikaMemNow(), 0);
  1911. }
  1912. TEST(parser, clean_compled_str) {
  1913. pikaMemInfo.heapUsedMax = 0;
  1914. Args* buffs = New_strBuff();
  1915. char* res = strsGetCleanCmd(buffs, (char*)"chars = ' .,-:;i+hHM$*#@ '\n");
  1916. EXPECT_STREQ(res, "chars=' .,-:;i+hHM$*#@ '\n");
  1917. printf("%s", res);
  1918. args_deinit(buffs);
  1919. EXPECT_EQ(pikaMemNow(), 0);
  1920. }
  1921. TEST(parser, class_) {
  1922. pikaMemInfo.heapUsedMax = 0;
  1923. Args* buffs = New_strBuff();
  1924. char* lines = (char*)
  1925. "class Test(PikaStdLib.PikaObj):\n"
  1926. " x = 1\n"
  1927. "\n"
  1928. ;
  1929. printf("%s", lines);
  1930. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1931. printf("%s", pikaAsm);
  1932. EXPECT_STREQ(pikaAsm, (char* )
  1933. "B0\n"
  1934. "0 DEF Test()\n"
  1935. "0 JMP 1\n"
  1936. "B1\n"
  1937. "0 RUN PikaStdLib.PikaObj\n"
  1938. "0 OUT self\n"
  1939. "B1\n"
  1940. "0 RAS self\n"
  1941. "B1\n"
  1942. "0 NUM 1\n"
  1943. "0 OUT x\n"
  1944. "B1\n"
  1945. "0 RAS $origin\n"
  1946. "B1\n"
  1947. "0 NEW self\n"
  1948. "0 RET\n"
  1949. "B0\n"
  1950. );
  1951. args_deinit(buffs);
  1952. EXPECT_EQ(pikaMemNow(), 0);
  1953. }
  1954. TEST(parser, class_def) {
  1955. pikaMemInfo.heapUsedMax = 0;
  1956. Args* buffs = New_strBuff();
  1957. char* lines = (char*)
  1958. "class Test():\n"
  1959. " x = 1\n"
  1960. " def hello():\n"
  1961. " print('hello')\n"
  1962. "\n"
  1963. ;
  1964. printf("%s", lines);
  1965. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  1966. printf("%s", pikaAsm);
  1967. EXPECT_STREQ(pikaAsm, (char*)
  1968. "B0\n"
  1969. "0 DEF Test()\n"
  1970. "0 JMP 1\n"
  1971. "B1\n"
  1972. "0 RUN PikaStdLib.PikaObj\n"
  1973. "0 OUT self\n"
  1974. "B1\n"
  1975. "0 RAS self\n"
  1976. "B1\n"
  1977. "0 NUM 1\n"
  1978. "0 OUT x\n"
  1979. "B1\n"
  1980. "0 DEF hello()\n"
  1981. "0 JMP 1\n"
  1982. "B2\n"
  1983. "1 STR hello\n"
  1984. "0 RUN print\n"
  1985. "B2\n"
  1986. "0 RET\n"
  1987. "B1\n"
  1988. "0 RAS $origin\n"
  1989. "B1\n"
  1990. "0 NEW self\n"
  1991. "0 RET\n"
  1992. "B0\n");
  1993. args_deinit(buffs);
  1994. EXPECT_EQ(pikaMemNow(), 0);
  1995. }
  1996. TEST(parser, nag_a) {
  1997. pikaMemInfo.heapUsedMax = 0;
  1998. Args* buffs = New_strBuff();
  1999. char* lines = (char*)"print(-a)\n";
  2000. printf("%s", lines);
  2001. char* pikaAsm = Parser_multiLineToAsm(buffs, (char*)lines);
  2002. printf("%s", pikaAsm);
  2003. args_deinit(buffs);
  2004. EXPECT_EQ(pikaMemNow(), 0);
  2005. }
  2006. TEST(asmer, asmer_to_instructUnit) {
  2007. char* asm_line = (char*)
  2008. "B2\n"
  2009. "2 NUM 2\n"
  2010. "2 NUM 3\n"
  2011. "1 RUN add\n"
  2012. "0 RUN test.on\n"
  2013. ;
  2014. Args buffs = {0};
  2015. ByteCodeFrame bf;
  2016. ByteCodeFrame_init(&bf);
  2017. ByteCodeFrame_appendFromAsm(&bf, asm_line);
  2018. constPool_print(&(bf.const_pool));
  2019. EXPECT_STREQ(constPool_getNext(&(bf.const_pool)), (char*)"2");
  2020. EXPECT_STREQ(constPool_getNext(&(bf.const_pool)), (char*)"3");
  2021. EXPECT_STREQ(constPool_getNext(&(bf.const_pool)), (char*)"add");
  2022. EXPECT_STREQ(constPool_getNext(&(bf.const_pool)), (char*)"test.on");
  2023. EXPECT_EQ((uintptr_t)constPool_getNext(&(bf.const_pool)), (uintptr_t)NULL);
  2024. ByteCodeFrame_deinit(&bf);
  2025. strsDeinit(&buffs);
  2026. EXPECT_EQ(pikaMemNow(), 0);
  2027. }