Esempio n. 1
0
def test_sparsearray_append():
    """ Tests that you can append to the sparse array """
    sa = SparseArray(LIST1)
    sa.append(LIST2)

    assert str(sa) == str(LIST1 + LIST2)
    assert len(sa) == 21
Esempio n. 2
0
def test_sparsearray_slice():
    """ Tests that you can get slices from the sparse array """
    sa = SparseArray(LIST1)
    sa.append(LIST2)
    list3 = LIST1 + LIST2

    assert sa[0:4] == list3[0:4]
    assert sa[5:10] == list3[5:10]
    assert sa[10:21] == list3[10:21]
    assert sa[-1::2] == list3[-1::2]
Esempio n. 3
0
def test_sparsearray_del():
    """ Tests that you can delete an element from the sparse array """
    sa = SparseArray(LIST1)
    del sa[4]
    print(sa)

    assert sa[4] == 0
    assert len(sa) == 18
Esempio n. 4
0
def test_sparsearray_set_get_value():
    """ Tests that you can get and set values to the sparse array """
    sa = SparseArray(LIST1)
    sa[1] = 2
    sa[2] = 0

    with pytest.raises(IndexError):
        print(sa[20])
    with pytest.raises(IndexError):
        sa[20] = 10

    sa[0] = 0
    assert sa[0] == 0

    assert sa[1] == 2
    assert sa[2] == 0
Esempio n. 5
0
def test_sparsearray_repr():
    """ Tests that repr(SparseArray) is correctly implemented """
    sa = SparseArray(LIST1)

    print(repr(LIST1))
    assert repr(sa) == repr(LIST1)
Esempio n. 6
0
def test_sparsearray_str():
    """ Tests that str(SparseArray) is correctly implemented """
    sa = SparseArray(LIST1)

    assert str(sa) == str(LIST1)
Esempio n. 7
0
def test_sparsearray_init():
    """ Tests that a sparsearray can be initialized """
    sa = SparseArray(LIST1)
    print(sa.array)
    assert sa.array[0] == sa.array[4] == sa.array[5] == sa.array[
        10] == sa.array[14] == 1
Esempio n. 8
0
def test_sparsearray_len():
    """ Tests that length is returned properly """
    sa = SparseArray(LIST1)
    print(sa)
    assert len(sa) == len(LIST1)
Esempio n. 9
0
def test_append():
    sa = SparseArray([2, 3, 0, 0, 0, 7, 0])
    sa.append(1)
    assert sa[8] == 1
    assert len(sa) == 8
Esempio n. 10
0
def test_set():
    sa = SparseArray([2, 3, 0, 0, 0, 7, 0])
    sa[0] = 1
    assert sa[0] == 1
Esempio n. 11
0
def test_get():
    sa = SparseArray([2, 3, 0, 0, 0, 7, 0])
    assert sa[3] == 0
    assert sa[0] == 2
Esempio n. 12
0
def test_del():
    sa = SparseArray([2, 3, 0])
    del sa[0]
    assert sa.array == {0: 3}
    assert len(sa) == 2
Esempio n. 13
0
def test_length():
    sa = SparseArray([2, 3, 0, 0, 0, 7, 0])
    assert len(sa) == 7
    del sa[1]
    assert len(sa) == 6