{-# LANGUAGE NoMonomorphismRestriction #-}

module Numeric.Units.Dimensional.Test where

import Numeric.Units.Dimensional.Prelude
import qualified Prelude
import Test.HUnit

testPower :: Test
testPower = TestLabel "Power test" $ TestList
    [ TestCase $ ((9::Double) *~ one) @=? (3 *~ one) ^ pos2
    , TestCase $ ((1::Double) *~ one) @=? (12.1231 *~ one) ^ zero
    , TestCase $ ((0.25::Double) *~ one) @=? (2 *~ one) ^ neg2
    ]

testDimensionless :: Test
testDimensionless = TestLabel "Dimensionless test" $ TestList
    [ TestCase $ (3 Prelude.** 2::Double) *~ one @=? (3 *~ one) ** (2 *~ one)
    ]

testShow :: Test
testShow = TestLabel "Test 'Show' instance" $ TestList
    [ TestCase $ show ((1.0::Double) *~ one) @?= "1.0"
    , TestCase $ show ((2.0::Double) *~ meter) @?= "2.0 m"
    , TestCase $ show ((2.0::Double) *~ (meter / second)) @?= "2.0 m s^-1"
    , TestCase $ show ((2.0::Double) *~ (meter ^ pos2 / second ^ pos2)) @?= "2.0 m^2 s^-2"
    --, TestCase $ show (undefined :: DimRep DVelocity) @?= "m s^-1"
    ]

testOrdering :: Test
testOrdering = TestLabel "Test 'Ord' instance" $ TestList
    [ TestCase $ compare ((1 :: Integer) *~ one) (3 *~ one) @?= LT
    , TestCase $ compare ((1 :: Double) *~ (kilo meter)) (1 *~ meter) @?= GT
    , TestCase $ compare ((0 :: Double) *~ second) (_0) @?= EQ 
    ]

testNFromTo :: Test
testNFromTo = TestLabel "Test enumeration function 'nFromTo'" $ TestList
    [ TestCase $ nFromTo' _1 _6 0    @?= [_1, _6]
    , TestCase $ nFromTo' _1 _6 (-1) @?= [_1, _6]
    , TestCase $ nFromTo' _1 _3 1    @?= [_1, _2, _3]
    , TestCase $ nFromTo' _1 _6 4    @?= [_1, _2, _3, _4, _5, _6]
    , TestCase $ nFromTo' _5 _2 2    @?= [_5, _4, _3, _2]
    , TestCase $ nFromTo' _0 _6 2    @?= [_0, _2, _4, _6]
    , TestCase $ nFromTo' _6 _0 2    @?= [_6, _4, _2, _0]
    , TestCase $ nFromTo' _1 _1 0    @?= [_1, _1]
    , TestCase $ nFromTo' _0 _0 2    @?= [_0, _0, _0, _0]
    ]
  where
    nFromTo' :: Dimensionless Double -> Dimensionless Double -> Int -> [Dimensionless Double]
    nFromTo' = nFromTo

-- Collect the test cases.
tests :: Test
tests = TestList
    [ testPower
    , testDimensionless
    , testShow
    , testOrdering
    , testNFromTo
    ]

main :: IO Bool -- True means everything passed
main = do
         res <- runTestTT tests
         return $ (errors res == 0) && (failures res == 0)