def test_merge_diagonal(self): a = ConnectedRegion(shape=(2, 4), value=1, rowptr=[0, 2, 4], colptr=[0, 1, 1, 2]) b = ConnectedRegion(shape=(2, 4), value=1, rowptr=[0, 2, 4], colptr=[2, 3, 3, 4]) crh.merge(a, b) assert_array_equal(crh.todense(a), [[1, 0, 1, 0], [0, 1, 0, 1]]) assert_equal(crh.nnz(a), 4) a = ConnectedRegion(shape=(2, 2), value=1, rowptr=[0, 2, 4], colptr=[0, 1, 1, 2]) b = ConnectedRegion(shape=(2, 2), value=1, rowptr=[0, 2, 4], colptr=[1, 2, 0, 1]) crh.merge(a, b) assert_array_equal(crh.todense(a), [[1, 1], [1, 1]]) assert_equal(crh.nnz(a), 4)
def test_nnz(self): assert_equal(crh.nnz(self.c), 8)