class TestMatDot(TestCase):
    def setUp(self):
        self.matdot = MatDot(indim=10, dim=15)
        self.data = np.random.random((100, 10))
        self.matdotout = self.matdot.predict(self.data)

    def test_matdot_shapes(self):
        self.assertEqual(self.matdotout.shape, (100, 15))

    def test_matdot_output(self):
        self.assertTrue(np.allclose(self.matdotout, np.dot(self.data, self.matdot.W.d.get_value())))
Exemple #2
0
class TestMatDot(TestCase):
    def setUp(self):
        self.matdot = MatDot(indim=10, dim=15)
        self.data = np.random.random((100, 10))
        self.matdotout = self.matdot.predict(self.data)

    def test_matdot_shapes(self):
        self.assertEqual(self.matdotout.shape, (100, 15))

    def test_matdot_output(self):
        self.assertTrue(
            np.allclose(self.matdotout,
                        np.dot(self.data, self.matdot.W.d.get_value())))