Upate csv test

jpeacock29 · jpeacock29 · commit 0107855da381 · 2019-04-27T18:56:29.000-04:00
diff --git a/amazonscraper/client.py b/amazonscraper/client.py
@@ -261,6 +261,8 @@ def _get_prices(self, product):
         """
         Given the HTML of a `product`, extract all prices.
         """
+        # XXX currently does not handle shipping prices or prices for the
+        # various formats of books.
 
         # match all prices of the form $X,XXX.XX:
         raw_prices = product.find_all(text=re.compile('\$[\d,]+.\d\d'))
diff --git a/test/test_amazonscraper.py b/test/test_amazonscraper.py
@@ -77,5 +77,7 @@ def test_amazonscraper_csv_header():
     products = amazonscraper.search(
                                 keywords="Python",
                                 max_product_nb=1)
-    assert "Product title,Rating,Number of customer reviews,Product URL,\
-Image URL,ASIN\n" in str(products.csv())
+    products.csv('test.csv')
+    with open('test.csv') as f:
+        csv_str = f.read()
+    assert "title,rating,review_nb,img,url,asin,prices_per_unit,units,prices_main"  in csv_str
