module Main (main) where
import System.Exit

import Test.HUnit
import Physics.MultipletCombiner

main :: IO ()
main = do
    counts <- runTestTT tests
    if (errors counts + failures counts == 0)
        then exitSuccess
        else exitFailure


comb1 = TestCase (assertEqual "[1] >< [1], " [[2],[0]] ([1] >< [1]))
comb2 = TestCase (assertEqual "[1,0] >< [0,1], " [[1,1],[0,0]] ([1,0] >< [0,1]))

multi1 = TestCase (assertEqual "n in [2], " 3 (multi [2]))
multi2 = TestCase (assertEqual "n in octet [1,1], " 8 (multi [1,1]))
multi3 = TestCase (assertEqual "n in decuplet [3,0], " 10 (multi [3,0]))

yt1 = TestCase (assertEqual "yt [0],   " "# \n# \n"
        (show $ ytSymbols [0]))
yt2 = TestCase (assertEqual "yt [1],   " "# # \n# \n"
        (show $ ytSymbols [1]))
yt3 = TestCase (assertEqual "yt [0,0], " "# \n# \n# \n"
        (show $ ytSymbols [0,0]))
yt4 = TestCase (assertEqual "yt [1,0], " "# # \n# \n# \n"
        (show $ ytSymbols [1,0]))
yt5 = TestCase (assertEqual "yt [2,1], " "# # # # \n# # \n# \n"
        (show $ ytSymbols [2,1]))

tests = TestList [TestLabel "comb1" comb1,
                TestLabel "comb2" comb2,
                TestLabel "multi1" multi1,
                TestLabel "multi2" multi2,
                TestLabel "multi3" multi3,
                TestLabel "yt1" yt1,
                TestLabel "yt2" yt2,
                TestLabel "yt3" yt3,
                TestLabel "yt4" yt4,
                TestLabel "yt5" yt5]