#!/usr/bin/env python
# -*- coding: utf-8 -*-

import unittest
import os


class TestDataSet(unittest.TestCase):
    """
    Class for testing the enron dataset.
    """
    def setUp(self):
        self.dataset_dir = 'enron_dataset'

    def tearDown(self):
        self.dataset_dir = ''

    def test_dataset_folder_exist(self):
        return self.assertEqual(True, os.path.exists(self.dataset_dir))


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

@@ -397,16 +397,128 @@
 _dir))%0A%0A
+    def test_dataset_is_a_folder(self):%0A        return self.assertEqual(True, os.path.isdir(self.dataset_dir))%0A%0A
 %0Aif __na

Add test dataset is a folder.