def generate_stores(self, num=None): generator = StoreGenerator(zipcode_objs=self.zipcode_objs, income_scaling_factor=sim_param.STORE_INCOME_SCALING_FACTOR) self.stores = generator.generate(n=num)
first_names.append((name, weight)) if cols[3] == "1": last_names.append((name, weight)) name_fl.close() return first_names, last_names if __name__ == "__main__": from zipcodes import load_zipcode_data from stores import StoreGenerator print "Loading zipcode data..." zipcode_objs = load_zipcode_data() print print "Generating Stores..." generator = StoreGenerator(zipcode_objs=zipcode_objs) stores = generator.generate(n=100) print print "Generating customers..." generator = CustomerGenerator(zipcode_objs=zipcode_objs, stores=stores) customers = generator.generate(10) print print "Customers" for c in customers: print c print