Python assert_array_items_equal示例

示例#1

文件： test_vlen_utf8.py 项目： x2wing/numcodecs

def test_encode_utf8():
    a = np.array([u'foo', None, u'bar'], dtype=object)
    codec = VLenUTF8()
    enc = codec.encode(a)
    dec = codec.decode(enc)
    expect = np.array([u'foo', u'', u'bar'], dtype=object)
    assert_array_items_equal(expect, dec)

示例#2

def test_encode_none():
    a = np.array([b'foo', None, b'bar'], dtype=object)
    codec = VLenBytes()
    enc = codec.encode(a)
    dec = codec.decode(enc)
    expect = np.array([b'foo', b'', b'bar'], dtype=object)
    assert_array_items_equal(expect, dec)

示例#3

文件： test_vlen_array.py 项目： x2wing/numcodecs

def test_encode_none():
    a = np.array([[1, 3], None, [4, 7]], dtype=object)
    codec = VLenArray(int)
    enc = codec.encode(a)
    dec = codec.decode(enc)
    expect = np.array(
        [np.array([1, 3]), np.array([]),
         np.array([4, 7])], dtype=object)
    assert_array_items_equal(expect, dec)