Python SparseNDArray.sum示例

示例#1

文件： test_sparse.py 项目： Haxine/mars-1

 def testSparseSum(self):
     s1 = SparseNDArray(self.s1)
     v1 = SparseNDArray(self.v1, shape=(3, ))
     self.assertEqual(s1.sum(), self.s1.sum())
     np.testing.assert_array_equal(
         s1.sum(axis=1),
         np.asarray(self.s1.sum(axis=1)).reshape(2))
     np.testing.assert_array_equal(
         s1.sum(axis=0),
         np.asarray(self.s1.sum(axis=0)).reshape(3))
     np.testing.assert_array_equal(v1.sum(), np.asarray(self.v1_data.sum()))

示例#2

文件： test_sparse.py 项目： qinxuye/mars

def test_sparse_sum():
    s1 = SparseNDArray(s1_data)
    v = SparseNDArray(v1, shape=(3, ))
    assert s1.sum() == s1.sum()
    np.testing.assert_array_equal(s1.sum(axis=1),
                                  np.asarray(s1.sum(axis=1)).reshape(2))
    np.testing.assert_array_equal(s1.sum(axis=0),
                                  np.asarray(s1.sum(axis=0)).reshape(3))
    np.testing.assert_array_equal(v.sum(), np.asarray(v1_data.sum()))

示例#3

文件： test_sparse.py 项目： wycharry/mars-1

 def testSparseSum(self):
     s1 = SparseNDArray(self.s1)
     self.assertEqual(s1.sum(), self.s1.sum())
     np.testing.assert_array_equal(
         s1.sum(axis=1),
         np.asarray(self.s1.sum(axis=1)).reshape(2))