def test_randomize_warn(self):
class _RandomClass(Randomizable):
def randomize(self, foo):
pass
c = Compose([_RandomClass(), _RandomClass()])
with self.assertWarns(Warning):
c.randomize()
def test_randomize_warn(self):
class _RandomClass(RandomizableTransform):
def randomize(self, foo1, foo2):
pass
def __call__(self, data):
pass
c = Compose([_RandomClass(), _RandomClass()])
with self.assertWarns(Warning):
c.randomize()
def test_random_compose(self):
class _Acc(Randomizable):
self.rand = 0.0
def randomize(self, data=None):
self.rand = self.R.rand()
def __call__(self, data):
self.randomize()
return self.rand + data
c = Compose([_Acc(), _Acc()])
self.assertNotAlmostEqual(c(0), c(0))
c.set_random_state(123)
self.assertAlmostEqual(c(1), 1.61381597)
c.set_random_state(223)
c.randomize()
self.assertAlmostEqual(c(1), 1.90734751)
def test_random_compose(self):
class _Acc(Randomizable):
self.rand = 0.0
def randomize(self):
self.rand = self.R.rand()
def __call__(self, data):
self.randomize()
return self.rand + data
c = Compose([_Acc(), _Acc()])
self.assertNotAlmostEqual(c(0), c(0))
c.set_random_state(123)
self.assertAlmostEqual(c(1), 2.39293837)
c.set_random_state(223)
c.randomize()
self.assertAlmostEqual(c(1), 2.57673391)
