inline-python-0.3.0.0: test/TST/Util.hs

-- |
module TST.Util where

import Control.Monad.IO.Class
import Control.Monad.Catch
import Data.Typeable
import Test.Tasty.HUnit

import Python.Inline

throwsPy :: Py () -> Py ()
throwsPy io = (io >> liftIO (assertFailure "Evaluation should raise python exception"))
  `catch` (\(_::PyError) -> pure ())

throwsException :: (Exception e, Eq e) => e -> Py () -> Py ()
throwsException e0 io = (io >> liftIO (assertFailure "Evaluation should raise python exception"))
  `catch` (\(SomeException e) -> case cast e of Just e' | e' == e0 -> pure ()
                                                _                  -> throwM e
          )

throwsPyIO :: IO () -> IO ()
throwsPyIO io = (io >> assertFailure "Evaluation should raise python exception")
  `catch` (\(_::PyError) -> pure ())

eq :: (Eq a, Show a, FromPy a) => Maybe a -> (Py PyObject) -> IO ()
eq a action = assertEqual "fromPy: " a =<< runPy (fromPy =<< action)