def testArctan2Execution(self): x = tensor(1) # scalar y = arctan2(x, x) self.assertFalse(y.issparse()) result = self.executor.execute_tensor(y, concat=True)[0] np.testing.assert_equal(result, np.arctan2(1, 1)) y = arctan2(0, x) self.assertFalse(y.issparse()) result = self.executor.execute_tensor(y, concat=True)[0] np.testing.assert_equal(result, np.arctan2(0, 1)) raw1 = np.array([[0, 1, 2]]) raw2 = sps.csr_matrix([[0, 1, 0]]) y = arctan2(raw1, raw2) self.assertFalse(y.issparse()) result = self.executor.execute_tensor(y, concat=True)[0] np.testing.assert_equal(result, np.arctan2(raw1, raw2.A)) y = arctan2(raw2, raw2) self.assertTrue(y.issparse()) result = self.executor.execute_tensor(y, concat=True)[0] np.testing.assert_equal(result, np.arctan2(raw2.A, raw2.A)) y = arctan2(0, raw2) self.assertTrue(y.issparse()) result = self.executor.execute_tensor(y, concat=True)[0] np.testing.assert_equal(result, np.arctan2(0, raw2.A))
def test_arctan2_execution(setup): x = tensor(1) # scalar y = arctan2(x, x) assert y.issparse() is False result = y.execute().fetch() np.testing.assert_equal(result, np.arctan2(1, 1)) y = arctan2(0, x) assert y.issparse() is False result = y.execute().fetch() np.testing.assert_equal(result, np.arctan2(0, 1)) raw1 = np.array([[0, 1, 2]]) raw2 = sps.csr_matrix([[0, 1, 0]]) y = arctan2(raw1, raw2) assert y.issparse() is False result = y.execute().fetch() np.testing.assert_equal(result, np.arctan2(raw1, raw2.A)) y = arctan2(raw2, raw2) assert y.issparse() is True result = y.execute().fetch() np.testing.assert_equal(result, np.arctan2(raw2.A, raw2.A)) y = arctan2(0, raw2) assert y.issparse() is True result = y.execute().fetch() np.testing.assert_equal(result, np.arctan2(0, raw2.A))