hpython-0.1: test/Optics.hs

{-# language OverloadedStrings, TemplateHaskell #-}
module Optics (opticsTests) where

import Hedgehog

import Control.Lens.Plated (transformOn)
import Control.Lens.Setter ((.~))
import Control.Monad.IO.Class (liftIO)
import qualified Data.Text.IO as Text

import Language.Python.Parse (parseModule)
import Language.Python.Render (showModule)
import Language.Python.Syntax.Statement (_Statements)
import Language.Python.Syntax.Whitespace (Whitespace (..))
import Language.Python.Optics (_Indent)

import Helpers (shouldBeParseSuccess)

opticsTests :: Group
opticsTests = $$discover

prop_optics_1 :: Property
prop_optics_1 =
  withTests 1 . property $ do
    str <- liftIO $ Text.readFile "test/files/indent_optics_in.py"

    tree <- shouldBeParseSuccess parseModule str

    str' <- liftIO $ Text.readFile "test/files/indent_optics_out.py"
    showModule
      (transformOn _Statements (_Indent .~ [Space, Space, Space, Space]) tree) === str'

prop_optics_2 :: Property
prop_optics_2 =
  withTests 1 . property $ do
    str <- liftIO $ Text.readFile "test/files/indent_optics_in2.py"

    tree <- shouldBeParseSuccess parseModule str
    -- annotateShow $! tree

    str' <- liftIO $ Text.readFile "test/files/indent_optics_out2.py"
    showModule
      (transformOn _Statements (_Indent .~ [Space, Space, Space, Space]) tree) === str'