import sys
import unittest
import os
sys.path.append('../')
import get_data as gd   # noqa


class FunctionRunTest(unittest.TestCase):
    """Test if functions run succesfully with correct input"""

    def test_get_tmdb_id_list(self):
        gd.get_tmdb_id_list(2011, 2012, 1, 2)

    def test_get_profit(self):
        gd.get_profit(2011, 2012, 1, 2)

    def test_get_info(self):
        gd.get_info(2011, 2012, 1, 2)

    def test_call_data(self):
        os.chdir('../')
        gd.call_data(2011, 2012, 1, 2)


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