def testDoesPDFAlreadyExist(self):
   remove_data('plot_x_vs_y.pdf')
   plot_x_vs_y(get_trips_and_weather(),'Annual Member','Min_TemperatureF',
       'plot_x_vs_y')
   result = plot_x_vs_y(get_trips_and_weather(),'Annual Member',
       'Min_TemperatureF','plot_x_vs_y')
   self.assertTrue(result)
    def test_invalid_plot_no_file(self):
        """
        Here we check that the pdf file is NOT generated
        when a valid data frame, but an invalid column
        are supplied to the function.
        """

        data = get_trips_and_weather()

        fileName = 'Temp_and_Membership_invalid'
        plot_x_vs_y(data,'Mean_Temperaturezzzz_F','Annual Member', fileName)
        filePath = os.getcwd() + '\\' + fileName + '.pdf'
        self.assertFalse(os.path.exists(filePath))
    def test_invalid_plot_exception_trhown(self):
        """
        Here we check that a TypeError exception is
        thrown when a valid data frame, but an invalid
        column are supplied to the function.
        """

        data = get_trips_and_weather()

        fileName = 'Temp_and_Membership_invalid'
        plot_x_vs_y(data,'Mean_Temperaturezzzz_F','Annual Member', fileName)
        filePath = os.getcwd() + '\\' + fileName + '.pdf'
        #self.assertFalse(os.path.exists(filePath))
        self.assertRaises(KeyError)
    def test_valid_plot(self):
        """
        Here we check that the pdf file is properly generated
        when a valid data frame and columns are supplied as
        arguments to the function.
        """

        # download the trip and weather data
        # NOTE: we turn off alerts about whether or not data
        # has already been downloaded within the pronto_utils.py
        # code in order to make unit test output cleaner
        data = get_trips_and_weather()

        fileName = 'Temp_and_Membership'
        plot_x_vs_y(data,'Mean_Temperature_F','Annual Member', fileName)
        filePath = os.getcwd() + '\\' + fileName + '.pdf'
        self.assertTrue(os.path.exists(filePath))
 def testWasPDFCreated(self):
     remove_data('plot_x_vs_y.pdf')
     plot_x_vs_y(get_trips_and_weather(),'Annual Member','Min_TemperatureF',
         'plot_x_vs_y')
     self.assertTrue(os.path.exists('plot_x_vs_y.pdf'))