def __init__(self): self.connection = init.connect_to_db() self.connection_cursor = self.connection.cursor()
def test_flat_select(): """Select all employees no dereferencing""" Employee.many() def test_embedded_document_select(): """Select all companies no dereferencing""" Company.many(projection={'departments': {'$sub': Department}}) def test_full_select(): """Select all employees and their referenced companies""" Employee.many(projection={ 'company': { '$ref': Company, 'departments': {'$sub': Department} } }) if __name__ == "__main__": # Connect to the database connect_to_db() # Build the test data build_test_data() # Run the tests #time_it(test_flat_select, 100) #time_it(test_embedded_document_select, 100) time_it(test_full_select, 100)
# Define tests def test_flat_select(): """Select all employees no dereferencing""" list(Employee.objects.no_dereference()) def test_embedded_document_select(): """Select all companies no dereferencing""" list(Company.objects.no_dereference()) def test_full_select(): """Select all employees and their referenced companies""" [e.company for e in Employee.objects] if __name__ == "__main__": # Connect to the database connect_to_db() # Build the test data build_test_data() # Run the tests #time_it(test_flat_select, 100) #time_it(test_embedded_document_select, 100) time_it(test_full_select, 100)
def __init__(self): self.connection = init.connect_to_db() self.connection_cursor = self.connection.cursor() self.dataset_directory = "../datasets/Netflix_prize_dataset/download" self.movie_titles_file = "movie_titles.txt" self.training_set_dir = self.dataset_directory + "/training_set"