Exemplo n.º 1
0
def test_fix_numpy_array_equality():
    complex_obj = [(numpy.array([0, 1, 2]), 'thing', []),
                   numpy.array([0, 1]),
                   [numpy.array([0, 1]),
                    numpy.array([0]), [0, 1, 2]]]

    assert helpers.fix_numpy_array_equality(complex_obj) == \
        [(helpers.sane_equality_array([0, 1, 2]), 'thing', []), helpers.sane_equality_array([0, 1]),
         [helpers.sane_equality_array([0, 1]), helpers.sane_equality_array([0]), [0, 1, 2]]]
Exemplo n.º 2
0
def test_SaneEqualityArray():
    assert helpers.sane_equality_array(
        [0, 1, 2]) == helpers.sane_equality_array([0, 1, 2])
    assert helpers.sane_equality_array([0]) == helpers.sane_equality_array([0])

    assert not helpers.sane_equality_array([0]) == helpers.sane_equality_array(
        [0, 1, 2])
    assert not helpers.sane_equality_array(
        [0, 1, 2]) == helpers.sane_equality_array([0])