Idris2/tests/idris2/total/total020/Check.idr

%default total

--- Functor-related ---

data X : Type -> Type -> Type where
  B : b -> X a b
  R : X a b -> X b c -> X a c

[WithFunction] Functor (X a) where
  map f $ B x = B $ f x
  map f $ R x y = R x (map f y)

[WithInfix] Functor (X a) where
  map f $ B x = B $ f x
  map f $ R x y = R x (f <$> y)

[WithFoldMap] Foldable (X a) where

  foldr = ?some_foldr_1

  foldMap f $ B x   = f x
  foldMap f rlr@(R l r) = do
    let r' = map @{WithFunction} f r
    foldMap id $ assert_smaller rlr r'

[WithConcat] Foldable (X a) where

  foldr = ?some_foldr_2

  foldMap f $ B x   = f x
  foldMap f rlr@(R l r) = do
    let r' = map @{WithFunction} f r
    concat $ assert_smaller rlr r'

--- Show-related ---

data Tree : Type -> Type where
  Leaf : a -> Tree a
  Node : a -> (left, right : Tree a) -> Tree a

Show a => Show (Tree a) where
  showPrec d (Leaf x)     = showCon d "Leaf" $ showArg x
  showPrec d (Node x l r) = showCon d "Node" $ showArg x ++ showArg l ++ showArg r