import unittest
from nala.bootstrapping import generate_documents


class TestBootstrapping(unittest.TestCase):
    def test_generate_documents_number(self):
        # commenting out for now since it takes about 6 mins on Travis CI
        test_dataset = generate_documents(1)
        self.assertEqual(len(test_dataset), 1)
        pass


if __name__ == '__main__':
    unittest.main()

import unittest
from nala.bootstrapping import generate_documents


class TestBootstrapping(unittest.TestCase):
    def test_generate_documents_number(self):
        # commenting out for now since it takes about 6 mins on Travis CI
        # test_dataset = generate_documents(1)
        # self.assertEqual(len(test_dataset), 1)
        pass


if __name__ == '__main__':
    unittest.main()

Remove long test for now