def test_get_products_by_query_no_match(self):
"""
Test DB. Get products by query, with query without matchs
"""
db = DB(filename=DATASET_TESTS_FILE)
query = 'aaaaaaa'
result = db.get_products_by_query(query=query)
self.assertEqual(result, [], msg=result)
def test_get_products_by_query(self):
"""
Test DB. Get products by query
"""
db = DB(filename=DATASET_TESTS_FILE)
query = 'Multi'
result = db.get_products_by_query(query=query)
self.assertEqual(result, [('785', 50), ('8588', 30), ('12535', 15)], msg=result)
def test_get_products_by_query2(self):
"""
Test DB. Get products by query
"""
db = DB(filename=DATASET_TESTS_FILE)
query = 'See By'
result = db.get_products_by_query(query=query)
self.assertEqual(result, [('63602', 30)], msg=result)
import os
from db.db import DB
from settings import DATASET_FILE, QUERIES_FILE, DATASET_TESTS_FILE
from utils.read_files import read_queries_file
__author__ = 'josebermudez'
here = os.path.dirname(os.path.abspath(__file__))
if __name__ == '__main__':
# Charge data base
db = DB(filename=os.path.join(here, DATASET_FILE))
# Get queries
queries = read_queries_file(filename=os.path.join(here, QUERIES_FILE))
# queries = ['Multi']
for num, query in enumerate(queries):
print num+1
result = db.get_products_by_query(query)
for key, value in result[:10]:
print "%i,%i,%s,%s" % (int(value), int(key), db.get_name_by_id(product_id=key), db.get_brand_by_id(product_id=key))
