{-# OPTIONS_GHC -fno-warn-incomplete-patterns #-}
{-# LANGUAGE ScopedTypeVariables #-}

module HaskellWorks.Data.Bits.BitWiseSpec (spec) where

import qualified Data.Bits                                 as B
import qualified Data.Vector.Storable                      as DVS
import           Data.Word
import           HaskellWorks.Data.Bits.BitLength
import           HaskellWorks.Data.Bits.PopCount.PopCount0
import           HaskellWorks.Data.Bits.PopCount.PopCount1
import           Test.Hspec
import           Test.QuickCheck

{-# ANN module ("HLint: ignore Redundant do" :: String) #-}

spec :: Spec
spec = describe "HaskellWorks.Data.Bits.BitWiseSpec" $ do
  describe "for popCount0" $ do
    it "for Word8 matches Data.Bits implementation" $ property $
      \(w :: Word8 ) -> popCount0 w == bitLength w - fromIntegral (B.popCount w)
    it "for Word16 matches Data.Bits implementation" $ property $
      \(w :: Word16) -> popCount0 w == bitLength w - fromIntegral (B.popCount w)
    it "for Word32 matches Data.Bits implementation" $ property $
      \(w :: Word32) -> popCount0 w == bitLength w - fromIntegral (B.popCount w)
    it "for Word64 matches Data.Bits implementation" $ property $
      \(w :: Word64) -> popCount0 w == bitLength w - fromIntegral (B.popCount w)
    it "for [Word8] matches Data.Bits implementation" $ property $
      \(w :: [Word8] ) -> popCount0 w == bitLength w - sum (fmap (fromIntegral . B.popCount) w)
    it "for [Word16] matches Data.Bits implementation" $ property $
      \(w :: [Word16]) -> popCount0 w == bitLength w - sum (fmap (fromIntegral . B.popCount) w)
    it "for [Word32] matches Data.Bits implementation" $ property $
      \(w :: [Word32]) -> popCount0 w == bitLength w - sum (fmap (fromIntegral . B.popCount) w)
    it "for [Word64] matches Data.Bits implementation" $ property $
      \(w :: [Word64]) -> popCount0 w == bitLength w - sum (fmap (fromIntegral . B.popCount) w)
    it "for [Word8] matches Data.Bits implementation" $ property $
      \(w :: [Word8] ) -> popCount0 w == popCount0 (DVS.fromList w)
    it "for [Word16] matches Data.Bits implementation" $ property $
      \(w :: [Word16]) -> popCount0 w == popCount0 (DVS.fromList w)
    it "for [Word32] matches Data.Bits implementation" $ property $
      \(w :: [Word32]) -> popCount0 w == popCount0 (DVS.fromList w)
    it "for [Word64] matches Data.Bits implementation" $ property $
      \(w :: [Word64]) -> popCount0 w == popCount0 (DVS.fromList w)
  describe "for popCount1" $ do
    it "for Word8 matches Data.Bits implementation" $ property $
      \(w :: Word8 ) -> popCount1 w == fromIntegral (B.popCount w)
    it "for Word16 matches Data.Bits implementation" $ property $
      \(w :: Word16) -> popCount1 w == fromIntegral (B.popCount w)
    it "for Word32 matches Data.Bits implementation" $ property $
      \(w :: Word32) -> popCount1 w == fromIntegral (B.popCount w)
    it "for Word64 matches Data.Bits implementation" $ property $
      \(w :: Word64) -> popCount1 w == fromIntegral (B.popCount w)
    it "for [Word8] matches Data.Bits implementation" $ property $
      \(w :: [Word8] ) -> popCount1 w == sum (fmap (fromIntegral . B.popCount) w)
    it "for [Word16] matches Data.Bits implementation" $ property $
      \(w :: [Word16]) -> popCount1 w == sum (fmap (fromIntegral . B.popCount) w)
    it "for [Word32] matches Data.Bits implementation" $ property $
      \(w :: [Word32]) -> popCount1 w == sum (fmap (fromIntegral . B.popCount) w)
    it "for [Word64] matches Data.Bits implementation" $ property $
      \(w :: [Word64]) -> popCount1 w == sum (fmap (fromIntegral . B.popCount) w)
    it "for [Word8] matches Data.Bits implementation" $ property $
      \(w :: [Word8] ) -> popCount1 w == popCount1 (DVS.fromList w)
    it "for [Word16] matches Data.Bits implementation" $ property $
      \(w :: [Word16]) -> popCount1 w == popCount1 (DVS.fromList w)
    it "for [Word32] matches Data.Bits implementation" $ property $
      \(w :: [Word32]) -> popCount1 w == popCount1 (DVS.fromList w)
    it "for [Word64] matches Data.Bits implementation" $ property $
      \(w :: [Word64]) -> popCount1 w == popCount1 (DVS.fromList w)