57 lines
1.9 KiB
Haskell
57 lines
1.9 KiB
Haskell
{-# 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 ingest facts and rules" $ do
|
|
let db = NaiveDatabase.withFactsAndRules
|
|
[ "parent(\"alice\", \"bob\")."
|
|
, "parent(\"bob\", \"carol\")." ]
|
|
[ "ancestor(X,Y) :- parent(X,Y)."
|
|
, "ancestor(X,Y) :- parent(X,Z), ancestor(Z,Y)." ]
|
|
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'
|