def test_get_asin_from_product(self):
# Test response as a list
papi = PAPI(deepcopy(TEST_RESPONSE))
products = papi.get_products_from_response()
asin0 = papi.get_asin_from_product(products[0])
asin1 = papi.get_asin_from_product(products[1])
asin2 = papi.get_asin_from_product(products[2])
self.assertEqual(asin0, "B00KUP84L2")
self.assertEqual(asin1, "B00HEATHXK")
self.assertEqual(asin2, "B00KURKYTK")
# IndexError and KeyError should raise KeyError
self.assertRaises(KeyError, papi.get_asin_from_product, {})
# Test response as a dict
papi2 = PAPI(deepcopy(TEST_RESPONSE_SM))
products = papi2.get_products_from_response()
asin = papi2.get_asin_from_product(products[0])
self.assertEqual(asin, "B00M2ALI02")
def test_get_asin_from_product(self):
# Test response as a list
papi = PAPI(deepcopy(TEST_RESPONSE))
products = papi.get_products_from_response()
asin0 = papi.get_asin_from_product(products[0])
asin1 = papi.get_asin_from_product(products[1])
asin2 = papi.get_asin_from_product(products[2])
self.assertEqual(asin0, 'B00KUP84L2')
self.assertEqual(asin1, 'B00HEATHXK')
self.assertEqual(asin2, 'B00KURKYTK')
# IndexError and KeyError should raise KeyError
self.assertRaises(KeyError, papi.get_asin_from_product, {})
# Test response as a dict
papi2 = PAPI(deepcopy(TEST_RESPONSE_SM))
products = papi2.get_products_from_response()
asin = papi2.get_asin_from_product(products[0])
self.assertEqual(asin, 'B00M2ALI02')
