view _test_onehotop.py @ 534:eaa5ad4089a1

Another bugfix in pylearn.embeddings.length()
author Joseph Turian <turian@gmail.com>
date Tue, 18 Nov 2008 03:49:37 -0500
parents 844bad76459c
children
line wrap: on
line source

from onehotop import one_hot

import unittest
from theano import compile
from theano import gradient
from theano import function
from theano.tensor import as_tensor

import random
import numpy.random

class T_OneHot(unittest.TestCase):
    def test0(self):
        x = as_tensor([3, 2, 1])
        y = as_tensor(5)
        o = one_hot(x, y)
        f = function([],o)
        self.failUnless(numpy.all(f() == numpy.asarray([[0, 0, 0, 1, 0], [0, 0, 1, 0, 0], [0, 1, 0, 0, 0]])))

if __name__ == '__main__':
    unittest.main()