from unittest import TestCase
from sitemap.async_sitemap import sitemap


class TestSiteMap(TestCase):
    def setUp(self):
        self.url = 'https://konstantinfarrell.github.io'
        self.bigger_site = 'https://raspberrypi.org'
        self.biggest_site = 'https://learnxinyminutes.com'

    def test_sitemap(self):
        results = sitemap(self.url, write=False)
        self.assertIsNotNone(results)

    def test_bigger_sitemap(self):
        results = sitemap(self.bigger_site, write=False)
        self.assertIsNotNone(results)

    def test_biggest_sitemap(self):
        results = sitemap(self.biggest_site, write=False)
        self.assertIsNotNone(results)

@@ -40,14 +40,8 @@
 map.
-async_
 site

remove old reference