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)