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)