add run id test

mozilla-ai · Jan 17, 2024 · e086c41 · e086c41
1 parent 3a40abd
commit e086c41
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/integrations/wandb/test_run_config.py b/tests/integrations/wandb/test_run_config.py
@@ -22,6 +22,11 @@ def test_wandb_path(wandb_run_config):
     assert wandb_run_config.wandb_path == "twitter.com/cortex-research/run-id"
 
 
+def test_ensure_run_id():
+    env = WandbRunConfig(name="defined", project="defined", entity="defined")
+    assert env.run_id is not None  # Pydantic validator fills this in
+
+
 def test_env_vars(wandb_run_config):
     env_vars = wandb_run_config.get_env_vars()
     expected = ["WANDB_NAME", "WANDB_PROJECT", "WANDB_ENTITY", "WANDB_RUN_ID"]