pyrites/haskell-experiments/test/Test/Datalog/NaiveDatabaseSpec.hs

{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE DuplicateRecordFields #-}
{-# LANGUAGE OverloadedRecordDot #-}
{-# HLINT ignore "Use const" #-}
{-# HLINT ignore "Unused LANGUAGE pragma" #-}
{-# HLINT ignore "Avoid lambda" #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE NoFieldSelectors #-}
{-# OPTIONS_GHC -Wno-unrecognised-pragmas #-}
{-# LANGUAGE TypeApplications #-}
{-# LANGUAGE ImportQualifiedPost #-}

module Test.Datalog.NaiveDatabaseSpec where

import Test.Hspec
import Datalog.NaiveDatabase
import qualified Datalog.NaiveDatabase as NaiveDatabase
import Data.Set qualified as Set
import Datalog.DatalogParser
import qualified Data.Map as Map

spec :: Spec
spec = do
  describe "dummy test" $ do
    it "..." $ do
      1 `shouldBe` (1 :: Int)
    it "can ingest facts into relations & a universe" $ do
      let db = NaiveDatabase.withFacts 
            [ "parent(\"alice\", \"bob\")."
            , "parent(\"bob\", \"carol\")." ]
      constants db `shouldBe` 
        (Set.fromList $ Sym <$> ["alice", "bob", "carol"])
      relations db `shouldBe`
        Map.fromList [
          ("parent",  
            Relation 2 $ Set.fromList $ [Sym <$> ["bob", "carol"]])
        ]

    it "can do basic queries" $ do
      let db = NaiveDatabase.withFacts 
            [ "parent(\"alice\", \"bob\")."
            , "parent(\"bob\", \"carol\")." ]
      query db "?- parent(alice,X)." `shouldBe` "#NYI" -- ideally, 'bob'