rick-and-morty-downloader-py/TestScraper.py

#!/usr/bin/env python
# TestScraper.py

import unittest
from scraper import Scraper


class KnownValues(unittest.TestCase):
    scraper = Scraper()

    def test_instance(self):
        self.assertIsInstance(self.scraper, Scraper)

    def test_bytes_to_multiples(self):
        self.assertEqual(Scraper.bytes_to_multiples(8302.2049371), '8.3kb')
        self.assertEqual(Scraper.bytes_to_multiples(8302.2049371), '8.3kb')
        self.assertEqual(Scraper.bytes_to_multiples(10), '10b')
        self.assertEqual(Scraper.bytes_to_multiples(10000), '10kb')
        self.assertEqual(Scraper.bytes_to_multiples(20000), '20kb')
        self.assertEqual(Scraper.bytes_to_multiples(10000000), '10mb')
        self.assertEqual(Scraper.bytes_to_multiples(1000000000), '1gb')
        self.assertEqual(Scraper.bytes_to_multiples(1250000000), '1.25gb')
        self.assertEqual(Scraper.bytes_to_multiples(160000000), '160mb')

    def test_get_speed_string(self):
        self.assertEqual(Scraper.get_speed_string(768000), '768kb/s')

    def test_get_completion(self):
        self.assertEqual(Scraper.get_completion(10, 100), '10% - 10b/100b')
        self.assertEqual(Scraper.get_completion(3, 10), '30% - 3b/10b')
        self.assertEqual(Scraper.get_completion(24, 289), '8.3% - 24b/289b')

    def test_calc_res(self):
        self.assertEqual(Scraper.calc_res('800x600'), 800 * 600)
        self.assertEqual(Scraper.calc_res('1920x1024'), 1920 * 1024)


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