def test_lambda_cut(): x = np.arange(21) - 10 mfx = fuzz.trimf(x, [-2, 3, 5]) # fuzz.lambda_cut test expected = np.r_[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0] result = fuzz.lambda_cut(mfx, 0.33) assert_allclose(expected, result) # fuzz.arglcut test expected = np.r_[10, 11, 12, 13, 14] result = fuzz.arglcut(mfx, 0.33) assert len(result) == 1 assert_allclose(expected, result[0])