streamlit-nightly 1.31.2.dev20240220__py2.py3-none-any.whl → 1.31.2.dev20240221__py2.py3-none-any.whl

streamlit/testing/v1/app_test.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 from __future__ import annotations
 
-import ast
 import hashlib
 import inspect
 import pathlib
@@ -142,12 +141,21 @@ class AppTest:
         dict-like syntax to set ``query_params`` values for the simulated app.
     """
 
-    def __init__(self, script_path: str, *, default_timeout: float):
+    def __init__(
+        self,
+        script_path: str,
+        *,
+        default_timeout: float,
+        args=None,
+        kwargs=None,
+    ):
         self._script_path = script_path
         self.default_timeout = default_timeout
         self.session_state = SafeSessionState(SessionState(), lambda: None)
         self.query_params: dict[str, Any] = {}
         self.secrets: dict[str, Any] = {}
+        self.args = args
+        self.kwargs = kwargs
 
         tree = ElementTree()
         tree._runner = self
@@ -180,17 +188,30 @@ class AppTest:
             executed via ``.run()``.
 
         """
+        return cls._from_string(script, default_timeout=default_timeout)
+
+    @classmethod
+    def _from_string(
+        cls, script: str, *, default_timeout: float = 3, args=None, kwargs=None
+    ) -> AppTest:
         hasher = hashlib.md5(bytes(script, "utf-8"), **HASHLIB_KWARGS)
         script_name = hasher.hexdigest()
 
         path = pathlib.Path(TMP_DIR.name, script_name)
         aligned_script = textwrap.dedent(script)
         path.write_text(aligned_script)
-        return AppTest(str(path), default_timeout=default_timeout)
+        return AppTest(
+            str(path), default_timeout=default_timeout, args=args, kwargs=kwargs
+        )
 
     @classmethod
     def from_function(
-        cls, script: Callable[[], None], *, default_timeout: float = 3
+        cls,
+        script: Callable[..., Any],
+        *,
+        default_timeout: float = 3,
+        args=None,
+        kwargs=None,
     ) -> AppTest:
         """
         Create an instance of ``AppTest`` to simulate an app page defined\
@@ -210,6 +231,12 @@ class AppTest:
             Default time in seconds before a script run is timed out. Can be
             overridden for individual ``.run()`` calls.
 
+        args: tuple
+            An optional tuple of args to pass to the script function.
+
+        kwargs: dict
+            An optional dict of kwargs to pass to the script function.
+
         Returns
         -------
         AppTest
@@ -217,14 +244,12 @@ class AppTest:
             executed via ``.run()``.
 
         """
-        # TODO: Simplify this using `ast.unparse()` once we drop 3.8 support
         source_lines, _ = inspect.getsourcelines(script)
         source = textwrap.dedent("".join(source_lines))
-        module = ast.parse(source)
-        fn_def = module.body[0]
-        body_lines = source_lines[fn_def.lineno :]
-        body = textwrap.dedent("".join(body_lines))
-        return cls.from_string(body, default_timeout=default_timeout)
+        module = source + f"\n{script.__name__}(*__args, **__kwargs)"
+        return cls._from_string(
+            module, default_timeout=default_timeout, args=args, kwargs=kwargs
+        )
 
     @classmethod
     def from_file(cls, script_path: str, *, default_timeout: float = 3) -> AppTest:
@@ -299,7 +324,9 @@ class AppTest:
             new_secrets._secrets = self.secrets
             st.secrets = new_secrets
 
-        script_runner = LocalScriptRunner(self._script_path, self.session_state)
+        script_runner = LocalScriptRunner(
+            self._script_path, self.session_state, args=self.args, kwargs=self.kwargs
+        )
         self._tree = script_runner.run(widget_state, self.query_params, timeout)
         self._tree._runner = self
         # Last event is SHUTDOWN, so the corresponding data includes query string

streamlit/testing/v1/local_script_runner.py

@@ -15,6 +15,7 @@ from __future__ import annotations
 
 import os
 import time
+import types
 from typing import Any
 from urllib import parse
 
@@ -37,6 +38,8 @@ class LocalScriptRunner(ScriptRunner):
         self,
         script_path: str,
         session_state: SafeSessionState,
+        args=None,
+        kwargs=None,
     ):
         """Initializes the ScriptRunner for the given script_path."""
 
@@ -45,6 +48,8 @@ class LocalScriptRunner(ScriptRunner):
         self.forward_msg_queue = ForwardMsgQueue()
         self.script_path = script_path
         self.session_state = session_state
+        self.args = args if args is not None else tuple()
+        self.kwargs = kwargs if kwargs is not None else dict()
 
         super().__init__(
             session_id="test session id",
@@ -135,6 +140,12 @@ class LocalScriptRunner(ScriptRunner):
         # are marked as active.
         runtime.get_instance().media_file_mgr.remove_orphaned_files()
 
+    def _new_module(self, name: str) -> types.ModuleType:
+        module = types.ModuleType(name)
+        module.__dict__["__args"] = self.args
+        module.__dict__["__kwargs"] = self.kwargs
+        return module
+
 
 def require_widgets_deltas(runner: LocalScriptRunner, timeout: float = 3) -> None:
     """Wait for the given ScriptRunner to emit a completion event. If the timeout

{streamlit_nightly-1.31.2.dev20240220.dist-info → streamlit_nightly-1.31.2.dev20240221.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: streamlit-nightly
-Version: 1.31.2.dev20240220
+Version: 1.31.2.dev20240221
 Summary: A faster way to build and share data apps
 Home-page: https://streamlit.io
 Author: Snowflake Inc

{streamlit_nightly-1.31.2.dev20240220.dist-info → streamlit_nightly-1.31.2.dev20240221.dist-info}/RECORD

@@ -301,7 +301,7 @@ streamlit/runtime/scriptrunner/magic_funcs.py,sha256=_npS_w-0riPNr1-dPyOSjqrwTXo
 streamlit/runtime/scriptrunner/script_cache.py,sha256=ZpaB4T50_GYfhMc2dajSMXWCmS3kaUJ_tPHNVt_flBg,2856
 streamlit/runtime/scriptrunner/script_requests.py,sha256=wxmxfScxKAw8Tl1yntpzMityA00qOAs3eb0f89tEbUM,7142
 streamlit/runtime/scriptrunner/script_run_context.py,sha256=aCa_AddUZvBNYs8dth9VJil59B5DZkKq46HWh7i4jb4,8479
-streamlit/runtime/scriptrunner/script_runner.py,sha256=Lbe4WlimiDcLpttS6tthEN3AUyOuZFEtPqoo9kANuEI,26790
+streamlit/runtime/scriptrunner/script_runner.py,sha256=yeON8dVhpAjdRpCK3TNV7nPw5rZ0bqxpRFUZKcKXMaQ,26812
 streamlit/runtime/state/__init__.py,sha256=UpfNfPrWJ6rVdD-qc0IP_bwZ4MrcNUjyU9wEKDK-eWM,1528
 streamlit/runtime/state/common.py,sha256=guvTqKgR02ysyuIvu-mpzry_LRh9oPrdxma2XekvThM,7565
 streamlit/runtime/state/query_params.py,sha256=mONSZ-HpPAt1qlVIatrCI2KuHryYd_b97gg0EjtuP4s,5846
@@ -310,9 +310,9 @@ streamlit/runtime/state/safe_session_state.py,sha256=StGh9V-tm7MCfohgw5FFfwz8tDw
 streamlit/runtime/state/session_state.py,sha256=IQFb-1z6za9dSpeIWfm_rtk0QpG07UfxohvcHh1yTJY,26197
 streamlit/runtime/state/session_state_proxy.py,sha256=k2JJ9ZX7rMYjpzeGy0WZaBKYaWnXxrbJC5jqdnRQ3jI,5119
 streamlit/runtime/state/widgets.py,sha256=dgPawgm1QC_OgH12xWWZ0I1sOvZFx8HrtRW3xwT5qPk,10975
-streamlit/static/asset-manifest.json,sha256=RwICtPgXDO7-y15FFT2xtpMlS9wawaSMX6XbK972ruQ,14227
+streamlit/static/asset-manifest.json,sha256=PTrwfhXJWBf2mrC2prXqC_Xxvtt6Qbmy0iW_cvDxWjA,14227
 streamlit/static/favicon.png,sha256=if5cVgw7azxKOvV5FpGixga7JLn23rfnHcy1CdWI1-E,1019
-streamlit/static/index.html,sha256=9fDWwQi9XCAzNl8ICTbsqsbkui-F7qDK48lEjg8PnuM,891
+streamlit/static/index.html,sha256=GBoagmPd2ISnAZtrffXNq88ZL7rsXw_FzcRAIOveefE,891
 streamlit/static/static/css/2411.8b8f33d6.chunk.css,sha256=4m2lbj1eVFXSaGCRBHZNhqyRz-4Ce9KogjJPxIq6On8,33275
 streamlit/static/static/css/43.e3b876c5.chunk.css,sha256=XExLUUHInaWJp_m8TtBWhQ88SUuxZl6Jnnw5NA6rwI4,2633
 streamlit/static/static/css/6692.65519639.chunk.css,sha256=7yVN3jCqR9w3ighikXanguy3-zs13V_poJHvLT7ojUU,11095
@@ -379,8 +379,8 @@ streamlit/static/static/js/9330.c0dd1723.chunk.js,sha256=GhnPu6wiRJ-Lf8CAPKlEbAk
 streamlit/static/static/js/9336.2d95d840.chunk.js,sha256=6nGdXfPdjgSauDXzdPVqijnQB2ErTgkzwfoinHGt484,13482
 streamlit/static/static/js/9656.8c935274.chunk.js,sha256=3VB6NT0EQErJUX2SYLQpmNAp19sWI45AbNOQLVjcvII,22113
 streamlit/static/static/js/9758.6e6d8662.chunk.js,sha256=3Gtwgkru2xQTl1mCxFjILQ_Ppb8DMKySHY5YVhm9ihY,2119
-streamlit/static/static/js/main.c1c3de9f.js,sha256=Cu9dVBRyhmXGYpOWnZkASqVFaaczlTWaoMZIaEE58ug,4368762
-streamlit/static/static/js/main.c1c3de9f.js.LICENSE.txt,sha256=YTeqT7R6idssTgnyi3gf0tRiq18-LiPeDOpsWcnbi34,3184
+streamlit/static/static/js/main.c1c90505.js,sha256=nFXWWvJe8uy8RileI_wzZ_HTzOBDjfjofOTQEpZPnCY,4368743
+streamlit/static/static/js/main.c1c90505.js.LICENSE.txt,sha256=YTeqT7R6idssTgnyi3gf0tRiq18-LiPeDOpsWcnbi34,3184
 streamlit/static/static/media/KaTeX_AMS-Regular.73ea273a72f4aca30ca5.woff2,sha256=DN04fJWQoan5eUVgAi27WWVKfYbxh6oMgUla1C06cwg,28076
 streamlit/static/static/media/KaTeX_AMS-Regular.853be92419a6c3766b9a.ttf,sha256=aFNIQLz90r_7bw6N60hoTdAefwTqKBMmdXevuQbeHRM,63632
 streamlit/static/static/media/KaTeX_AMS-Regular.d562e886c52f12660a41.woff,sha256=MNqR6EyJP4deJSaJ-uvcWQsocRReitx_mp1NvYzgslE,33516
@@ -473,9 +473,9 @@ streamlit/static/static/media/logo.83ae4f2fb87e38be7cbb8a5d2beb64d2.svg,sha256=_
 streamlit/static/static/media/rocket.b75b17d2b0a063c6cea230d1a9d77f1e.svg,sha256=6ZIRbOQuBNsdJnpfrNMq-6mYpYDN8qJGVsxkZIn8bP8,39315
 streamlit/testing/__init__.py,sha256=Vrf1yVMOcTyhUPnYvsfyeL96Vpd5z8KoSV5ZzTcTQgU,616
 streamlit/testing/v1/__init__.py,sha256=XGxNOq4VfmwlVj9K6vXB8dAH1YbI_8AIsvw_vbzQoNA,690
-streamlit/testing/v1/app_test.py,sha256=9M7hwxFMjM9i6qVWMPTIqQO8F2hlHRkp9-cksQhw8FQ,33625
+streamlit/testing/v1/app_test.py,sha256=CE_BcAhcdR7QaL6lI00_WrIzqxGo9HT_XySbWoKd1xM,34187
 streamlit/testing/v1/element_tree.py,sha256=PxVskkUEoRQqBUO8EDe9owdm7uyePBCr36Gq_8X8OrE,57730
-streamlit/testing/v1/local_script_runner.py,sha256=r4GH2TYuyyTE8QMgkrUvPKXwKX_31XblUZM7X-rXzUQ,5938
+streamlit/testing/v1/local_script_runner.py,sha256=oa00hdDCB0pjVwBR4XGNNxAnOW4YuQCF29lwcHsvTDM,6329
 streamlit/testing/v1/util.py,sha256=IwxXIlGsVNtC1RmtmgdeBHmJYPjK0wiqNL4HMW6IJZI,1791
 streamlit/vendor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 streamlit/vendor/ipython/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -503,9 +503,9 @@ streamlit/web/server/server_util.py,sha256=FptUF-CjFh78VjeTQDi3R78m7E64MDe3wcklg
 streamlit/web/server/stats_request_handler.py,sha256=cL__KbJFIhdhf1Zt6skbLehUqT-jo56x1HARxogZDOI,3680
 streamlit/web/server/upload_file_request_handler.py,sha256=YPDmKWUnaGH9d4QNcMEsY5k1YIz_q-xW1K5fmgHaDzc,4966
 streamlit/web/server/websocket_headers.py,sha256=07SkWLcOxbyldl7UcBzrMKY9ZojypCQACiKoh5FcH7Y,1870
-streamlit_nightly-1.31.2.dev20240220.data/scripts/streamlit.cmd,sha256=ZEYM3vBJSp-k7vwSJ3ba5NzEk9-qHdSeLvGYAAe1mMw,676
-streamlit_nightly-1.31.2.dev20240220.dist-info/METADATA,sha256=tu4Vjkuyt0BBl-wi7ZnmR6A5D9kIH9LyvI_BluXnVXc,8528
-streamlit_nightly-1.31.2.dev20240220.dist-info/WHEEL,sha256=-G_t0oGuE7UD0DrSpVZnq1hHMBV9DD2XkS5v7XpmTnk,110
-streamlit_nightly-1.31.2.dev20240220.dist-info/entry_points.txt,sha256=uNJ4DwGNXEhOK0USwSNanjkYyR-Bk7eYQbJFDrWyOgY,53
-streamlit_nightly-1.31.2.dev20240220.dist-info/top_level.txt,sha256=V3FhKbm7G2LnR0s4SytavrjIPNIhvcsAGXfYHAwtQzw,10
-streamlit_nightly-1.31.2.dev20240220.dist-info/RECORD,,
+streamlit_nightly-1.31.2.dev20240221.data/scripts/streamlit.cmd,sha256=ZEYM3vBJSp-k7vwSJ3ba5NzEk9-qHdSeLvGYAAe1mMw,676
+streamlit_nightly-1.31.2.dev20240221.dist-info/METADATA,sha256=kw1zdzc-CBsaRjLqIPtdVc17NqYSWiHt2CDK0Wy97Ns,8528
+streamlit_nightly-1.31.2.dev20240221.dist-info/WHEEL,sha256=-G_t0oGuE7UD0DrSpVZnq1hHMBV9DD2XkS5v7XpmTnk,110
+streamlit_nightly-1.31.2.dev20240221.dist-info/entry_points.txt,sha256=uNJ4DwGNXEhOK0USwSNanjkYyR-Bk7eYQbJFDrWyOgY,53
+streamlit_nightly-1.31.2.dev20240221.dist-info/top_level.txt,sha256=V3FhKbm7G2LnR0s4SytavrjIPNIhvcsAGXfYHAwtQzw,10
+streamlit_nightly-1.31.2.dev20240221.dist-info/RECORD,,