testing.py

import unittest
import calcs
import aiounittest

class TestDiscordBotProcessInputsFuncs(aiounittest.AsyncTestCase):

    async def test_rating_is_numeric(self):
        self.assertEqual(await calcs.process_inputs("nihalsarin2004", "wrong", "Rapid",testing=True), 
        (True, "Error: rating must be a positive integer between 500 and 3200.", None, None))

    async def test_rating_is_below_3200(self):
        self.assertEqual(await calcs.process_inputs("ZackAttack614", "3201", "rapid",testing=True), 
        (True, "Error: rating must be a positive integer between 500 and 3200.", None, None))

    async def test_variant_type(self):
        self.assertEqual(await calcs.process_inputs("johndavis_59", "2000", "Puzzles",testing=True), 
        (True, "Error: variant not supported. Try bullet, blitz, rapid, or classical.",None,None))

    async def test_variant_type_2(self):
        self.assertEqual(await calcs.process_inputs("HumanSponge", "2000", "fun_chess_variant_5",testing=True), 
        (True, "Error: variant not supported. Try bullet, blitz, rapid, or classical.",None,None))

    async def test_default_variant_is_rapid(self):
        self.assertEqual(await calcs.process_inputs("EricRosen", "2100",testing=True), 
         await calcs.process_inputs("EricRosen", "2100", "rapid",testing=True))

    async def test_switching_inputs(self):
        self.assertEqual(await calcs.process_inputs("2000", "milocannestra",testing=True), 
        (True,"Error: rating must be a positive integer between 500 and 3200.",None,None))

    async def test_negative_target_rating(self):
        self.assertEqual(await calcs.process_inputs("grandmastergauri", "-1000",testing=True), 
        (True,"Error: rating must be a positive integer between 500 and 3200.",None,None))

    async def test_decimal_target_rating(self):
        self.assertEqual(await calcs.process_inputs("EricRosen", "2800.8",testing=True), 
        (True,"Error: rating must be a positive integer between 500 and 3200.",None,None))

class TestDiscordBotScoreFuncs(aiounittest.AsyncTestCase):
    async def test_variant_in_user_history(self):
        self.assertEqual(await calcs.score("milocannestra", 2000, "classical", calcs.model_params,testing=True), 
        (True,"Error: user milocannestra has no rating history for variant Classical.", None, None))

    async def test_user_hit_rating(self):
        self.assertEqual(await calcs.score("penguingim1", 1000, "Blitz", calcs.model_params,testing=True), 
        (True,"Error: penguingim1 has already achieved the target rating 1000 Blitz.", None, None))

    async def test_target_less_than_1000_plus_current_rating(self):
        self.assertEqual(await calcs.score("KrnAmericanChessNoob", 2900, "Bullet", calcs.model_params,testing=True), 
        (True,"Error: please submit a target rating gain of less than +1000 points.", None, None))

    async def test_user_doesnt_exist(self):
        self.assertEqual(await calcs.score("rwafrsdsgga", 3000, "Rapid", calcs.model_params,testing=True), 
        (True,"Error: can't retrieve lichess data for user rwafrsdsgga.", None, None))
    
    async def test_limited_data_warning(self):
        error_bool,error_msg,prob_success,predicted_date = await calcs.score("penguingim1",2600,"Rapid",calcs.model_params,testing=True)
        self.assertEqual(error_msg,"Warning: results may be unreliable due to limited Rapid data for user penguingim1.")

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