1#@!#!123s

: / usr / local / lib64 / python3.6 / site-packages / thinc / tests / linear /

Filename : test_sparse_array.py

back

# coding: utf8
from __future__ import unicode_literals

from ...linear.sparse import SparseArray


def test_init():
    sp = SparseArray(10, 40.0)
    assert sp[10] == 40.0
    assert sp[0] == 0.0
    assert sp[11] == 0.0


def test_setitem():
    sp = SparseArray(10, 40.0)
    assert sp[10] == 40.0
    assert sp[0] == 0.0
    assert sp[11] == 0.0
    sp[0] = 12.0
    assert sp[0] == 12.0
    assert sp[10] == 40.0
    sp[0] = 14.0
    assert sp[0] == 14.0
    assert sp[10] == 40.0
    sp[52] = 6.0
    assert sp[0] == 14.0
    assert sp[1] == 0.0
    assert sp[10] == 40.0
    assert sp[52] == 6.0


def test_clone():
    sp1 = SparseArray(10, 40.0)
    sp2 = SparseArray(200, 2.0)
    sp1 << sp2
    assert sp1[200] == 2.0
    assert sp2[200] == 2.0
    sp1[2] = 2.0
    assert sp1[2] == 2.0
    assert 2 not in sp2