tests

Lightning-AI · Aug 5, 2020 · 4af6496 · 4af6496
1 parent cc0a5d1
commit 4af6496
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/loggers/test_csv.py b/tests/loggers/test_csv.py
@@ -4,7 +4,9 @@
 import torch
 import os
 
+from pytorch_lightning.core.saving import load_hparams_from_yaml
 from pytorch_lightning.loggers import CSVLogger
+from pytorch_lightning.loggers.csv_logs import ExperimentWriter
 
 
 def test_file_logger_automatic_versioning(tmpdir):
@@ -68,6 +70,10 @@ def test_file_logger_log_metrics(tmpdir, step_idx):
     logger.log_metrics(metrics, step_idx)
     logger.save()
 
+    path_yaml = os.path.join(logger.log_dir, ExperimentWriter.NAME_HPARAMS_FILE)
+    params = load_hparams_from_yaml(path_yaml)
+    assert all([n in params for n in metrics])
+
 
 def test_file_logger_log_hyperparams(tmpdir):
     logger = CSVLogger(tmpdir)