def test_is_equal_incorrect(self): # Tests that a ValueError is raised when called on one 1D array with # another array of different shape and the wrong data type array1 = Array((3, ), 1, 2, 3) array2 = Array((2, ), 2, 3) with pytest.raises(ValueError): array1.is_equal(array2) array1.is_equal("string")
def test_is_equal(arg, other, expected): """ Verify that comparison of arrays evaluates correctly """ my_arr = Array(arg[0], *arg[1]) other_arr = Array(other[0], *other[1]) assert my_arr.is_equal(other_arr).flatten == expected
def test_is_equal_2d_array(self): # Tests two 2D arrays with integers array1 = Array((2, 3), 1, 2, 3, 4, 5, 6) array2 = Array((2, 3), 2, 3, 3, 5, 6, 6) result = array1.is_equal(array2) expected = Array((2, 3), False, False, True, False, False, True) assert result == expected
def test_is_equal_int(self): # Tests one 1D array with integers and a integer as parameter array1 = Array((3, ), 1, 2, 3) number = 2 result = array1.is_equal(number) expected = Array((3, ), False, True, False) assert result == expected
def test_is_equal_array(self): # Tests two 1D arrays with integers array1 = Array((3, ), 1, 2, 3) array2 = Array((3, ), 1, 5, -3) result = array1.is_equal(array2) expected = Array((3, ), True, False, False) assert result == expected