text: # coding=utf-8
  # Copyright 2019 The TensorFlow Datasets Authors.
  #
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #     http://www.apache.org/licenses/LICENSE-2.0
  #
  # Unless required by applicable law or agreed to in writing, software
  # distributed under the License is distributed on an "AS IS" BASIS,
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  # See the License for the specific language governing permissions and
  # limitations under the License.
  
  """Tests for WMT translate dataset module."""
  
  from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  
  from tensorflow_datasets import testing
  from tensorflow_datasets.translate import wmt
  
  
  class TranslateWmtCustomConfigTest(testing.DatasetBuilderTestCase):
  
    @classmethod
    def setUpClass(cls):
      super(TranslateWmtCustomConfigTest, cls).setUpClass()
  
      config = wmt.WmtConfig(
          name="small",
          language_pair=("cs", "en"),
          description="Example of custom config",
          subsets={
              "train": ["paracrawl_v3"],
              "validation": ["newstest2009", "newstest2010"],
          },
          version="0.0.1",
      )
      wmt.WmtTranslate.BUILDER_CONFIGS = [config]
  
    @classmethod
    def tearDownClass(cls):
      super(TranslateWmtCustomConfigTest, cls).tearDownClass()
      wmt.WmtTranslate.BUILDER_CONFIGS.pop()
  
    DATASET_CLASS = wmt.WmtTranslate
    # OVERLAPPING_SPLITS = ["validation"]
  
    DL_EXTRACT_RESULT = {
        "czeng17_filter": ["czeng"],
        "europarl_v9": ["sentences.cs-en.tsv"],
        "paracrawl_v3": ["sentences.cs-en.tmx"],
        "commoncrawl": ["commoncrawl"],
        "newscommentary_v14": ["sentences.cs-en.tsv"],
        "wikititles_v1": ["sentences.cs-en.tsv"],
        "newssyscomb2009": ["validation"],
        "newstest2008": ["validation"],
        "newstest2009": ["validation"],
        "newstest2010": ["validation"],
        "newstest2011": ["validation"],
        "newstest2012": ["validation"],
        "newstest2013": ["validation"],
        "newstest2014": ["validation"],
        "newstest2015": ["validation"],
        "newstest2016": ["validation"],
        "newstest2017": ["validation"],
        "newstest2018": ["validation"],
    }
  
    SPLITS = {
        "train": 2,
        "validation": 4,
    }
  
  
  if __name__ == "__main__":
    testing.test_main()
  