move Small function isntance to Test.QuickCheck.Function

nick8325 · Mar 21, 2024 · 7e01bf9 · 7e01bf9
1 parent 6c4eb24
commit 7e01bf9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/Test/QuickCheck/Function.hs b/src/Test/QuickCheck/Function.hs
@@ -66,6 +66,7 @@ module Test.QuickCheck.Function
 
 import Test.QuickCheck.Arbitrary
 import Test.QuickCheck.Poly
+import Test.QuickCheck.Modifiers (Small(..))
 
 import Control.Applicative
 import Data.Char
@@ -478,6 +479,11 @@ instance Function OrdB where
 instance Function OrdC where
   function = functionMap unOrdC OrdC
 
+-- Instances for modifiers
+
+instance Function a => Function (Small a) where
+  function = functionMap getSmall Small
+
 -- instance Arbitrary
 
 instance (Function a, CoArbitrary a, Arbitrary b) => Arbitrary (a:->b) where

diff --git a/src/Test/QuickCheck/Modifiers.hs b/src/Test/QuickCheck/Modifiers.hs
@@ -78,7 +78,6 @@ module Test.QuickCheck.Modifiers
 import Test.QuickCheck.Gen
 import Test.QuickCheck.Arbitrary
 import Test.QuickCheck.Exception
-import Test.QuickCheck.Function
 
 import Data.List
   ( sort
@@ -388,9 +387,6 @@ instance Integral a => Arbitrary (Small a) where
 instance CoArbitrary a => CoArbitrary (Small a) where
   coarbitrary (Small a) = coarbitrary a
 
-instance Function a => Function (Small a) where
-  function = functionMap getSmall Small
-
 --------------------------------------------------------------------------
 -- | @Shrink2 x@: allows 2 shrinking steps at the same time when shrinking x
 newtype Shrink2 a = Shrink2 {getShrink2 :: a}