detectron2/tests/utils/test_tensorboardx.py

import os
import tempfile
import unittest

from detectron2.utils.events import TensorboardXWriter


# TODO Fix up capitalization
class TestTensorboardXWriter(unittest.TestCase):
    def test_no_files_created(self) -> None:
        with tempfile.TemporaryDirectory() as tmp_dir:
            writer = TensorboardXWriter(tmp_dir)
            writer.close()

            self.assertFalse(os.listdir(tmp_dir))

    def test_single_write(self) -> None:
        with tempfile.TemporaryDirectory() as tmp_dir:
            writer = TensorboardXWriter(tmp_dir)
            writer._writer.add_scalar("testing", 1, 1)
            writer.close()

            self.assertTrue(os.listdir(tmp_dir))