def test_constant_equiv(): n = 10 constant = 1.5 w_const = FnConstWeighting(constant) w_const2 = FnConstWeighting(constant) const_sparse_mat = sp.sparse.dia_matrix(([constant] * n, [0]), shape=(n, n)) const_dense_mat = constant * np.eye(n) w_matrix_sp = FnMatrixWeighting(const_sparse_mat) w_matrix_de = FnMatrixWeighting(const_dense_mat) # Equal -> True assert w_const.equiv(w_const) assert w_const.equiv(w_const2) # Equivalent matrix representation -> True assert w_const.equiv(w_matrix_sp) assert w_const.equiv(w_matrix_de) w_different_const = FnConstWeighting(2.5) assert not w_const.equiv(w_different_const) # Bogus input assert not w_const.equiv(True) assert not w_const.equiv(object) assert not w_const.equiv(None)