import tensorflow as tf

class BaseTest:
    def __init__(self, sess, model, data, config, summary_logger):
        self.model = model
        self.summary_logger = summary_logger
        self.config = config
        self.sess = sess
        self.data = data
        self.init = tf.group(tf.global_variables_initializer(), tf.local_variables_initializer())
        self.sess.run(self.init)

    def test(self):
        """
        implement the logic of epoch:
        -loop over the number of iterations in the config and call the train step
        -add any summaries you want using the summary
        """
        raise NotImplementedError
