'''
GotoNewest tests
'''

import unittest
import sys
sys.path.append('../src/')
import gn

TEST_DIR = '/tmp/test'

class TestGotoNewest(unittest.TestCase):
    ''' Test class for GotoNewest
    '''

    def test_empty_base_dir(self):
        ''' If the base directory is empty, raise an
        AttributeError
        '''
        self.assertRaises(AttributeError, gn.transfer, None)

    def test_base_dir_with_no_subdirs(self):
        ''' If the base directory has no subdirectories,
        raise an AttributeError
        '''
        self.assertRaises(AttributeError, gn.transfer, TEST_DIR)


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

'''
GotoNewest tests
'''

import unittest
import sys
sys.path.append('../src/')
import gn

TEST_DIR = '/tmp/test'

class TestGotoNewest(unittest.TestCase):
    ''' Test class for GotoNewest
    '''

    def test_empty_base_dir(self):
        ''' If the base directory is empty, raise an
        AttributeError
        '''
        self.assertRaises(AttributeError, gn.transfer, None)

    def test_base_dir_with_no_subdirs(self):
        ''' If the base directory has no subdirectories,
        raise an AttributeError
        '''
        self.assertRaises(AttributeError, gn.transfer, TEST_DIR)

    def test_one_subdirectory(self):
        ''' If there is only one subdirectory in the base
        directory, return the directory
        '''
        self.assertRaises('temp', gn.transfer, TEST_DIR)


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

Add test for one subdirectory
Add test for one subdirectory
