IDSIA / sacred

    ex.run()

def test_log_file_writer_as_context_manager_with_exception(ex, tf):

    """ Check that Tensorflow log directory is captured by LogFileWriter context manager"""

    TEST_LOG_DIR = "/tmp/sacred_test"

    @ex.main

    def run_experiment(_run):

        assert _run.info.get("tensorflow", None) is None

        with tf.Session() as s:

            # Capturing the log directory should be done only in scope of the context manager

            try:

                with LogFileWriter(ex):

                    swr = tf.summary.FileWriter(logdir=TEST_LOG_DIR, graph=s.graph)

                    assert swr is not None

                    assert _run.info["tensorflow"]["logdirs"] == [TEST_LOG_DIR]

                    raise ValueError("I want to be raised!")

            except ValueError:

                pass

            # This should not be captured:

            tf.summary.FileWriter("/tmp/whatever", s.graph)

            assert _run.info["tensorflow"]["logdirs"] == [TEST_LOG_DIR]

    ex.run()

# Tests whether logdir is stored into the info dictionary when creating a new FileWriter object,

# but this time on a method of a class

# Tests whether logdir is stored into the info dictionary when creating a new FileWriter object

def test_log_file_writer(ex, tf):

    TEST_LOG_DIR = "/dev/null"

    TEST_LOG_DIR2 = "/tmp/sacred_test"

    @ex.main

    @LogFileWriter(ex)

    def run_experiment(_run):

        assert _run.info.get("tensorflow", None) is None

        with tf.Session() as s:

            with LogFileWriter(ex):

                swr = tf.summary.FileWriter(logdir=TEST_LOG_DIR, graph=s.graph)

            assert swr is not None

            assert _run.info["tensorflow"]["logdirs"] == [TEST_LOG_DIR]

            tf.summary.FileWriter(TEST_LOG_DIR2, s.graph)

            assert _run.info["tensorflow"]["logdirs"] == [TEST_LOG_DIR, TEST_LOG_DIR2]

    ex.run()

def test_log_summary_writer_as_context_manager(ex, tf):