class LocaldataTestCase(unittest.TestCase):
def setUp(self):
data_file = os.path.abspath(os.path.dirname(__file__) + os.path.sep + "../../") + '/yelp_example/yelp_3000_AZ.csv'
self.localdata = LocalData(data_file, 'csv', "business_id", ["name"],
["name", "full_address"])
def tearDown(self):
self.localdata = None
def test_loadLocalData(self):
self.localdata.setlocalData(None, None, None)
self.localdata.read_csv()
localdata_ids, localdata_query, localdata_er = self.localdata.getlocalData()
assert len(localdata_ids) == 3000
class LocaldataTestCase(unittest.TestCase):
def setUp(self):
data_file = os.path.abspath(
os.path.dirname(__file__) + os.path.sep +
"../../") + '/example/yelp_3000.pkl'
self.localdata = LocalData(data_file, 'pkl', "row['business_id']",
["row['name']"],
["row['name']", "row['full_address']"])
def tearDown(self):
self.localdata = None
def test_loadLocalData(self):
self.localdata.setlocalData(None, None, None)
self.localdata.read_pickle()
localdata_ids, localdata_query, localdata_er = self.localdata.getlocalData(
)
assert len(localdata_ids) == 3000