test_var_operations.py 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946
  1. """Integration tests for var operations."""
  2. from typing import Generator
  3. import pytest
  4. from selenium.webdriver.common.by import By
  5. from reflex.testing import AppHarness
  6. def VarOperations():
  7. """App with var operations."""
  8. from typing import TypedDict
  9. import reflex as rx
  10. from reflex.vars.base import LiteralVar
  11. from reflex.vars.sequence import ArrayVar
  12. class Object(rx.Base):
  13. name: str = "hello"
  14. class Person(TypedDict):
  15. name: str
  16. age: int
  17. class VarOperationState(rx.State):
  18. int_var1: rx.Field[int] = rx.field(10)
  19. int_var2: rx.Field[int] = rx.field(5)
  20. int_var3: rx.Field[int] = rx.field(7)
  21. float_var1: rx.Field[float] = rx.field(10.5)
  22. float_var2: rx.Field[float] = rx.field(5.5)
  23. list1: rx.Field[list] = rx.field([1, 2])
  24. list2: rx.Field[list] = rx.field([3, 4])
  25. list3: rx.Field[list] = rx.field(["first", "second", "third"])
  26. list4: rx.Field[list] = rx.field([Object(name="obj_1"), Object(name="obj_2")])
  27. optional_list: rx.Field[list | None] = rx.field(None)
  28. optional_dict: rx.Field[dict[str, str] | None] = rx.field(None)
  29. optional_list_value: rx.Field[list[str] | None] = rx.field(["red", "yellow"])
  30. optional_dict_value: rx.Field[dict[str, str] | None] = rx.field({"name": "red"})
  31. str_var1: rx.Field[str] = rx.field("first")
  32. str_var2: rx.Field[str] = rx.field("second")
  33. str_var3: rx.Field[str] = rx.field("ThIrD")
  34. str_var4: rx.Field[str] = rx.field("a long string")
  35. dict1: rx.Field[dict[int, int]] = rx.field({1: 2})
  36. dict2: rx.Field[dict[int, int]] = rx.field({3: 4})
  37. html_str: rx.Field[str] = rx.field("<div>hello</div>")
  38. people: rx.Field[list[Person]] = rx.field(
  39. [{"name": "Alice", "age": 30}, {"name": "Bob", "age": 25}]
  40. )
  41. app = rx.App(_state=rx.State)
  42. @rx.memo
  43. def memo_comp(list1: list[int], int_var1: int, id: str):
  44. return rx.text(list1, int_var1, id=id)
  45. @rx.memo
  46. def memo_comp_nested(int_var2: int, id: str):
  47. return memo_comp(list1=[3, 4], int_var1=int_var2, id=id)
  48. @app.add_page
  49. def index():
  50. return rx.vstack(
  51. rx.el.input(
  52. id="token",
  53. value=VarOperationState.router.session.client_token,
  54. is_read_only=True,
  55. ),
  56. # INT INT
  57. rx.text(
  58. VarOperationState.int_var1 + VarOperationState.int_var2,
  59. id="int_add_int",
  60. ),
  61. rx.text(
  62. VarOperationState.int_var1 * VarOperationState.int_var2,
  63. id="int_mult_int",
  64. ),
  65. rx.text(
  66. VarOperationState.int_var1 - VarOperationState.int_var2,
  67. id="int_sub_int",
  68. ),
  69. rx.text(
  70. VarOperationState.int_var1**VarOperationState.int_var2,
  71. id="int_exp_int",
  72. ),
  73. rx.text(
  74. VarOperationState.int_var1 / VarOperationState.int_var2,
  75. id="int_div_int",
  76. ),
  77. rx.text(
  78. VarOperationState.int_var1 // VarOperationState.int_var3,
  79. id="int_floor_int",
  80. ),
  81. rx.text(
  82. VarOperationState.int_var1 % VarOperationState.int_var2,
  83. id="int_mod_int",
  84. ),
  85. rx.text(
  86. VarOperationState.int_var1 | VarOperationState.int_var2,
  87. id="int_or_int",
  88. ),
  89. rx.text(
  90. (VarOperationState.int_var1 > VarOperationState.int_var2).to_string(),
  91. id="int_gt_int",
  92. ),
  93. rx.text(
  94. (VarOperationState.int_var1 < VarOperationState.int_var2).to_string(),
  95. id="int_lt_int",
  96. ),
  97. rx.text(
  98. (VarOperationState.int_var1 >= VarOperationState.int_var2).to_string(),
  99. id="int_gte_int",
  100. ),
  101. rx.text(
  102. (VarOperationState.int_var1 <= VarOperationState.int_var2).to_string(),
  103. id="int_lte_int",
  104. ),
  105. rx.text(
  106. VarOperationState.int_var1 & VarOperationState.int_var2,
  107. id="int_and_int",
  108. ),
  109. rx.text(
  110. (VarOperationState.int_var1 | VarOperationState.int_var2).to_string(),
  111. id="int_or_int",
  112. ),
  113. rx.text(
  114. (VarOperationState.int_var1 == VarOperationState.int_var2).to_string(),
  115. id="int_eq_int",
  116. ),
  117. rx.text(
  118. (VarOperationState.int_var1 != VarOperationState.int_var2).to_string(),
  119. id="int_neq_int",
  120. ),
  121. # INT FLOAT OR FLOAT INT
  122. rx.text(
  123. VarOperationState.float_var1 + VarOperationState.int_var2,
  124. id="float_add_int",
  125. ),
  126. rx.text(
  127. VarOperationState.float_var1 * VarOperationState.int_var2,
  128. id="float_mult_int",
  129. ),
  130. rx.text(
  131. VarOperationState.float_var1 - VarOperationState.int_var2,
  132. id="float_sub_int",
  133. ),
  134. rx.text(
  135. VarOperationState.float_var1**VarOperationState.int_var2,
  136. id="float_exp_int",
  137. ),
  138. rx.text(
  139. VarOperationState.float_var1 / VarOperationState.int_var2,
  140. id="float_div_int",
  141. ),
  142. rx.text(
  143. VarOperationState.float_var1 // VarOperationState.int_var3,
  144. id="float_floor_int",
  145. ),
  146. rx.text(
  147. VarOperationState.float_var1 % VarOperationState.int_var2,
  148. id="float_mod_int",
  149. ),
  150. rx.text(
  151. (VarOperationState.float_var1 > VarOperationState.int_var2).to_string(),
  152. id="float_gt_int",
  153. ),
  154. rx.text(
  155. (VarOperationState.float_var1 < VarOperationState.int_var2).to_string(),
  156. id="float_lt_int",
  157. ),
  158. rx.text(
  159. (
  160. VarOperationState.float_var1 >= VarOperationState.int_var2
  161. ).to_string(),
  162. id="float_gte_int",
  163. ),
  164. rx.text(
  165. (
  166. VarOperationState.float_var1 <= VarOperationState.int_var2
  167. ).to_string(),
  168. id="float_lte_int",
  169. ),
  170. rx.text(
  171. (
  172. VarOperationState.float_var1 == VarOperationState.int_var2
  173. ).to_string(),
  174. id="float_eq_int",
  175. ),
  176. rx.text(
  177. (
  178. VarOperationState.float_var1 != VarOperationState.int_var2
  179. ).to_string(),
  180. id="float_neq_int",
  181. ),
  182. rx.text(
  183. (VarOperationState.float_var1 & VarOperationState.int_var2).to_string(),
  184. id="float_and_int",
  185. ),
  186. rx.text(
  187. (VarOperationState.float_var1 | VarOperationState.int_var2).to_string(),
  188. id="float_or_int",
  189. ),
  190. # INT, DICT
  191. rx.text(
  192. (VarOperationState.int_var1 | VarOperationState.dict1).to_string(),
  193. id="int_or_dict",
  194. ),
  195. rx.text(
  196. (VarOperationState.int_var1 & VarOperationState.dict1).to_string(),
  197. id="int_and_dict",
  198. ),
  199. rx.text(
  200. (VarOperationState.int_var1 == VarOperationState.dict1).to_string(),
  201. id="int_eq_dict",
  202. ),
  203. rx.text(
  204. (VarOperationState.int_var1 != VarOperationState.dict1).to_string(),
  205. id="int_neq_dict",
  206. ),
  207. # FLOAT FLOAT
  208. rx.text(
  209. VarOperationState.float_var1 + VarOperationState.float_var2,
  210. id="float_add_float",
  211. ),
  212. rx.text(
  213. VarOperationState.float_var1 * VarOperationState.float_var2,
  214. id="float_mult_float",
  215. ),
  216. rx.text(
  217. VarOperationState.float_var1 - VarOperationState.float_var2,
  218. id="float_sub_float",
  219. ),
  220. rx.text(
  221. VarOperationState.float_var1**VarOperationState.float_var2,
  222. id="float_exp_float",
  223. ),
  224. rx.text(
  225. VarOperationState.float_var1 / VarOperationState.float_var2,
  226. id="float_div_float",
  227. ),
  228. rx.text(
  229. VarOperationState.float_var1 // VarOperationState.float_var2,
  230. id="float_floor_float",
  231. ),
  232. rx.text(
  233. VarOperationState.float_var1 % VarOperationState.float_var2,
  234. id="float_mod_float",
  235. ),
  236. rx.text(
  237. (
  238. VarOperationState.float_var1 > VarOperationState.float_var2
  239. ).to_string(),
  240. id="float_gt_float",
  241. ),
  242. rx.text(
  243. (
  244. VarOperationState.float_var1 < VarOperationState.float_var2
  245. ).to_string(),
  246. id="float_lt_float",
  247. ),
  248. rx.text(
  249. (
  250. VarOperationState.float_var1 >= VarOperationState.float_var2
  251. ).to_string(),
  252. id="float_gte_float",
  253. ),
  254. rx.text(
  255. (
  256. VarOperationState.float_var1 <= VarOperationState.float_var2
  257. ).to_string(),
  258. id="float_lte_float",
  259. ),
  260. rx.text(
  261. (
  262. VarOperationState.float_var1 == VarOperationState.float_var2
  263. ).to_string(),
  264. id="float_eq_float",
  265. ),
  266. rx.text(
  267. (
  268. VarOperationState.float_var1 != VarOperationState.float_var2
  269. ).to_string(),
  270. id="float_neq_float",
  271. ),
  272. rx.text(
  273. (
  274. VarOperationState.float_var1 & VarOperationState.float_var2
  275. ).to_string(),
  276. id="float_and_float",
  277. ),
  278. rx.text(
  279. (
  280. VarOperationState.float_var1 | VarOperationState.float_var2
  281. ).to_string(),
  282. id="float_or_float",
  283. ),
  284. # FLOAT STR
  285. rx.text(
  286. VarOperationState.float_var1 | VarOperationState.str_var1,
  287. id="float_or_str",
  288. ),
  289. rx.text(
  290. VarOperationState.float_var1 & VarOperationState.str_var1,
  291. id="float_and_str",
  292. ),
  293. rx.text(
  294. (
  295. VarOperationState.float_var1 == VarOperationState.str_var1
  296. ).to_string(),
  297. id="float_eq_str",
  298. ),
  299. rx.text(
  300. (
  301. VarOperationState.float_var1 != VarOperationState.str_var1
  302. ).to_string(),
  303. id="float_neq_str",
  304. ),
  305. # FLOAT LIST
  306. rx.text(
  307. (VarOperationState.float_var1 | VarOperationState.list1).to_string(),
  308. id="float_or_list",
  309. ),
  310. rx.text(
  311. (VarOperationState.float_var1 & VarOperationState.list1).to_string(),
  312. id="float_and_list",
  313. ),
  314. rx.text(
  315. (VarOperationState.float_var1 == VarOperationState.list1).to_string(),
  316. id="float_eq_list",
  317. ),
  318. rx.text(
  319. (VarOperationState.float_var1 != VarOperationState.list1).to_string(),
  320. id="float_neq_list",
  321. ),
  322. # FLOAT DICT
  323. rx.text(
  324. (VarOperationState.float_var1 | VarOperationState.dict1).to_string(),
  325. id="float_or_dict",
  326. ),
  327. rx.text(
  328. (VarOperationState.float_var1 & VarOperationState.dict1).to_string(),
  329. id="float_and_dict",
  330. ),
  331. rx.text(
  332. (VarOperationState.float_var1 == VarOperationState.dict1).to_string(),
  333. id="float_eq_dict",
  334. ),
  335. rx.text(
  336. (VarOperationState.float_var1 != VarOperationState.dict1).to_string(),
  337. id="float_neq_dict",
  338. ),
  339. # STR STR
  340. rx.text(
  341. VarOperationState.str_var1 + VarOperationState.str_var2,
  342. id="str_add_str",
  343. ),
  344. rx.text(
  345. (VarOperationState.str_var1 > VarOperationState.str_var2).to_string(),
  346. id="str_gt_str",
  347. ),
  348. rx.text(
  349. (VarOperationState.str_var1 < VarOperationState.str_var2).to_string(),
  350. id="str_lt_str",
  351. ),
  352. rx.text(
  353. (VarOperationState.str_var1 >= VarOperationState.str_var2).to_string(),
  354. id="str_gte_str",
  355. ),
  356. rx.text(
  357. (VarOperationState.str_var1 <= VarOperationState.str_var2).to_string(),
  358. id="str_lte_str",
  359. ),
  360. rx.text(
  361. (
  362. VarOperationState.float_var1 == VarOperationState.float_var2
  363. ).to_string(),
  364. id="str_eq_str",
  365. ),
  366. rx.text(
  367. (
  368. VarOperationState.float_var1 != VarOperationState.float_var2
  369. ).to_string(),
  370. id="str_neq_str",
  371. ),
  372. rx.text(
  373. VarOperationState.str_var1.contains("fir").to_string(),
  374. id="str_contains",
  375. ),
  376. rx.text(
  377. VarOperationState.str_var1 | VarOperationState.str_var1,
  378. id="str_or_str",
  379. ),
  380. rx.text(
  381. VarOperationState.str_var1 & VarOperationState.str_var2,
  382. id="str_and_str",
  383. ),
  384. # STR, INT
  385. rx.text(
  386. VarOperationState.str_var1 * VarOperationState.int_var2,
  387. id="str_mult_int",
  388. ),
  389. rx.text(
  390. VarOperationState.str_var1 & VarOperationState.int_var2,
  391. id="str_and_int",
  392. ),
  393. rx.text(
  394. VarOperationState.str_var1 | VarOperationState.int_var2,
  395. id="str_or_int",
  396. ),
  397. rx.text(
  398. (VarOperationState.str_var1 == VarOperationState.int_var1).to_string(),
  399. id="str_eq_int",
  400. ),
  401. rx.text(
  402. (VarOperationState.str_var1 != VarOperationState.int_var1).to_string(),
  403. id="str_neq_int",
  404. ),
  405. # STR, LIST
  406. rx.text(
  407. VarOperationState.str_var1 | VarOperationState.list1,
  408. id="str_or_list",
  409. ),
  410. rx.text(
  411. (VarOperationState.str_var1 & VarOperationState.list1).to_string(),
  412. id="str_and_list",
  413. ),
  414. rx.text(
  415. (VarOperationState.str_var1 == VarOperationState.list1).to_string(),
  416. id="str_eq_list",
  417. ),
  418. rx.text(
  419. (VarOperationState.str_var1 != VarOperationState.list1).to_string(),
  420. id="str_neq_list",
  421. ),
  422. # STR, DICT
  423. rx.text(
  424. VarOperationState.str_var1 | VarOperationState.dict1,
  425. id="str_or_dict",
  426. ),
  427. rx.text(
  428. (VarOperationState.str_var1 & VarOperationState.dict1).to_string(),
  429. id="str_and_dict",
  430. ),
  431. rx.text(
  432. (VarOperationState.str_var1 == VarOperationState.dict1).to_string(),
  433. id="str_eq_dict",
  434. ),
  435. rx.text(
  436. (VarOperationState.str_var1 != VarOperationState.dict1).to_string(),
  437. id="str_neq_dict",
  438. ),
  439. # LIST, LIST
  440. rx.text(
  441. (VarOperationState.list1 + VarOperationState.list2).to_string(),
  442. id="list_add_list",
  443. ),
  444. rx.text(
  445. (VarOperationState.list1 & VarOperationState.list2).to_string(),
  446. id="list_and_list",
  447. ),
  448. rx.text(
  449. (VarOperationState.list1 | VarOperationState.list2).to_string(),
  450. id="list_or_list",
  451. ),
  452. rx.text(
  453. (VarOperationState.list1 > VarOperationState.list2).to_string(),
  454. id="list_gt_list",
  455. ),
  456. rx.text(
  457. (VarOperationState.list1 < VarOperationState.list2).to_string(),
  458. id="list_lt_list",
  459. ),
  460. rx.text(
  461. (VarOperationState.list1 >= VarOperationState.list2).to_string(),
  462. id="list_gte_list",
  463. ),
  464. rx.text(
  465. (VarOperationState.list1 <= VarOperationState.list2).to_string(),
  466. id="list_lte_list",
  467. ),
  468. rx.text(
  469. (VarOperationState.list1 == VarOperationState.list2).to_string(),
  470. id="list_eq_list",
  471. ),
  472. rx.text(
  473. (VarOperationState.list1 != VarOperationState.list2).to_string(),
  474. id="list_neq_list",
  475. ),
  476. rx.text(
  477. VarOperationState.list1.contains(1).to_string(),
  478. id="list_contains",
  479. ),
  480. rx.text(VarOperationState.list4.pluck("name").to_string(), id="list_pluck"),
  481. rx.text(VarOperationState.list1.reverse().to_string(), id="list_reverse"),
  482. # LIST, INT
  483. rx.text(
  484. (VarOperationState.list1 * VarOperationState.int_var2).to_string(),
  485. id="list_mult_int",
  486. ),
  487. rx.text(
  488. (VarOperationState.list1 | VarOperationState.int_var1).to_string(),
  489. id="list_or_int",
  490. ),
  491. rx.text(
  492. (VarOperationState.list1 & VarOperationState.int_var1).to_string(),
  493. id="list_and_int",
  494. ),
  495. rx.text(
  496. (VarOperationState.list1 == VarOperationState.int_var1).to_string(),
  497. id="list_eq_int",
  498. ),
  499. rx.text(
  500. (VarOperationState.list1 != VarOperationState.int_var1).to_string(),
  501. id="list_neq_int",
  502. ),
  503. # LIST, DICT
  504. rx.text(
  505. (VarOperationState.list1 | VarOperationState.dict1).to_string(),
  506. id="list_or_dict",
  507. ),
  508. rx.text(
  509. (VarOperationState.list1 & VarOperationState.dict1).to_string(),
  510. id="list_and_dict",
  511. ),
  512. rx.text(
  513. (VarOperationState.list1 == VarOperationState.dict1).to_string(),
  514. id="list_eq_dict",
  515. ),
  516. rx.text(
  517. (VarOperationState.list1 != VarOperationState.dict1).to_string(),
  518. id="list_neq_dict",
  519. ),
  520. # DICT, DICT
  521. rx.text(
  522. (VarOperationState.dict1 | VarOperationState.dict2).to_string(),
  523. id="dict_or_dict",
  524. ),
  525. rx.text(
  526. (VarOperationState.dict1 & VarOperationState.dict2).to_string(),
  527. id="dict_and_dict",
  528. ),
  529. rx.text(
  530. (VarOperationState.dict1 == VarOperationState.dict2).to_string(),
  531. id="dict_eq_dict",
  532. ),
  533. rx.text(
  534. (VarOperationState.dict1 != VarOperationState.dict2).to_string(),
  535. id="dict_neq_dict",
  536. ),
  537. rx.text(
  538. VarOperationState.dict1.contains(1).to_string(),
  539. id="dict_contains",
  540. ),
  541. rx.text(VarOperationState.str_var3.lower(), id="str_lower"),
  542. rx.text(VarOperationState.str_var3.upper(), id="str_upper"),
  543. rx.text(VarOperationState.str_var4.split(" ").to_string(), id="str_split"),
  544. rx.text(VarOperationState.list3.join(""), id="list_join"),
  545. rx.text(VarOperationState.list3.join(","), id="list_join_comma"),
  546. # Index from an op var
  547. rx.text(
  548. VarOperationState.list3[VarOperationState.int_var1 % 3],
  549. id="list_index_mod",
  550. ),
  551. rx.html(
  552. VarOperationState.html_str,
  553. id="html_str",
  554. ),
  555. rx.el.mark("second"),
  556. rx.text(ArrayVar.range(2, 5).join(","), id="list_join_range1"),
  557. rx.text(ArrayVar.range(2, 10, 2).join(","), id="list_join_range2"),
  558. rx.text(ArrayVar.range(5, 0, -1).join(","), id="list_join_range3"),
  559. rx.text(ArrayVar.range(0, 3).join(","), id="list_join_range4"),
  560. rx.box(
  561. rx.foreach(
  562. ArrayVar.range(0, 2),
  563. lambda x: rx.text(VarOperationState.list1[x], as_="p"),
  564. ),
  565. id="foreach_list_arg",
  566. ),
  567. rx.box(
  568. rx.foreach(
  569. ArrayVar.range(0, 2),
  570. lambda x, ix: rx.text(VarOperationState.list1[ix], as_="p"),
  571. ),
  572. id="foreach_list_ix",
  573. ),
  574. rx.box(
  575. rx.foreach(
  576. LiteralVar.create(list(range(0, 3))).to(ArrayVar, list[int]),
  577. lambda x: rx.foreach(
  578. ArrayVar.range(x),
  579. lambda y: rx.text(VarOperationState.list1[y], as_="p"),
  580. ),
  581. ),
  582. id="foreach_list_nested",
  583. ),
  584. memo_comp(
  585. list1=VarOperationState.list1,
  586. int_var1=VarOperationState.int_var1,
  587. id="memo_comp",
  588. ),
  589. memo_comp_nested(
  590. int_var2=VarOperationState.int_var2,
  591. id="memo_comp_nested",
  592. ),
  593. # foreach in a match
  594. rx.box(
  595. rx.match(
  596. VarOperationState.list3.length(),
  597. (0, rx.text("No choices")),
  598. (1, rx.text("One choice")),
  599. rx.foreach(VarOperationState.list3, lambda choice: rx.text(choice)),
  600. ),
  601. id="foreach_in_match",
  602. ),
  603. # Literal range var in a foreach
  604. rx.box(rx.foreach(range(42, 80, 27), rx.text.span), id="range_in_foreach1"),
  605. rx.box(rx.foreach(range(42, 80, 3), rx.text.span), id="range_in_foreach2"),
  606. rx.box(rx.foreach(range(42, 20, -6), rx.text.span), id="range_in_foreach3"),
  607. rx.box(rx.foreach(range(42, 43, 5), rx.text.span), id="range_in_foreach4"),
  608. # Literal dict in a foreach
  609. rx.box(rx.foreach({"a": 1, "b": 2}, rx.text.span), id="dict_in_foreach1"),
  610. # State Var dict in a foreach
  611. rx.box(
  612. rx.foreach(VarOperationState.dict1, rx.text.span),
  613. id="dict_in_foreach2",
  614. ),
  615. rx.box(
  616. rx.foreach(
  617. VarOperationState.dict1.merge(VarOperationState.dict2),
  618. rx.text.span,
  619. ),
  620. id="dict_in_foreach3",
  621. ),
  622. rx.box(
  623. rx.foreach("abcdef", lambda x: rx.text.span(x + " ")),
  624. id="str_in_foreach",
  625. ),
  626. rx.box(
  627. rx.foreach(VarOperationState.str_var1, lambda x: rx.text.span(x + " ")),
  628. id="str_var_in_foreach",
  629. ),
  630. rx.box(
  631. rx.foreach(
  632. VarOperationState.people,
  633. lambda person: rx.text.span(
  634. "Hello " + person["name"], person["age"] + 3
  635. ),
  636. ),
  637. id="typed_dict_in_foreach",
  638. ),
  639. rx.box(
  640. rx.foreach(VarOperationState.optional_list, rx.text.span),
  641. id="optional_list",
  642. ),
  643. rx.box(
  644. rx.foreach(VarOperationState.optional_dict, rx.text.span),
  645. id="optional_dict",
  646. ),
  647. rx.box(
  648. rx.foreach(VarOperationState.optional_list_value, rx.text.span),
  649. id="optional_list_value",
  650. ),
  651. rx.box(
  652. rx.foreach(VarOperationState.optional_dict_value, rx.text.span),
  653. id="optional_dict_value",
  654. ),
  655. rx.box(
  656. rx.text.span(f"{rx.Var.create(13212312312.1231231)}"),
  657. id="float_format",
  658. ),
  659. rx.box(
  660. rx.text.span(f"{rx.Var.create(13212312312.1231231):.0f}"),
  661. id="float_format_0f",
  662. ),
  663. rx.box(
  664. rx.text.span(f"{rx.Var.create(13212312312.1231231):.1f}"),
  665. id="float_format_1f",
  666. ),
  667. rx.box(
  668. rx.text.span(f"{rx.Var.create(13212312312.1231231):.2f}"),
  669. id="float_format_2f",
  670. ),
  671. rx.box(
  672. rx.text.span(f"{rx.Var.create(13212312312.1231231):,}"),
  673. id="float_format_comma",
  674. ),
  675. rx.box(
  676. rx.text.span(f"{rx.Var.create(13212312312.1231231):_}"),
  677. id="float_format_underscore",
  678. ),
  679. rx.box(
  680. rx.text.span(f"{rx.Var.create(13212312312.1231231):,.0f}"),
  681. id="float_format_comma_0f",
  682. ),
  683. rx.box(
  684. rx.text.span(f"{rx.Var.create(13212312312.1231231):,.1f}"),
  685. id="float_format_comma_1f",
  686. ),
  687. rx.box(
  688. rx.text.span(f"{rx.Var.create(13212312312.1231231):,.2f}"),
  689. id="float_format_comma_2f",
  690. ),
  691. rx.box(
  692. rx.text.span(f"{rx.Var.create(13212312312.1231231):_.0f}"),
  693. id="float_format_underscore_0f",
  694. ),
  695. rx.box(
  696. rx.text.span(f"{rx.Var.create(13212312312.1231231):_.1f}"),
  697. id="float_format_underscore_1f",
  698. ),
  699. rx.box(
  700. rx.text.span(f"{rx.Var.create(13212312312.1231231):_.2f}"),
  701. id="float_format_underscore_2f",
  702. ),
  703. )
  704. @pytest.fixture(scope="module")
  705. def var_operations(tmp_path_factory) -> Generator[AppHarness, None, None]:
  706. """Start VarOperations app at tmp_path via AppHarness.
  707. Args:
  708. tmp_path_factory: pytest tmp_path_factory fixture
  709. Yields:
  710. running AppHarness instance
  711. """
  712. with AppHarness.create(
  713. root=tmp_path_factory.mktemp("var_operations"),
  714. app_source=VarOperations,
  715. ) as harness:
  716. assert harness.app_instance is not None, "app is not running"
  717. yield harness
  718. @pytest.fixture
  719. def driver(var_operations: AppHarness):
  720. """Get an instance of the browser open to the var operations app.
  721. Args:
  722. var_operations: harness for VarOperations app
  723. Yields:
  724. WebDriver instance.
  725. """
  726. driver = var_operations.frontend()
  727. try:
  728. token_input = driver.find_element(By.ID, "token")
  729. assert token_input
  730. # wait for the backend connection to send the token
  731. token = var_operations.poll_for_value(token_input)
  732. assert token is not None
  733. yield driver
  734. finally:
  735. driver.quit()
  736. def test_var_operations(driver, var_operations: AppHarness):
  737. """Test that the var operations produce the right results.
  738. Args:
  739. driver: selenium WebDriver open to the app
  740. var_operations: AppHarness for the var operations app
  741. """
  742. tests = [
  743. # int, int
  744. ("int_add_int", "15"),
  745. ("int_mult_int", "50"),
  746. ("int_sub_int", "5"),
  747. ("int_exp_int", "100000"),
  748. ("int_div_int", "2"),
  749. ("int_floor_int", "1"),
  750. ("int_mod_int", "0"),
  751. ("int_gt_int", "true"),
  752. ("int_lt_int", "false"),
  753. ("int_gte_int", "true"),
  754. ("int_lte_int", "false"),
  755. ("int_and_int", "5"),
  756. ("int_or_int", "10"),
  757. ("int_eq_int", "false"),
  758. ("int_neq_int", "true"),
  759. # int, float
  760. ("float_add_int", "15.5"),
  761. ("float_mult_int", "52.5"),
  762. ("float_sub_int", "5.5"),
  763. ("float_exp_int", "127628.15625"),
  764. ("float_div_int", "2.1"),
  765. ("float_floor_int", "1"),
  766. ("float_mod_int", "0.5"),
  767. ("float_gt_int", "true"),
  768. ("float_lt_int", "false"),
  769. ("float_gte_int", "true"),
  770. ("float_lte_int", "false"),
  771. ("float_eq_int", "false"),
  772. ("float_neq_int", "true"),
  773. ("float_and_int", "5"),
  774. ("float_or_int", "10.5"),
  775. # int, dict
  776. ("int_or_dict", "10"),
  777. ("int_and_dict", '{"1":2}'),
  778. ("int_eq_dict", "false"),
  779. ("int_neq_dict", "true"),
  780. # float, float
  781. ("float_add_float", "16"),
  782. ("float_mult_float", "57.75"),
  783. ("float_sub_float", "5"),
  784. ("float_exp_float", "413562.49323606625"),
  785. ("float_div_float", "1.9090909090909092"),
  786. ("float_floor_float", "1"),
  787. ("float_mod_float", "5"),
  788. ("float_gt_float", "true"),
  789. ("float_lt_float", "false"),
  790. ("float_gte_float", "true"),
  791. ("float_lte_float", "false"),
  792. ("float_eq_float", "false"),
  793. ("float_neq_float", "true"),
  794. ("float_and_float", "5.5"),
  795. ("float_or_float", "10.5"),
  796. # float, str
  797. ("float_or_str", "10.5"),
  798. ("float_and_str", "first"),
  799. ("float_eq_str", "false"),
  800. ("float_neq_str", "true"),
  801. # float, list
  802. ("float_or_list", "10.5"),
  803. ("float_and_list", "[1,2]"),
  804. ("float_eq_list", "false"),
  805. ("float_neq_list", "true"),
  806. # float, dict
  807. ("float_or_dict", "10.5"),
  808. ("float_and_dict", '{"1":2}'),
  809. ("float_eq_dict", "false"),
  810. ("float_neq_dict", "true"),
  811. # str, str
  812. ("str_add_str", "firstsecond"),
  813. ("str_gt_str", "false"),
  814. ("str_lt_str", "true"),
  815. ("str_gte_str", "false"),
  816. ("str_lte_str", "true"),
  817. ("str_eq_str", "false"),
  818. ("str_neq_str", "true"),
  819. ("str_and_str", "second"),
  820. ("str_or_str", "first"),
  821. ("str_contains", "true"),
  822. ("str_lower", "third"),
  823. ("str_upper", "THIRD"),
  824. ("str_split", '["a","long","string"]'),
  825. # str, int
  826. ("str_mult_int", "firstfirstfirstfirstfirst"),
  827. ("str_and_int", "5"),
  828. ("str_or_int", "first"),
  829. ("str_eq_int", "false"),
  830. ("str_neq_int", "true"),
  831. # str, list
  832. ("str_and_list", "[1,2]"),
  833. ("str_or_list", "first"),
  834. ("str_eq_list", "false"),
  835. ("str_neq_list", "true"),
  836. # str, dict
  837. ("str_or_dict", "first"),
  838. ("str_and_dict", '{"1":2}'),
  839. ("str_eq_dict", "false"),
  840. ("str_neq_dict", "true"),
  841. # list, list
  842. ("list_add_list", "[1,2,3,4]"),
  843. ("list_gt_list", "false"),
  844. ("list_lt_list", "true"),
  845. ("list_gte_list", "false"),
  846. ("list_lte_list", "true"),
  847. ("list_eq_list", "false"),
  848. ("list_neq_list", "true"),
  849. ("list_and_list", "[3,4]"),
  850. ("list_or_list", "[1,2]"),
  851. ("list_contains", "true"),
  852. ("list_pluck", '["obj_1","obj_2"]'),
  853. ("list_reverse", "[2,1]"),
  854. ("list_join", "firstsecondthird"),
  855. ("list_join_comma", "first,second,third"),
  856. ("list_join_range1", "2,3,4"),
  857. ("list_join_range2", "2,4,6,8"),
  858. ("list_join_range3", "5,4,3,2,1"),
  859. ("list_join_range4", "0,1,2"),
  860. # list, int
  861. ("list_mult_int", "[1,2,1,2,1,2,1,2,1,2]"),
  862. ("list_or_int", "[1,2]"),
  863. ("list_and_int", "10"),
  864. ("list_eq_int", "false"),
  865. ("list_neq_int", "true"),
  866. # list, dict
  867. ("list_and_dict", '{"1":2}'),
  868. ("list_or_dict", "[1,2]"),
  869. ("list_eq_dict", "false"),
  870. ("list_neq_dict", "true"),
  871. # dict, dict
  872. ("dict_or_dict", '{"1":2}'),
  873. ("dict_and_dict", '{"3":4}'),
  874. ("dict_eq_dict", "false"),
  875. ("dict_neq_dict", "true"),
  876. ("dict_contains", "true"),
  877. # index from an op var
  878. ("list_index_mod", "second"),
  879. # html component with var
  880. ("html_str", "hello"),
  881. # index into list with foreach
  882. ("foreach_list_arg", "1\n2"),
  883. ("foreach_list_ix", "1\n2"),
  884. ("foreach_list_nested", "1\n1\n2"),
  885. # rx.memo component with state
  886. ("memo_comp", "1210"),
  887. ("memo_comp_nested", "345"),
  888. # foreach in a match
  889. ("foreach_in_match", "first\nsecond\nthird"),
  890. # literal range in a foreach
  891. ("range_in_foreach1", "4269"),
  892. ("range_in_foreach2", "42454851545760636669727578"),
  893. ("range_in_foreach3", "42363024"),
  894. ("range_in_foreach4", "42"),
  895. ("dict_in_foreach1", "a1b2"),
  896. ("dict_in_foreach2", "12"),
  897. ("dict_in_foreach3", "1234"),
  898. ("str_in_foreach", "a b c d e f"),
  899. ("str_var_in_foreach", "f i r s t"),
  900. ("typed_dict_in_foreach", "Hello Alice33Hello Bob28"),
  901. # fstring operations
  902. ("float_format", "13212312312.123123"),
  903. ("float_format_0f", "13212312312"),
  904. ("float_format_1f", "13212312312.1"),
  905. ("float_format_2f", "13212312312.12"),
  906. ("float_format_comma", "13,212,312,312.123"),
  907. ("float_format_underscore", "13_212_312_312.123"),
  908. ("float_format_comma_0f", "13,212,312,312"),
  909. ("float_format_comma_1f", "13,212,312,312.1"),
  910. ("float_format_comma_2f", "13,212,312,312.12"),
  911. ("float_format_underscore_0f", "13_212_312_312"),
  912. ("float_format_underscore_1f", "13_212_312_312.1"),
  913. ("float_format_underscore_2f", "13_212_312_312.12"),
  914. ]
  915. for tag, expected in tests:
  916. print(tag)
  917. assert driver.find_element(By.ID, tag).text == expected
  918. # Highlight component with var query (does not plumb ID)
  919. assert driver.find_element(By.TAG_NAME, "mark").text == "second"