fixup! feat(Window.split_window): set up environment

Author: zappolowski

tests/test_window.py

@@ -318,20 +318,31 @@ def test_empty_window_name(session: Session) -> None:
     has_lt_version("3.0"),
     reason="needs -e flag for split-window which was introduced in 3.0",
 )
-def test_split_window_with_environment(session: Session) -> None:
+@pytest.mark.parametrize(
+    "environment",
+    [
+        {"ENV_VAR": "pane"},
+        {"ENV_VAR_1": "pane_1", "ENV_VAR_2": "pane_2"},
+    ],
+)
+def test_split_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(window_name="split_window_with_environment")
     pane = window.split_window(
         shell=f"{env} PS1='$ ' sh",
-        environment={"ENV_VAR": "pane"},
+        environment=environment,
     )
     assert pane is not None
     # wait a bit for the prompt to be ready as the test gets flaky otherwise
     time.sleep(0.05)
-    pane.send_keys("echo $ENV_VAR")
-    assert pane.capture_pane() == ["$ echo $ENV_VAR", "pane", "$"]
+    for k, v in environment.items():
+        pane.send_keys(f"echo ${k}")
+        assert pane.capture_pane()[-2] == v
 
 
 @pytest.mark.skipif(