# Path: baleen/exceptions.py
# class ExportError(BaleenError):
#     """
#     Something went wrong with the export of the corpus
#     """
#     pass


import unittest
    import mock
    from unittest import mock
from baleen.export import *
from baleen.exceptions import ExportError
##########################################################################
## Imports
##########################################################################


try:
except ImportError:



##########################################################################
## Export Tests
##########################################################################

class ExportTests(unittest.TestCase):

    def test_scheme_specification(self):
        """
        Assert that only known schemes are allowed.
        """

        # Make sure good schemes don't error
        for scheme in SCHEMES:
            try:
                exporter = MongoExporter("/tmp/corpus", scheme=scheme)
            except ExportError:
                self.fail("Could not use expected scheme, {}".format(scheme))

        # Make sure bad schemes do error
        for scheme in ('text', 'txt', 'bson', 'xml', 'yaml'):

# Path: baleen/exceptions.py
# class ExportError(BaleenError):
#     """
#     Something went wrong with the export of the corpus
#     """
#     pass


# Path: tests/test_export.py
import unittest
    import mock
    from unittest import mock
from baleen.export import *
from baleen.exceptions import ExportError

##########################################################################
## Imports
##########################################################################


try:
except ImportError:



##########################################################################
## Export Tests
##########################################################################

class ExportTests(unittest.TestCase):

    def test_scheme_specification(self):
        """
        Assert that only known schemes are allowed.
        """

        # Make sure good schemes don't error
        for scheme in SCHEMES:
            try:
                exporter = MongoExporter("/tmp/corpus", scheme=scheme)
            except ExportError:
                self.fail("Could not use expected scheme, {}".format(scheme))

        # Make sure bad schemes do error
        for scheme in ('text', 'txt', 'bson', 'xml', 'yaml'):

            with self.assertRaises(ExportError):