def test_data_initialization(self): '''All validations of the data initialization function are tested ''' self.assertRaises(UnboundLocalError, handlers.data_initialization, self.invalid_file_path) # Should throw an error because the file is not present self.data_initialization_return = handlers.data_initialization(self.valid_file_path) # Valid instantiation self.assertEquals(len(self.data_initialization_return), 2) # 2 parameters are returned through valid instantiation self.assertEquals(self.data_initialization_return[0].shape, self.dataframe.shape)
def setUp(self): '''Sets up the testing environment''' self.valid_file_path = TESTDATA_FILENAME self.invalid_file_path = "fake.csv" self.dataframe, self.crimes_object = handlers.data_initialization(TESTDATA_FILENAME) self.valid_university_name = "University of California-Berkeley" self.invalid_university_name = "Joke" self.valid_branch_name = "Main Campus" self.invalid_branch_name = "Jill"
def setUp(self):
'''Sets up the testing environment'''
self.valid_file_path = TESTDATA_FILENAME
self.invalid_file_path = "fake.csv"
self.dataframe, self.crimes_object = handlers.data_initialization(
TESTDATA_FILENAME)
self.valid_university_name = "University of California-Berkeley"
self.invalid_university_name = "Joke"
self.valid_branch_name = "Main Campus"
self.invalid_branch_name = "Jill"
def test_data_initialization(self):
'''All validations of the data initialization function are tested '''
self.assertRaises(
UnboundLocalError, handlers.data_initialization,
self.invalid_file_path
) # Should throw an error because the file is not present
self.data_initialization_return = handlers.data_initialization(
self.valid_file_path) # Valid instantiation
self.assertEquals(
len(self.data_initialization_return),
2) # 2 parameters are returned through valid instantiation
self.assertEquals(self.data_initialization_return[0].shape,
self.dataframe.shape)
""" Varun D N - [email protected] """ """The main program orchestrating the different components of the project""" import functionalities as func import handlers import os if __name__ == "__main__": """The main program running the software""" dataframe, crimes_obj = handlers.data_initialization("data/oncampuscrime101112_cleaned.csv") # Directory to hold the output images newpath = "output" if not os.path.exists(newpath): os.makedirs(newpath) try: func.interface(dataframe, crimes_obj) except (NameError, KeyError): print "Thanks for using the software"
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import handlers
import college
if __name__ == '__main__':
'''Handling the functionalities'''
dataframe, crimes_obj = handlers.data_initialization("data/oncampuscrime101112_cleaned.csv")
coll =college.College(pd.Series({"a":3},crimes_obj))
university_name = "University of Alabama in Huntsville"
branch_name = "Main_Campus"
college_instance = handlers.college_details(dataframe, university_name, branch_name)
college_obj = coll.College(college_instance, crimes_obj)
crime_per_student = handlers.all_crimes_per_student_over_years("On Campus", "Crime", college_instance, crimes_obj)
answers_obj = plots.Answers(crimes_obj, college_obj, pltparam, crime_per_student_without_average, crime_per_student_with_average, crimes_per_student_by_category)
d = handlers.average_crimes_per_student_by_category(dataframe, 'State' ,crimes_obj, overall_average = True)
answers_obj.simpleBarChart(d,"MA")
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import handlers
if __name__ == '__main__':
'''Handling the functionalities'''
# SEAN's PART
dataframe, crimes_obj = handlers.data_initialization("data/oncampuscrime101112.xls")
<<<<<<< HEAD
college_name = "University of Alabama in Huntsville"
=======
college_name = "Samford University"
# VARUN's PART
>>>>>>> upstream/master
college_instance = dataframe[dataframe.INSTNM == college_name]
crime_per_student = handlers.all_crimes_per_student_over_years("On Campus", "Crime", college_instance, crimes_obj)
for crime in crime_per_student.keys():
print crime
print crime_per_student[crime]
print ""
<<<<<<< HEAD
print dataframe['INSTNM'][:10]
#print crime_per_student
