class TestSBN(RWSLayerTest, unittest.TestCase): def setUp(self): self.n_samples = 10 self.layer = SBN( n_X=16, n_Y=8, ) self.layer.setup()