Testing.hs

-- Note: this is an automatically generated file. Do not edit.

-- | Python test code generation codec for pytest-based generation tests

module Hydra.Python.Testing where

import qualified Hydra.Ast as Ast
import qualified Hydra.Classes as Classes
import qualified Hydra.Coders as Coders
import qualified Hydra.Constants as Constants
import qualified Hydra.Core as Core
import qualified Hydra.Error.Checking as Checking
import qualified Hydra.Error.Core as ErrorCore
import qualified Hydra.Error.Packaging as ErrorPackaging
import qualified Hydra.Errors as Errors
import qualified Hydra.Formatting as Formatting
import qualified Hydra.Graph as Graph
import qualified Hydra.Json.Model as Model
import qualified Hydra.Overlay.Haskell.Lib.Chars as Chars
import qualified Hydra.Overlay.Haskell.Lib.Eithers as Eithers
import qualified Hydra.Overlay.Haskell.Lib.Equality as Equality
import qualified Hydra.Overlay.Haskell.Lib.Lists as Lists
import qualified Hydra.Overlay.Haskell.Lib.Logic as Logic
import qualified Hydra.Overlay.Haskell.Lib.Optionals as Optionals
import qualified Hydra.Overlay.Haskell.Lib.Strings as Strings
import qualified Hydra.Names as Names
import qualified Hydra.Packaging as Packaging
import qualified Hydra.Parsing as Parsing
import qualified Hydra.Paths as Paths
import qualified Hydra.Python.Syntax as Syntax
import qualified Hydra.Query as Query
import qualified Hydra.Relational as Relational
import qualified Hydra.Serialization as Serialization
import qualified Hydra.Tabular as Tabular
import qualified Hydra.Test.Utils as Utils
import qualified Hydra.Testing as Testing
import qualified Hydra.Topology as Topology
import qualified Hydra.Typed as Typed
import qualified Hydra.Typing as Typing
import qualified Hydra.Util as Util
import qualified Hydra.Validation as Validation
import qualified Hydra.Variants as Variants
import Prelude hiding  (Enum, Ordering, decodeFloat, encodeFloat, fail, map, pure, sum)
import qualified Data.Scientific as Sci

-- | Build the complete Python test module content
buildPythonTestModule :: t0 -> Testing.TestGroup -> String -> String
buildPythonTestModule testModule testGroup testBody =

      let groupName_ = Testing.testGroupName testGroup
          header =
                  Strings.cat [
                    Strings.cat2 "# " Constants.warningAutoGeneratedFile,
                    "\n",
                    (Strings.cat2 "# " groupName_),
                    "\n\n"]
      in (Strings.cat [
        header,
        testBody,
        "\n"])

-- | Format a test name for Python (snake_case with test_ prefix)
formatPythonTestName :: String -> String
formatPythonTestName name =
    Strings.cat2 "test_" (Strings.fromList (Lists.map (\c -> Logic.ifElse (Chars.isAlphaNum c) (Chars.toLower c) 95) (Strings.toList name)))

-- | Generate a single pytest test case from a test case with metadata
generatePythonTestCase :: [String] -> Testing.TestCaseWithMetadata -> Either t0 [String]
generatePythonTestCase groupPath tcm =

      let name_ = Testing.testCaseWithMetadataName tcm
          tcase = Testing.testCaseWithMetadataCase tcm
      in case tcase of
        Testing.TestCaseUniversal v0 ->
          let actual_ = Testing.universalTestCaseActual v0 ()
              expected_ = Testing.universalTestCaseExpected v0 ()
              fullName = Logic.ifElse (Lists.null groupPath) name_ (Strings.intercalate "__" (Lists.concat2 groupPath [
                    name_]))
              formattedName = formatPythonTestName fullName
          in (Right [
            Strings.cat [
              "def ",
              formattedName,
              "():"],
            (Strings.cat [
              "    assert (",
              actual_,
              ") == (",
              expected_,
              ")"])])

-- | Generate a Python test file for a test group
generatePythonTestFile :: Packaging.Module -> Testing.TestGroup -> t0 -> Either t1 (String, String)
generatePythonTestFile testModule testGroup _g = generateTestFileWithPythonCodec testModule testGroup

-- | Generate test hierarchy for Python with nested subgroups
generatePythonTestGroupHierarchy :: [String] -> Testing.TestGroup -> Either t0 String
generatePythonTestGroupHierarchy groupPath testGroup =

      let cases_ = Testing.testGroupCases testGroup
          subgroups = Testing.testGroupSubgroups testGroup
      in (Eithers.bind (Eithers.mapList (\tc -> generatePythonTestCase groupPath tc) cases_) (\testCaseLines -> Eithers.bind (Eithers.mapList (\subgroup ->
        let groupName = Testing.testGroupName subgroup
            header = Strings.cat2 "# " groupName
        in (Eithers.map (\content -> Strings.cat [
          header,
          "\n\n",
          content]) (generatePythonTestGroupHierarchy (Lists.concat2 groupPath [
          groupName]) subgroup))) subgroups) (\subgroupBlocks ->
        let testCasesStr = Strings.intercalate "\n\n" (Lists.concat testCaseLines)
            subgroupsStr = Strings.intercalate "\n\n" subgroupBlocks
        in (Right (Strings.cat [
          testCasesStr,
          (Logic.ifElse (Logic.or (Equality.equal testCasesStr "") (Equality.equal subgroupsStr "")) "" "\n\n"),
          subgroupsStr])))))

-- | Generate a complete test file for Python
generateTestFileWithPythonCodec :: Packaging.Module -> Testing.TestGroup -> Either t0 (String, String)
generateTestFileWithPythonCodec testModule testGroup =
    Eithers.map (\testBody ->
      let testModuleContent = buildPythonTestModule testModule testGroup testBody
          ns_ = Packaging.moduleName testModule
          parts = Strings.splitOn "." (Packaging.unModuleName ns_)
          dirParts = Optionals.fromOptional [] (Lists.maybeInit parts)
          fileName =
                  Strings.cat [
                    "test_",
                    (Optionals.fromOptional "" (Lists.maybeLast parts)),
                    ".py"]
          filePath =
                  Strings.cat [
                    Strings.intercalate "/" dirParts,
                    "/",
                    fileName]
      in (filePath, testModuleContent)) (generatePythonTestGroupHierarchy [] testGroup)