def test_greedy(self):
print 'testing greedy function'
#my sample data
my_data = Data()
test_data = 'test.json'
my_data.extract_data(test_data)
uni_set = my_data.get_all_stores()
set_sub = my_data.get_zips()
#sorted_set = sorted(set_sub.values(), key=operator.attrgetter('num_nearby_stores'), reverse = True)
#for zip in sorted_set:
# zip.print_nearby_stores()
min_zip_check = greedy(uni_set, set_sub)
#test on given input file
real_data = Data()
data_file = 'gistfile1.json'
real_data.extract_data(data_file)
uni_set = real_data.get_all_stores()
set_sub = real_data.get_zips()
min_zip_check = greedy(uni_set, set_sub)
def test_extract_data(self):
print 'testing extract data fuction'
#my sample test
my_data = Data()
test_data_file = 'test.json'
my_data.extract_data(test_data_file)
print my_data.get_all_stores()
subset_stores = my_data.get_zips()
for key in subset_stores.keys():
print key
subset_stores[key].print_nearby_stores()
#print subset_stores[key]
#test on given input file
real_data = Data()
data_file = 'gistfile1.json'
real_data.extract_data(data_file)
