hedges should work with functions now

Author: amogorkon

META.stay

@@ -3,7 +3,7 @@ name: fuzzylogic
 description: Fuzzy Logic for Python 3
 license: MIT
 url: https://github.com/amogorkon/fuzzylogic
-version: 0.1.1post3
+version: 0.1.1post5
 author: Anselm Kiefner
 author_email: fuzzylogic-pypi@anselm.kiefner.de
 python_requires: >3.6

src/fuzzylogic.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fuzzylogic
-Version: 0.1.1.post3
+Version: 0.1.1.post5
 Summary: Fuzzy Logic for Python 3
 Home-page: https://github.com/amogorkon/fuzzylogic
 Author: Anselm Kiefner

src/fuzzylogic/hedges.py

@@ -2,34 +2,48 @@
 """
 Lingual hedges modify curves of membership values.
 
-These work with sets only. It's more trouble than it is worth making
-these work with pure functions, so meta-functionality was removed.
+These should work with Sets and functions.
 """
 
 from .classes import Set
 
 def very(g):
     """Sharpen memberships so that only the values close 1 stay at the top."""
-    def s_f(g):
+    if isinstance(g, Set):
+        def s_f(g):
+            def f(x):
+                return g(x) ** 2
+            return f
+        return Set(s_f(g.func))
+    else:
         def f(x):
             return g(x) ** 2
         return f
-    return Set(s_f(g.func))
 
 
 def plus(g):
     """Sharpen memberships like 'very' but not as strongly."""
-    def s_f(g):
+    if isinstance(g, Set):
+        def s_f(g):
+            def f(x):
+                return g(x) ** 1.25
+            return f
+        return Set(s_f(g.func))
+    else:
         def f(x):
             return g(x) ** 1.25
         return f
-    return Set(s_f(g.func))
 
 
 def minus(g):
     """Increase membership support so that more values hit the top."""
-    def s_f(g):
+    if isinstance(g, Set):
+        def s_f(g):
+            def f(x):
+                return g(x) ** 0.75
+            return f 
+        return Set(s_f(g.func))
+    else:
         def f(x):
             return g(x) ** 0.75
-        return f 
-    return Set(s_f(g.func))
+        return f