wikimedia/pywikibot

#!/usr/bin/env python3
"""Test cases for the WikiStats dataset."""
#
# (C) Pywikibot team, 2014-2022
#
# Distributed under the terms of the MIT license.
#
from __future__ import annotations

import sys
import unittest
from contextlib import suppress

from pywikibot.data.wikistats import WikiStats
from tests.aspects import TestCase


class WikiStatsTestCase(TestCase):

    """Test WikiStats dump."""

    net = True

    def test_sort_numeric(self):
        """Test sorted results."""
        keys = ('good', 'prefix', 'total')

        ws = WikiStats()
        data = ws.sorted('wikipedia', 'total')
        top = data[0]
        bottom = data[-1]

        for key in keys:
            with self.subTest(key=key):
                self.assertIn(key, top)
                self.assertIn(key, bottom)

        for key in top:
            self.assertIsInstance(key, (str, type(None)))

        self.assertIsInstance(top['good'], str)
        self.assertIsInstance(top['total'], str)
        self.assertIsInstance(bottom['good'], str)
        self.assertIsInstance(bottom['total'], str)

        self.assertGreater(int(top['total']), int(bottom['good']))
        self.assertGreater(int(top['good']), int(bottom['good']))
        self.assertGreater(int(top['total']), int(bottom['total']))

    def test_sort_alphabetic(self):
        """Test alphabetic sorted results."""
        ws = WikiStats()
        for reverse in (True, False):
            last = ' ~'[reverse]  # first and last printable ASCII
            data = ws.sorted('wikisource', 'prefix', reverse=reverse)
            with self.subTest(reverse=reverse):
                for entry in data:
                    code = entry['prefix']
                    if reverse:
                        self.assertGreater(last, code)
                    else:
                        self.assertLess(last, code)
                    last = code

    def test_sorting_order(self):
        """Test sorting order of languages_by_size."""
        family = 'wikipedia'
        ws = WikiStats()
        data = ws.get_dict(family)
        last = sys.maxsize
        last_code = ''
        for code in ws.languages_by_size(family):
            curr = int(data[code]['good'])
            self.assertGreaterEqual(
                last, curr,
                f'{code} ({curr}) is greater than {last_code} ({last}).')
            last = curr
            last_code = code

    def test_wikipedia(self):
        """Test WikiStats wikipedia data content."""
        ws = WikiStats()
        data = ws.get_dict('wikipedia')
        self.assertIsInstance(data, dict)
        self.assertIn('en', data)
        self.assertIn('ht', data)
        self.assertGreater(int(data['en']['total']), int(data['en']['good']))
        data = data['en']
        self.assertTrue(all(isinstance(key, str)
                            for key in data.keys() if key is not None))
        self.assertIsInstance(data['total'], str)
        self.assertIn('prefix', data)
        self.assertIn('total', data)

    def test_wikisource(self):
        """Test WikiStats wikisource data content."""
        ws = WikiStats()
        data = ws.get_dict('wikisource')
        self.assertIsInstance(data, dict)
        self.assertIn('en', data)
        self.assertIn('id', data)
        self.assertGreater(int(data['fr']['total']), int(data['fr']['good']))
        data = data['fr']
        self.assertTrue(all(isinstance(key, str)
                            for key in data.keys() if key is not None))
        self.assertIsInstance(data['total'], str)
        self.assertIn('prefix', data)
        self.assertIn('total', data)


if __name__ == '__main__':
    with suppress(SystemExit):
        unittest.main()