event_file_inspector_test.py 文件源码

def testSessionLogSummaries(self):
    data = [
        {
            'session_log': tf.SessionLog(status=tf.SessionLog.START),
            'step': 0
        },
        {
            'session_log': tf.SessionLog(status=tf.SessionLog.CHECKPOINT),
            'step': 1
        },
        {
            'session_log': tf.SessionLog(status=tf.SessionLog.CHECKPOINT),
            'step': 2
        },
        {
            'session_log': tf.SessionLog(status=tf.SessionLog.CHECKPOINT),
            'step': 3
        },
        {
            'session_log': tf.SessionLog(status=tf.SessionLog.STOP),
            'step': 4
        },
        {
            'session_log': tf.SessionLog(status=tf.SessionLog.START),
            'step': 5
        },
        {
            'session_log': tf.SessionLog(status=tf.SessionLog.STOP),
            'step': 6
        },
    ]

    self._WriteScalarSummaries(data)
    units = efi.get_inspection_units(self.logdir)
    self.assertEqual(1, len(units))
    printable = efi.get_dict_to_print(units[0].field_to_obs)
    self.assertEqual(printable['sessionlog:start']['steps'], [0, 5])
    self.assertEqual(printable['sessionlog:stop']['steps'], [4, 6])
    self.assertEqual(printable['sessionlog:checkpoint']['num_steps'], 3)