Add tests

Signed-off-by: Saurabh <saurabhkoshatwar1996@gmail.com>

Author: saurabhkoshatwar

tests/unit/launcher/test_user_args.py

@@ -6,8 +6,18 @@
 import pytest
 import subprocess
 
+from types import SimpleNamespace
+
 from deepspeed.accelerator import get_accelerator
+from deepspeed.launcher.multinode_runner import MultiNodeRunner
+
+class DummyRunner(MultiNodeRunner):
+    def backend_exists(self):
+        return True
 
+    def get_cmd(self, environment, active_resources):
+        return []
+    
 if not get_accelerator().is_available():
     pytest.skip("only supported in accelerator environments.", allow_module_level=True)
 
@@ -37,6 +47,11 @@ def cmd(user_script_fp, prompt, multi_node):
         cmd = ("deepspeed", "--num_nodes", "1", "--num_gpus", "1", user_script_fp, "--prompt", prompt)
     return cmd
 
+@pytest.fixture
+def dummy_runner():
+    args = SimpleNamespace(user_args=[], user_script="dummy_script.py")
+    return DummyRunner(args, "dummy_world_info")
+
 
 @pytest.mark.parametrize("prompt", [
     '''"I am 6' tall"''', """'I am 72" tall'""", """'"translate English to Romanian: "'""",
@@ -64,3 +79,20 @@ def test_bash_string_args(tmpdir, user_script_fp):
     p = subprocess.Popen(["bash", bash_fp], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     out, err = p.communicate()
     assert "ARG PARSE SUCCESS" in out.decode("utf-8"), f"User args not parsed correctly: {err.decode('utf-8')}"
+
+
+def test_add_export_with_special_characters(dummy_runner):
+    """
+    Values with special characters (e.g., 64(x2)) must be quoted to avoid bash syntax errors.
+    """
+    dummy_runner.add_export("SLURM_JOB_CPUS_PER_NODE", "64(x2)")
+    assert dummy_runner.exports["SLURM_JOB_CPUS_PER_NODE"] == "\"64(x2)\""
+
+
+def test_add_export_no_special_characters(dummy_runner):
+    """
+    Values without special characters should remain unquoted (e.g., PYTHONPATH).
+    This avoids issues where unnecessary quotes break module imports.
+    """
+    dummy_runner.add_export("PYTHONPATH", "/usr/local/lib/python3.9/site-packages")
+    assert dummy_runner.exports["PYTHONPATH"] == "/usr/local/lib/python3.9/site-packages"