{-# LANGUAGE OverloadedStrings #-}
module SequenceFormats.RareAlleleHistogramSpec (spec) where

import SequenceFormats.RareAlleleHistogram (RareAlleleHistogram(..), readHistogram, writeHistogramFile)

import qualified Data.Map as Map
import Test.Hspec

spec :: Spec
spec = do
    testReadHistogram
    testWriteHistogramFile

testReadHistogram :: Spec
testReadHistogram = describe "readHistogram" $
    it "should read correct data" $
        readHistogram "testDat/example.histogram.txt" `shouldReturn` mockHistogramDat

testWriteHistogramFile :: Spec
testWriteHistogramFile = describe "writeHistogramFile" $
    it "should read the correct histogram after writing" $ do
        let fn = "/tmp/histogramWriteTest.txt"
        writeHistogramFile fn mockHistogramDat
        readHistogram fn `shouldReturn`mockHistogramDat

mockHistogramDat :: RareAlleleHistogram
mockHistogramDat = RareAlleleHistogram names nVec 1 10 [] [] 1146826657 counts jnEstimates
  where
    names = ["EUR", "SEA", "SIB", "CHK", "SAM"]
    nVec = [66, 42, 44, 8, 28]
    counts = Map.fromList [
        ([1,0,0,0,0], 773148),
        ([0,1,0,0,0], 527207),
        ([0,0,1,0,0], 368640),
        ([0,0,0,0,1], 213918),
        ([2,0,0,0,0], 158795),
        ([0,2,0,0,0], 73241)]
    jnEstimates = Just $ Map.fromList [
        ([1,0,0,0,0], (6.7e-4, 3.6e-6)),
        ([0,1,0,0,0], (4.5e-4, 2.5e-6)),
        ([0,0,1,0,0], (3.2e-4, 2.4e-6)),
        ([0,0,0,0,1], (1.8e-4, 3.6e-6)),
        ([2,0,0,0,0], (1.3e-4, 1.4e-6)),
        ([0,2,0,0,0], (6.3e-5, 7.3e-7))]