Fix fromInfinite error

chimera.cabal

@@ -95,6 +95,7 @@ test-suite chimera-test
     build-depends:
         base >=4.5 && <5,
         chimera,
+        infinite-list,
         QuickCheck >=2.10 && <2.15,
         tasty <1.6,
         tasty-hunit <0.11,

src/Data/Chimera/Internal.hs

@@ -572,7 +572,10 @@
   where
     go0 (x :< xs) = G.singleton x : go 0 xs
 
-    go k xs = G.fromListN kk ys : go (k + 1) zs
+    go k xs =
+      if k == bits
+        then []
+        else G.fromListN kk ys : go (k + 1) zs
       where
         kk = 1 `shiftL` k
         (ys, zs) = Inf.splitAt kk xs
@@ -611,7 +614,7 @@
   => v a
   -> Chimera v a
   -> Chimera v a
 prependVector (G.uncons -> Nothing) ch = ch
 prependVector (G.uncons -> Just (pref0, pref)) (Chimera as) =
   Chimera $
     fromListN (bits + 1) $

test/Test.hs

@@ -12,7 +12,9 @@ import Test.Tasty.QuickCheck as QC hiding ((.&.))
 import Data.Bits
 import Data.Foldable
 import Data.Function (fix)
+import qualified Data.List.Infinite as I
 import qualified Data.List as L
+import qualified Data.List.NonEmpty as NE
 import qualified Data.Vector.Generic as G
 
 import Data.Chimera.ContinuousMapping
@@ -158,6 +160,13 @@ chimeraTests = testGroup "Chimera"
         (if fromIntegral jx < length xs then xs !! fromIntegral jx else x) ===
           Ch.index (Ch.fromListWithDef x xs :: UChimera Bool) jx
 
+  , QC.testProperty "fromInfinite" $
+    \x xs ix ->
+      let jx = ix `mod` 65536 in
+        let ys = I.cycle (x NE.:| xs) in
+          (ys I.!! jx) ===
+            Ch.index (Ch.fromInfinite ys :: UChimera Bool) jx
+
   , QC.testProperty "fromVectorWithDef" $
     \x xs ix ->
       let jx = ix `mod` 65536 in