fixup! feat(Session.new_window): set up environment

Author: zappolowski

tests/test_session.py

@@ -264,20 +264,31 @@ def test_cmd_inserts_sesion_id(session: Session) -> None:
     has_lt_version("3.0"),
     reason="needs -e flag for new-window which was introduced in 3.0",
 )
-def test_new_window_with_environment(session: Session) -> None:
+@pytest.mark.parametrize(
+    "environment",
+    [
+        {"ENV_VAR": "window"},
+        {"ENV_VAR_1": "window_1", "ENV_VAR_2": "window_2"},
+    ],
+)
+def test_new_window_with_environment(
+    session: Session,
+    environment: t.Dict[str, str],
+) -> None:
     env = shutil.which("env")
     assert env is not None, "Cannot find usable `env` in PATH."
 
     window = session.new_window(
         attach=True,
         window_name="window_with_environment",
         window_shell=f"{env} PS1='$ ' sh",
-        environment={"ENV_VAR": "window"},
+        environment=environment,
     )
     pane = window.attached_pane
     assert pane is not None
-    pane.send_keys("echo $ENV_VAR")
-    assert pane.capture_pane() == ["$ echo $ENV_VAR", "window", "$"]
+    for k, v in environment.items():
+        pane.send_keys(f"echo ${k}")
+        assert pane.capture_pane()[-2] == v
 
 
 @pytest.mark.skipif(