class TestNetworkUtil(unittest.TestCase): def setUp(self): self.nu = NetworkUtil(2, 4) def _buildXORData(self): ndu = NetworkDataUtil() return ndu.createXORData() def test_XOR(self): ds = self._buildXORData() self.nu.train(ds) results, _ = self.nu.activate(ds) assert_array_equal(results, [[2, 0, 100], [0, 2, 100]])
def setUp(self): self.nu = NetworkUtil(2, 4)