import numpy as py
from perzeptron import *

data = np.mat([[1, 0, 0],[1, 0, 1], [1, 1, 0], [1, 1, 1]])

result_OR = np.array([0, 1, 1, 1])
result_AND = np.array([0, 0, 0, 1])
result_XOR = np.array([0, 1, 1, 0])

t = 0.5
my = 0.1
iter = 1000
eps = 0.01

def isBetween(v,r):
    return v >= r - eps and v <= r + eps

def test_or_perzeptron():
    p = Perzeptron(t, my, result_OR, iter, lineare)
    p.training(data)

    assert isBetween(p.calculate(data[0].T), 0) == True
    assert isBetween(p.calculate(data[1].T), 1) == True
    assert isBetween(p.calculate(data[2].T), 1) == True
    assert isBetween(p.calculate(data[3].T), 1) == True

def test_and_perzeptron():
    p = Perzeptron(t, my, result_AND, iter, lineare)
    p.training(data)

    assert isBetween(p.calculate(data[0].T), 0) == True
    assert isBetween(p.calculate(data[1].T), 0) == True
    assert isBetween(p.calculate(data[2].T), 0) == True
    assert isBetween(p.calculate(data[3].T), 1) == True