def test_MlpManager(self): mng = MlpManager(ns=1) mng.createMlp(self.output, self.factors2, self.output, [10]) assert_array_equal(mng.getMlpTopology(), [2*9 + 2*9, 10, 3]) mng = MlpManager() mng.createMlp(self.output, self.factors, self.output, [10]) assert_array_equal(mng.getMlpTopology(), [2*1 + 1*1, 10, 3])