class TestInnsExtract(unittest.TestCase):
def setUp(self):
unittest.TestCase.setUp(self)
with open('inns.html', 'r') as f:
self.tree = etree.parse(f, etree.HTMLParser())
self.hotels = InnsHotels('3100')
self.soup = None
def test_extract_page(self):
(total, pages) = self.hotels._extractPages(self.tree)
self.assertEqual((40,8), (total,pages))
def test_extract_hotels(self):
hotels = self.hotels._extractHotels(self.tree)
self.assertEqual(5, len(hotels))
self.assertEqual('北京国际展览中心店', hotels[0]['name'])
self.assertEqual('0622', hotels[0]['id'])
self.assertIsNotNone(hotels[0]['address'])
self.assertIsNotNone(hotels[0]['bigScore'])
self.assertIsNotNone(hotels[0]['telephone'])
self.assertIsNotNone(hotels[0]['intro'])
def test_extract_prices(self):
prices = self.hotels._extractPrices(self.tree)
self.assertEqual(5, len(prices))
hotelPrice = prices[0]
self.assertEqual(269, hotelPrice[u'商务房A'])
def test_getInnsCities(self):
hotels = InnsHotels('3100')
cities = hotels._extractCities()
self.assertEqual(161, len(cities))
def test_inns_fetch(self):
inns = INNS('4401')
print inns.extract()
def setUp(self):
unittest.TestCase.setUp(self)
with open('inns.html', 'r') as f:
self.tree = etree.parse(f, etree.HTMLParser())
self.hotels = InnsHotels('3100')
self.soup = None
