def __init__(self, *args, **kwargs):
super(DbTestCase, self).__init__(*args, **kwargs)
self._db_connection_provider = None
self.clock = test_util.FakeClock()
self.sleep = test_util.FakeSleep(self.clock)
self.Retrier = functools.partial(util.Retrier, sleep=self.sleep)
self.tbase = db.TensorBase(db_connection_provider=self.connect,
retrier_factory=self.Retrier)
def __init__(self, *args, **kwargs):
super(LoaderTestCase, self).__init__(*args, **kwargs)
self.clock = test_util.FakeClock()
self.sleep = test_util.FakeSleep(self.clock)
def __init__(self, *args, **kwargs):
super(LogFormatterTest, self).__init__(*args, **kwargs)
self.clock = test_util.FakeClock()
self.stubs = tf.test.StubOutForTesting()
self.formatter = util.LogFormatter()
self.formatter.converter = time.gmtime
def __init__(self, *args, **kwargs):
super(RetrierTest, self).__init__(*args, **kwargs)
self.clock = test_util.FakeClock()
self.sleep = test_util.FakeSleep(self.clock)
self.Retrier = functools.partial(util.Retrier, sleep=self.sleep)
