test_requests.py

from copy import deepcopy
from urllib.parse import urlencode

import requests

SERVER = 'http://127.0.0.1:8000'


def main():

    params = {
              # 'guesser': guesser_type,
              # 'contexts[]': [' '.join(context) for context in context_1],
              'no_of_subwords': 1,
              'prev_guesses[]': [],
              'retry_wrong': False,
              'top_n': 10
             }

    ####################################################################################################################

    missing_word = 9 * '#'
    context_1 = [
        ['b)', 'az', 'adatállományok', 'helyreállításának', 'lehetőségét', missing_word, 'intézkedésekről', ',', 'ezen',
         'belül', 'a']]

    params_1 = deepcopy(params)
    params_1['contexts[]'] = [' '.join(context) for context in context_1]
    params_1['missing_token'] = missing_word

    for guesser_type in ('bert', 'cbow'):
        params_1['guesser'] = guesser_type
        output_1 = requests.post(f'{SERVER}/guess', json=params_1)

        print('POST', guesser_type, output_1)
        print(output_1.json())

    for guesser_type in ('bert', 'cbow'):
        params_1['guesser'] = guesser_type
        output_2 = requests.get(f'{SERVER}/guess?{urlencode(params_1, doseq=True)}')

        print('GET', guesser_type, output_2)
        print(output_2.json())

    ####################################################################################################################

    missing_word = 9 * '#'
    context_2 = [
        ['Nyugalmuk', 'sokszor', 'békés', 'alvásba', missing_word, ',', 'rengeteget', 'pihennek', ',', 'komótos'],
        ['szélén', '–', 'ahol', 'rengeteg', 'kullancs', missing_word, 'elő', ',', 'akkor', 'lehet', ',']]

    previous_guesses_2 = ['került', 'fordul']

    params_2 = deepcopy(params)
    params_2['contexts[]'] = [' '.join(context) for context in context_2]
    params_2['prev_guesses[]'] = previous_guesses_2
    params_2['missing_token'] = missing_word

    for guesser_type in ('bert', 'cbow'):
        params_2['guesser'] = guesser_type
        output_3 = requests.post(f'{SERVER}/guess', json=params_2)

        print('POST', guesser_type, output_3)
        print(output_3.json())

    for guesser_type in ('bert', 'cbow'):
        params_2['guesser'] = guesser_type
        output_4 = requests.get(f'{SERVER}/guess?{urlencode(params_2, doseq=True)}')

        print('GET', guesser_type, output_4)
        print(output_4.json())

    ####################################################################################################################

    params_3 = {'word': 'Kormányok'}

    for guesser_type in ('bert', 'cbow'):
        params_3['guesser'] = guesser_type
        output_5 = requests.post(f'{SERVER}/no_of_subwords', json=params_3)

        print('POST', guesser_type, output_5)
        print(output_5.json())

    for guesser_type in ('bert', 'cbow'):
        params_3['guesser'] = guesser_type
        output_6 = requests.get(f'{SERVER}/no_of_subwords?{urlencode(params_3, doseq=True)}')

        print('GET', guesser_type, output_6)
        print(output_6.json())

    ####################################################################################################################

    params_4 = {'word1': 'alma', 'word2': 'körte'}

    for guesser_type in ('bert', 'cbow'):
        params_4['guesser'] = guesser_type
        output_7 = requests.post(f'{SERVER}/word_similarity', json=params_4)

        print('POST', guesser_type, output_7)
        print(output_7.json())

    for guesser_type in ('bert', 'cbow'):
        params_4['guesser'] = guesser_type
        output_8 = requests.get(f'{SERVER}/word_similarity?{urlencode(params_4, doseq=True)}')

        print('GET', guesser_type, output_8)
        print(output_8.json())


if __name__ == '__main__':
    main()