def test_on_backward(self): def example(state): return state state = 'test' self.assertTrue( callbacks.on_backward(example).on_backward(state) == state)
def test_multi(self): def example(state): return state state = 'test' c = callbacks.on_backward(callbacks.on_sample(callbacks.on_start(example))) self.assertTrue(c.on_backward(state) == state) self.assertTrue(c.on_sample(state) == state) self.assertTrue(c.on_start(state) == state)
def test_on_backward(self): def example(state): return state state = 'test' self.assertTrue(callbacks.on_backward(example).on_backward(state) == state)