module Ob.Tests.Small (
    tests
  ) where

import Control.Monad (unless)
import Data.List (nub, foldl')
import Prelude hiding (tail)
import Test.Hspec
import qualified Urbit.Ob.Ob as Ob

a = 2 ^ 2 - 1
b = 2 ^ 2
c = a * b

eff j m =
  let v0 = [5, 9, 2, 6, 4, 0, 8, 7, 1, 10, 3, 11]
      v1 = [2, 1, 0, 3, 10, 4, 9, 5, 7, 11, 6, 8]
      v2 = [10, 6, 7, 1, 0, 11, 3, 9, 5, 2, 8, 4]
      v3 = [11, 0, 3, 5, 9, 8, 6, 10, 4, 1, 2, 7]

  in  case j of
        0 -> v0 !! fromIntegral m
        1 -> v1 !! fromIntegral m
        2 -> v2 !! fromIntegral m
        _ -> v3 !! fromIntegral m

feis = Ob.capFe 4 a b c eff

tail = Ob.capFen 4 a b c eff

tests :: Spec
tests = do
  let emm  = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
      perm = fmap feis emm
      inv  = fmap tail perm
      rinv = fmap feis inv
      distincts = nub perm

  describe "feis" $ do
    it "produces distinct elements" $
      length distincts `shouldBe` length perm

    it "permutes successfully" $
      foldl' (\acc x -> x `elem` emm && acc) True perm `shouldBe` True

  describe "feis" $
    it "inverts tail" $
      rinv `shouldBe` perm

  describe "tail" $
    it "inverts feis" $
      emm `shouldBe` inv