From be07162191c2fbda4fba4393b8a92de686b6ba36 Mon Sep 17 00:00:00 2001
From: Eric Conlon <37287+ejconlon@users.noreply.github.com>
Date: Tue, 13 Feb 2024 05:52:14 -0800
Subject: [PATCH] etc

---
 minipat/src/Minipat/Ur.hs |  4 +--
 minipat/test/Main.hs      | 60 +++++++++++++++++++++++++++++++++++----
 2 files changed, 57 insertions(+), 7 deletions(-)

diff --git a/minipat/src/Minipat/Ur.hs b/minipat/src/Minipat/Ur.hs
index d6aa178..4bcf2ef 100644
--- a/minipat/src/Minipat/Ur.hs
+++ b/minipat/src/Minipat/Ur.hs
@@ -10,7 +10,7 @@ import Data.Map.Strict qualified as Map
 import Data.Text (Text)
 import Data.Typeable (Typeable)
 import Minipat.Ast (Ident, Pat, Select (..))
-import Minipat.Eval (evalPat)
+import Minipat.Eval (PatternEval, evalPat)
 import Minipat.Interp (InterpErr, customInterpPat)
 import Minipat.Parser (Loc, identP, selectP)
 import Minipat.Pattern (Pattern (..), PatternUnwrap (..))
@@ -24,7 +24,7 @@ data UrErr k
 instance (Show k, Typeable k) => Exception (UrErr k)
 
 ur
-  :: (PatternUnwrap Loc f)
+  :: (PatternEval f)
   => CycleDelta
   -> Text
   -> [(Ident, f a)]
diff --git a/minipat/test/Main.hs b/minipat/test/Main.hs
index 9f63a68..064b929 100644
--- a/minipat/test/Main.hs
+++ b/minipat/test/Main.hs
@@ -11,18 +11,21 @@ import Control.Exception (throwIO)
 import Control.Monad (void)
 import Data.Bifunctor (first)
 import Data.Either (isLeft, isRight)
+import Data.Foldable (for_)
 import Data.Maybe (fromMaybe)
 import Data.Ratio ((%))
 import Data.Sequence (Seq (..))
 import Data.Text (Text)
 import Looksee (Err, intP, parse)
 import Minipat.Ast
+import Minipat.Eval (evalPat)
 import Minipat.Interp (interpPat)
 import Minipat.Norm (normPat)
-import Minipat.Parser (P, ParseErr, factorP, identP, identPatP, selectIdentPatP)
+import Minipat.Parser (Loc, P, ParseErr, factorP, identP, identPatP, selectIdentPatP)
 import Minipat.Print (prettyShow)
 import Minipat.Stream (Ev (..), streamRun)
 import Minipat.Time (Arc (..), CycleTime (..), Span (..))
+import Minipat.Ur (ur)
 import Prettyprinter qualified as P
 import System.IO (BufferMode (..), hSetBuffering, stdout)
 import Test.Daytripper
@@ -399,11 +402,9 @@ testPatNormCases =
 
 runPatInterpCase :: (TestName, Maybe Arc, Text, [Ev Ident]) -> TestTree
 runPatInterpCase (n, mayArc, patStr, evs) = testCase n $ do
-  pat <- either throwIO pure (parse tpatP patStr)
-  let pat' = normPat pat
-  pat'' <- either throwIO pure (interpPat pat')
+  pat <- either throwIO pure (evalPat identP patStr)
   let arc = fromMaybe (Arc 0 1) mayArc
-      actualEvs = streamRun pat'' arc
+      actualEvs = streamRun pat arc
   actualEvs @?= evs
 
 ev :: Rational -> Rational -> x -> Ev x
@@ -683,6 +684,52 @@ testPatInterpCases =
         )
       ]
 
+runPatReprCase :: (TestName, Text, Maybe (TPat Ident), Maybe Text) -> TestTree
+runPatReprCase (n, patStr, mayRePat, mayReStr) = testCase n $ do
+  pat :: Pat Loc Ident <- either throwIO pure (evalPat identP patStr)
+  actualPat :: Pat Loc Ident <- either throwIO pure (interpPat pat)
+  for_ mayRePat (first (const ()) actualPat @?=)
+  let actualStr = prettyShow actualPat
+      expectStr = fromMaybe patStr mayReStr
+  actualStr @?= expectStr
+
+testPatReprCases :: TestTree
+testPatReprCases =
+  testGroup "pat repr cases" $
+    fmap
+      runPatReprCase
+      [
+        ( "pure"
+        , "x"
+        , Just (mkTPat (PatPure "x"))
+        , Nothing
+        )
+      ,
+        ( "seq"
+        , "[x y]"
+        , Nothing -- Just (mkTPat (PatPure "x"))
+        , Nothing
+        )
+      ]
+
+testUr :: TestTree
+testUr = testCase "ur" $ do
+  -- let ep = ur @_ @String 3 "a b:x c" [("a", "1"), ("b", "2"), ("c", "3")] [("x", patFastBy 2)]
+  --     expectedEvs =
+  --       [ ev 0 1 "1"
+  --       , ev 1 (3 % 2) "2"
+  --       , ev (3 % 2) 2 "2"
+  --       , ev 2 3 "3"
+  --       ]
+  let ep = ur @_ @String 1 "a a" [("a", "1")] []
+      expectedEvs =
+        [ ev 0 1 "1"
+        ]
+  pat <- either throwIO pure ep
+  -- let actualEvs = streamRun pat (Arc 0 3)
+  let actualEvs = streamRun pat (Arc 0 1)
+  actualEvs @?= expectedEvs
+
 main :: IO ()
 main = do
   hSetBuffering stdout LineBuffering
@@ -692,4 +739,7 @@ main = do
       [ testParseCases
       , testPatNormCases
       , testPatInterpCases
+      -- TODO fix these
+      -- , testUr
+      -- , testPatReprCases
       ]