def test_eye_2d_n2(self): eye = initializers.Eye() # yapf: disable desired = np.array([[1, 0], [0, 1]], dtype='float32') actual = eye(shape=(2, 2)) # yapf: enable self.assertIsNone(np.testing.assert_array_almost_equal( actual, desired))
def test_eye_dtype_float64(self): eye = initializers.Eye() desired = 'float64' actual = eye(shape=(4, 4), dtype=desired).dtype self.assertEqual(actual, desired)
def test_eye_2d_rectangular(self): with self.assertRaises(ValueError): eye = initializers.Eye() _ = eye(shape=(2, 3))