import unittest
from rivers import *

class RiverTest(unittest.TestCase):
    def test_rivers(self):
        rivers()

    def test_polling(self):
        polling()


if __name__ == "__main__":
    unittest.main()
