# Path: studio/torch/summary.py
# class TensorBoardWriter(object):
# class Reporter(object):
#     def __init__(self, logdir):
#     def add(self, step, key, value):
#     def flush(self):
#     def close(self):
#     def __init__(self, log_interval=10, logdir=None, smooth_interval=10):
#     def record(self, step, **kwargs):
#     def add(self, step, key, value):
#     def report(self, stdout=None):
#             def smooth(values):


import unittest
from io import StringIO
from studio.torch import summary



class ReporterTest(unittest.TestCase):

    def test_summary_report(self):
        r = summary.Reporter(log_interval=2, smooth_interval=2)
        out = StringIO()
        r.add(0, 'k', 0.1)

# Path: studio/torch/summary.py
# class TensorBoardWriter(object):
# class Reporter(object):
#     def __init__(self, logdir):
#     def add(self, step, key, value):
#     def flush(self):
#     def close(self):
#     def __init__(self, log_interval=10, logdir=None, smooth_interval=10):
#     def record(self, step, **kwargs):
#     def add(self, step, key, value):
#     def report(self, stdout=None):
#             def smooth(values):


# Path: studio/torch/summary_test.py
import unittest
from io import StringIO
from studio.torch import summary




class ReporterTest(unittest.TestCase):

    def test_summary_report(self):
        r = summary.Reporter(log_interval=2, smooth_interval=2)
        out = StringIO()
        r.add(0, 'k', 0.1)

        r.add(1, 'k', 0.2)