-- 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 OverloadedStrings #-}

module Duckling.Duration.FR.Corpus
  ( corpus
  , negativeCorpus
  ) where

import Prelude
import Data.String

import Duckling.Duration.Types
import Duckling.Lang
import Duckling.Resolve
import Duckling.Testing.Types
import Duckling.TimeGrain.Types (Grain(..))

corpus :: Corpus
corpus = (testContext {lang = FR}, allExamples)

negativeCorpus :: NegativeCorpus
negativeCorpus = (testContext {lang = FR}, examples)
  where
    examples =
      [ "les jours"
      , "en secondaire"
      , "minutes"
      , "pendant des mois"
      ]

allExamples :: [Example]
allExamples = concat
  [ examples (DurationData 1 Second)
             [ "une sec"
             , "1 seconde"
             , "1\""
             ]
  , examples (DurationData 2 Minute)
             [ "2 mins"
             , "deux minutes"
             , "2'"
             ]
  , examples (DurationData 30 Day)
             [ "30 jours"
             ]
  , examples (DurationData 7 Week)
             [ "sept semaines"
             ]
  , examples (DurationData 1 Month)
             [ "1 mois"
             , "un mois"
             ]
  , examples (DurationData 3 Quarter)
             [ "3 trimestres"
             ]
  , examples (DurationData 2 Year)
             [ "2 ans"
             ]
  ]