BLKSerene/Wordless

# ----------------------------------------------------------------------
# Wordless: Tests - Figures - Statistics
# Copyright (C) 2018-2024  Ye Lei (叶磊)
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
# ----------------------------------------------------------------------

import random

from tests import wl_test_init
from wordless.wl_figs import wl_figs_stats
from wordless.wl_nlp import wl_texts

def test_wl_fig_stats():
    main = wl_test_init.Wl_Test_Main(switch_lang_utils = 'fast')

    for tab in [
        'wordlist_generator',
        'ngram_generator',
        'collocation_extractor',
        'colligation_extractor',
        'keyword_extractor'
    ]:
        print(f'[{tab}]')

        if tab in ['wordlist_generator', 'ngram_generator', 'keyword_extractor']:
            graph_types = ['Line Chart', 'Word Cloud']
        elif tab in ['collocation_extractor', 'colligation_extractor']:
            graph_types = ['Line Chart', 'Word Cloud', 'Network Graph']

        if tab in ['wordlist_generator', 'ngram_generator']:
            use_datas = ["Juilland's D", "Juilland's U"]
        elif tab in ['collocation_extractor', 'colligation_extractor', 'keyword_extractor']:
            use_datas = ['Test Statistic', 'p-value', 'Bayes Factor', 'Effect Size']

        for graph_type in graph_types:
            stat_files_items = {}

            use_data = random.choice(use_datas)

            if use_data == 'p-value':
                val_max = 1
            else:
                val_max = 100

            if graph_type == 'Network Graph':
                for node in range(10):
                    node = wl_texts.Wl_Token(str(node))

                    for collocate in range(10):
                        collocate = wl_texts.Wl_Token(str(collocate))
                        stat_files_items[(node, collocate)] = [
                            random.uniform(0, val_max),
                            random.uniform(0, val_max),
                            random.uniform(0, val_max)
                        ]
            else:
                for item in range(100):
                    item = wl_texts.Wl_Token(str(item))
                    stat_files_items[item] = [
                        random.uniform(0, val_max),
                        random.uniform(0, val_max),
                        random.uniform(0, val_max)
                    ]

            wl_test_init.select_test_files(main, no_files = [0, 1])

            fig_settings = main.settings_custom[tab]['fig_settings']
            fig_settings['graph_type'] = graph_type
            fig_settings['sort_by_file'] = random.choice([*main.wl_file_area.get_selected_file_names(), 'Total'])
            fig_settings['use_data'] = use_data
            fig_settings['use_cumulative'] = random.choice([True, False])
            fig_settings['use_pct'] = random.choice([True, False])

            print(f"Files: {' | '.join(wl_test_init.get_test_file_names(main))}")
            print(f"Graph type: {fig_settings['graph_type']}")
            print(f"Sort by file: {fig_settings['sort_by_file']}")
            print(f"Use data: {fig_settings['use_data']}")
            print(f"Use cumulative data: {fig_settings['use_cumulative']}")
            print(f"Use percentage data: {fig_settings['use_pct']}\n")

            wl_figs_stats.wl_fig_stats(
                main,
                stat_files_items = stat_files_items,
                tab = tab
            )

if __name__ == '__main__':
    test_wl_fig_stats()