Python get_listings_dataframe Examples

Example #1

File: test_get_cleaned_listings.py Project: mag3141592/Zillowbnb

 def test_wrong_check_dtype(self):
     """
     Tests that get_listings_dataframe will not run if datatypes are incorrect
     :param self:
     :return boolean:
     """
     with self.assertRaises(ValueError):
         get_cleaned_listings.get_listings_dataframe('Incorrect', constants.LISTING_COLUMNS)

Example #2

File: test_get_cleaned_listings.py Project: mag3141592/Zillowbnb

 def test_listings_col(self):
     """
     Test cleaned_listings data has all 40 columns
     :param self:
     :return boolean:
     """
     listings = get_cleaned_listings.get_listings_dataframe(DATA, constants.LISTING_COLUMNS)
     self.assertTrue(listings.shape[1] == 40)

Example #3

File: test_get_cleaned_listings.py Project: mag3141592/Zillowbnb

 def test_clean_listings_row(self):
     """
     Tests cleaned_listings data for more than one row
     :param self:
     :return boolean:
     """
     listings = get_cleaned_listings.get_listings_dataframe(DATA, constants.LISTING_COLUMNS)
     self.assertTrue(listings.shape[0] >= 1)

Example #4

File: test_get_cleaned_listings.py Project: mag3141592/Zillowbnb

 def test_listings_col_types(self):
     """
     Test cleaned_listings data has the correct data column types
     :param self:
     :return boolean:
     """
     valid_types = ['int', 'O', 'O', 'float', 'float', 'O', 'O', 'int',
                    'int', 'int', 'float', 'float', 'float', 'float', 'float',
                    'float', 'float', 'float', 'float', 'float', 'float',
                    'float', 'float', 'float', 'float', 'float', 'float',
                    'float', 'float', 'float', 'float', 'float', 'float',
                    'float', 'float', 'float', 'float', 'float', 'float', 'float']
     listings = get_cleaned_listings.get_listings_dataframe(DATA, constants.LISTING_COLUMNS)
     types = listings.dtypes
     self.assertTrue((types == valid_types).all())

Example #5

File: test_train_model.py Project: mag3141592/Zillowbnb

This module runs unit test for training the machine learning model.
"""

import unittest

import zillowbnb.test.submodule_path

import constants as co
import convert_to_matrix as cm
import get_data as gd
import get_cleaned_listings as gcl
import train_model as tm

DATA = gd.download_dataset(co.DATASET_PROPERTIES, co.LISTINGS_DATA)

DATAFRAME = gcl.get_listings_dataframe(DATA, co.LISTING_COLUMNS)

X_VAR, Y_VAR = cm.to_matrix(DATAFRAME, co.LISTING_COLUMNS)


class TrainModelTest(unittest.TestCase):
    """
    This class runs unit tests for the train_model module.
    """
    def test_input_size(self):
        """
        Tests that train_model will not run if sizes of x_var and y_var
        do not match.
        :params self:
        :returns boolean:
        """