from pySVM import *
import sys

SV = SparseVector

v = SV([1.0, 2.0])
w = SV([1.0, 0.0, 2.0])

lk = LinearKernel()
if lk(v,w) != 1.0:
    print "1"
    sys.exit(1)


rk = RadialKernel()
if rk(v,v) != 1.0:
    print "2"
    sys.exit(1)

pk = PolynomialKernel()

if pk(v,v) != 36.0:
    print "3"
    sys.exit(1)



fun = lambda v1,v2: dot(v1,v2)
ppk = PythonKernel(fun)
del fun

if ppk(v,v)-dot(v,v):
    print "4"
    sys.exit(1)