mirror of
https://github.com/facebook/duckling.git
synced 2024-12-01 08:19:36 +03:00
73 lines
2.0 KiB
Haskell
73 lines
2.0 KiB
Haskell
-- Copyright (c) 2016-present, Facebook, Inc.
|
|
-- All rights reserved.
|
|
--
|
|
-- This source code is licensed under the BSD-style license found in the
|
|
-- LICENSE file in the root directory of this source tree. An additional grant
|
|
-- of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
|
|
{-# LANGUAGE NoRebindableSyntax #-}
|
|
{-# LANGUAGE TupleSections #-}
|
|
|
|
module Duckling.Testing.Types
|
|
( Corpus
|
|
, Datetime
|
|
, Example
|
|
, NegativeCorpus
|
|
, TestPredicate
|
|
, dt
|
|
, examples
|
|
, examplesCustom
|
|
, parserCheck
|
|
, refTime
|
|
, testContext
|
|
, zTime
|
|
) where
|
|
|
|
import Data.Aeson
|
|
import Data.Fixed (Pico)
|
|
import qualified Data.Time as Time
|
|
import Data.Text (Text)
|
|
import Prelude
|
|
|
|
import Duckling.Lang
|
|
import Duckling.Resolve
|
|
import Duckling.Types
|
|
|
|
type TestPredicate = Context -> ResolvedToken -> Bool
|
|
type Example = (Text, TestPredicate)
|
|
type Corpus = (Context, [Example])
|
|
type NegativeCorpus = (Context, [Text])
|
|
|
|
examplesCustom :: TestPredicate -> [Text] -> [Example]
|
|
examplesCustom check = map (, check)
|
|
|
|
simpleCheck :: ToJSON a => a -> TestPredicate
|
|
simpleCheck json _ (Resolved {jsonValue = v}) = toJSON json == v
|
|
|
|
parserCheck :: Eq a => a -> (Value -> Maybe a) -> TestPredicate
|
|
parserCheck expected parse _ (Resolved {jsonValue = v}) =
|
|
maybe False (expected ==) $ parse v
|
|
|
|
examples :: ToJSON a => a -> [Text] -> [Example]
|
|
examples output = examplesCustom (simpleCheck output)
|
|
|
|
type Datetime = (Integer, Int, Int, Int, Int, Pico)
|
|
|
|
dt :: Datetime -> Time.UTCTime
|
|
dt (year, month, days, hours, minutes, seconds) = Time.UTCTime day diffTime
|
|
where
|
|
day = Time.fromGregorian year month days
|
|
diffTime = Time.timeOfDayToTime $ Time.TimeOfDay hours minutes seconds
|
|
|
|
zTime :: Datetime -> Int -> Time.ZonedTime
|
|
zTime datetime offset = fromUTC (dt datetime) $ Time.hoursToTimeZone offset
|
|
|
|
refTime :: Datetime -> Int -> DucklingTime
|
|
refTime datetime offset = fromZonedTime $ zTime datetime offset
|
|
|
|
-- Tuesday Feb 12, 2013 at 4:30am is the "now" for the tests
|
|
testContext :: Context
|
|
testContext = Context
|
|
{lang = EN, referenceTime = refTime (2013, 2, 12, 4, 30, 0) (-2)}
|