test

Author: wwwillchen

mesop/component_helpers/helper.py

@@ -387,7 +387,7 @@ def compute_fn_id(fn: Callable[..., Any]) -> str:
     partial_args = (
       f"{args_str}{', ' if args_str and kwargs_str else ''}{kwargs_str}"
     )
-    source_code = f"partial({inspect.getsource(fn.func)}, {partial_args})"
+    source_code = f"partial(<<{inspect.getsource(fn.func)}>>, {partial_args})"
     print("source_code", source_code)
     fn_name = fn.func.__name__
     fn_module = fn.func.__module__

mesop/examples/functools_partial.py

@@ -11,7 +11,7 @@ class State:
 @me.page(path="/functools_partial")
 def main():
   state = me.state(State)
-  me.text(text=f"{state.count}")
+  me.text(text=f"value={state.count}")
   me.button("increment 2*4", on_click=partial(increment_click, 2, amount=4))
   me.button("increment 2*10", on_click=partial(increment_click, 2, amount=10))
 

mesop/tests/e2e/functools_partial_test.ts

@@ -0,0 +1,11 @@
+import {test, expect} from '@playwright/test';
+
+test('functools partial', async ({page}) => {
+  await page.goto('/functools_partial');
+
+  await page.getByRole('button', {name: 'increment 2*4'}).click();
+  await expect(page.getByText('8')).toBeVisible();
+
+  await page.getByRole('button', {name: 'increment 2*10'}).click();
+  await expect(page.getByText('28')).toBeVisible();
+});