def test_Identity4D (self):
np_a = np.reshape(self.np_a, (2,2,2,3))
dc_a = dc.reshape(self.dc_a, (2,2,2,3))
npr = np_a
dcr = dc.identity(dc_a)
np.testing.assert_allclose(npr.flatten(), np.array(dcr.data()).astype(np.float32),
rtol=1e-3, atol=1e-3)
def test_Identity1D(self):
npr = self.np_a
dcr = dc.identity(self.dc_a)
np.testing.assert_allclose(npr,
np.array(dcr.data()).astype(np.float32),
rtol=1e-3,
atol=1e-3)
