Skip to content

Commit

Permalink
More unit tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@gbtami
gbtami committed Dec 25, 2024
1 parent c2ee837 commit 9b84a00
Showing 1 changed file with 110 additions and 51 deletions.
161 changes: 110 additions & 51 deletions tests/test_auto_pairing.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,9 @@
from pychess_global_app_state_utils import get_app_state
from const import VARIANTS
from glicko2.glicko2 import DEFAULT_PERF
from auto_pair import add_to_auto_pairings
from auto_pair import add_to_auto_pairings, find_matching_seek, find_matching_user

ONE_TEST_ONLY = False

PERFS = {variant: DEFAULT_PERF for variant in VARIANTS}
HIGH_PERFS = {variant: {"gl": {"r": 2300}} for variant in VARIANTS}
Expand All @@ -34,6 +35,15 @@
"all": {"variants": ALL_VARIANT, "tcs": ALL_TC, "rrmin": -1000, "rrmax": 1000},
"chess": {"variants": [("chess", False)], "tcs": ALL_TC, "rrmin": -1000, "rrmax": 1000},
"chess960": {"variants": [("chess", True)], "tcs": ALL_TC, "rrmin": -1000, "rrmax": 1000},
"zh": {"variants": [("crazyhouse", False)], "tcs": ALL_TC, "rrmin": -1000, "rrmax": 1000},
"zh960": {"variants": [("crazyhouse", True)], "tcs": ALL_TC, "rrmin": -1000, "rrmax": 1000},
"koth": {"variants": [("kingofthehill", False)], "tcs": ALL_TC, "rrmin": -1000, "rrmax": 1000},
"koth960": {
"variants": [("kingofthehill", True)],
"tcs": ALL_TC,
"rrmin": -1000,
"rrmax": 1000,
},
"chess-500+500": {"variants": [("chess", False)], "tcs": ALL_TC, "rrmin": -500, "rrmax": 500},
"chess-800+800": {"variants": [("chess", False)], "tcs": ALL_TC, "rrmin": -800, "rrmax": 800},
}
Expand All @@ -50,10 +60,10 @@ async def startup(self, app):
self.eplayer = User(app_state, username="eplayer", perfs=PERFS)

# low rated player
self.lplayer = User(get_app_state(self.app), username="lplayer", perfs=LOW_PERFS)
self.lplayer = User(app_state, username="lplayer", perfs=LOW_PERFS)

# high rated player
self.hplayer = User(get_app_state(self.app), username="hplayer", perfs=HIGH_PERFS)
self.hplayer = User(app_state, username="hplayer", perfs=HIGH_PERFS)

app_state.users["aplayer"] = self.aplayer
app_state.users["bplayer"] = self.bplayer
Expand All @@ -71,6 +81,7 @@ async def get_application(self):
async def tearDownAsync(self):
await self.client.close()

@unittest.skipIf(ONE_TEST_ONLY, "1 test only")
def test_add_to_auto_pairings(self):
app_state = get_app_state(self.app)

Expand All @@ -94,6 +105,7 @@ def test_add_to_auto_pairings(self):
self.assertIsNone(matching_seek)
self.assertTrue(self.aplayer.ready_for_auto_pairing)

@unittest.skipIf(ONE_TEST_ONLY, "1 test only")
def remove_from_auto_pairings(self):
app_state = get_app_state(self.app)

Expand All @@ -105,76 +117,72 @@ def remove_from_auto_pairings(self):
self.assertNotIn(self.bplayer, app_state.auto_pairings[variant_tc])
self.assretFalse(self.bplayer.ready_for_auto_pairing)

@unittest.skipIf(ONE_TEST_ONLY, "1 test only")
def test_auto_compatible_with_seek0(self):
app_state = get_app_state(self.app)

seek = Seek("id", self.aplayer, "chess") # accepts any ratings

# auto_compatible_with_seek() checks rating ranges and blocked users only!
add_to_auto_pairings(get_app_state(self.app), self.bplayer, DATA["chess960"])
add_to_auto_pairings(app_state, self.bplayer, DATA["chess960"])
result = self.bplayer.auto_compatible_with_seek(seek)
self.assertTrue(result)

# auto_compatible_with_seek() checks rating ranges and blocked users only!
add_to_auto_pairings(get_app_state(self.app), self.cplayer, DATA["chess"])
add_to_auto_pairings(app_state, self.cplayer, DATA["chess"])
result = self.cplayer.auto_compatible_with_seek(seek)
self.assertTrue(result)

# now auto pairing creator blocks the seek creator
self.dplayer.blocked.add("aplayer")
add_to_auto_pairings(get_app_state(self.app), self.dplayer, DATA["chess960"])
add_to_auto_pairings(app_state, self.dplayer, DATA["chess960"])
result = self.dplayer.auto_compatible_with_seek(seek)
self.assertFalse(result)

# now seek creator blocks auto pairing creator
self.aplayer.blocked.add("eplayer")
add_to_auto_pairings(get_app_state(self.app), self.eplayer, DATA["chess960"])
add_to_auto_pairings(app_state, self.eplayer, DATA["chess960"])
result = self.eplayer.auto_compatible_with_seek(seek)
self.assertFalse(result)

@unittest.skipIf(ONE_TEST_ONLY, "1 test only")
def test_auto_compatible_with_seek1(self):
app_state = get_app_state(self.app)

# low rating player and auto pairing range is not OK
seek = Seek(
"id", self.lplayer, "chess", rrmin=-200, rrmax=0
) # low rated player, accepts 500-700
add_to_auto_pairings(
get_app_state(self.app), self.aplayer, DATA["chess-500+500"]
) # accepts 1000-2000
seek = Seek("id", self.lplayer, "chess", rrmin=-200, rrmax=0) # accepts 500-700
add_to_auto_pairings(app_state, self.aplayer, DATA["chess-500+500"]) # accepts 1000-2000
result = self.aplayer.auto_compatible_with_seek(seek)
self.assertFalse(result)

# low rating player and auto pairing range is OK
seek = Seek(
"id", self.lplayer, "chess", rrmin=-200, rrmax=900
) # low rated player, accepts 500-1500
add_to_auto_pairings(
get_app_state(self.app), self.bplayer, DATA["chess-800+800"]
) # accepts 700-2300
seek = Seek("id", self.lplayer, "chess", rrmin=-200, rrmax=900) # accepts 500-1500
add_to_auto_pairings(app_state, self.bplayer, DATA["chess-800+800"]) # accepts 700-2300
result = self.bplayer.auto_compatible_with_seek(seek)
self.assertTrue(result)

@unittest.skipIf(ONE_TEST_ONLY, "1 test only")
def test_auto_compatible_with_seek2(self):
app_state = get_app_state(self.app)

# high rating player and auto pairing ranges don't overlap
seek = Seek(
"id", self.hplayer, "chess", rrmin=-200, rrmax=200
) # high rated player, accepts 2100-2500
add_to_auto_pairings(
get_app_state(self.app), self.aplayer, DATA["chess-500+500"]
) # accepts 1000-2000
seek = Seek("id", self.hplayer, "chess", rrmin=-200, rrmax=200) # accepts 2100-2500
add_to_auto_pairings(app_state, self.aplayer, DATA["chess-500+500"]) # accepts 1000-2000
result = self.aplayer.auto_compatible_with_seek(seek)
self.assertFalse(result)

# high rating player and auto pairing ranges are overlapping
seek = Seek(
"id", self.hplayer, "chess", rrmin=-900, rrmax=200
) # high rated player, accepts 1400-2500
add_to_auto_pairings(
get_app_state(self.app), self.bplayer, DATA["chess-800+800"]
) # accepts 700-2300
seek = Seek("id", self.hplayer, "chess", rrmin=-900, rrmax=200) # accepts 1400-2500
add_to_auto_pairings(app_state, self.bplayer, DATA["chess-800+800"]) # accepts 700-2300
result = self.bplayer.auto_compatible_with_seek(seek)
self.assertTrue(result)

@unittest.skipIf(ONE_TEST_ONLY, "1 test only")
def test_auto_compatible_with_other_user0(self):
add_to_auto_pairings(get_app_state(self.app), self.aplayer, DATA["chess"])
add_to_auto_pairings(get_app_state(self.app), self.bplayer, DATA["chess960"])
app_state = get_app_state(self.app)

add_to_auto_pairings(app_state, self.aplayer, DATA["chess"])
add_to_auto_pairings(app_state, self.bplayer, DATA["chess960"])

# auto_compatible_with_other_user() checks rating ranges and blocked users only!
result = self.aplayer.auto_compatible_with_other_user(self.bplayer, "chess", False)
Expand All @@ -184,16 +192,13 @@ def test_auto_compatible_with_other_user0(self):
result = self.aplayer.auto_compatible_with_other_user(self.bplayer, "chess", True)
self.assertTrue(result)

@unittest.skipIf(ONE_TEST_ONLY, "1 test only")
def test_auto_compatible_with_other_user1(self):
add_to_auto_pairings(
get_app_state(self.app), self.lplayer, DATA["chess"]
) # accepts any range
add_to_auto_pairings(
get_app_state(self.app), self.aplayer, DATA["chess-500+500"]
) # accepts 1000-2000
add_to_auto_pairings(
get_app_state(self.app), self.bplayer, DATA["chess-800+800"]
) # accepts 700-2300
app_state = get_app_state(self.app)

add_to_auto_pairings(app_state, self.lplayer, DATA["chess"]) # accepts any range
add_to_auto_pairings(app_state, self.aplayer, DATA["chess-500+500"]) # accepts 1000-2000
add_to_auto_pairings(app_state, self.bplayer, DATA["chess-800+800"]) # accepts 700-2300

# low rating player and default player
result = self.lplayer.auto_compatible_with_other_user(self.aplayer, "chess", False)
Expand All @@ -203,16 +208,13 @@ def test_auto_compatible_with_other_user1(self):
result = self.lplayer.auto_compatible_with_other_user(self.bplayer, "chess", False)
self.assertTrue(result)

@unittest.skipIf(ONE_TEST_ONLY, "1 test only")
def test_auto_compatible_with_other_user2(self):
add_to_auto_pairings(
get_app_state(self.app), self.hplayer, DATA["chess"]
) # accepts any range
add_to_auto_pairings(
get_app_state(self.app), self.aplayer, DATA["chess-500+500"]
) # accepts 1000-2000
add_to_auto_pairings(
get_app_state(self.app), self.bplayer, DATA["chess-800+800"]
) # accepts 700-2300
app_state = get_app_state(self.app)

add_to_auto_pairings(app_state, self.hplayer, DATA["chess"]) # accepts any range
add_to_auto_pairings(app_state, self.aplayer, DATA["chess-500+500"]) # accepts 1000-2000
add_to_auto_pairings(app_state, self.bplayer, DATA["chess-800+800"]) # accepts 700-2300

# high rating player and default player
result = self.hplayer.auto_compatible_with_other_user(self.aplayer, "chess", False)
Expand All @@ -222,6 +224,63 @@ def test_auto_compatible_with_other_user2(self):
result = self.hplayer.auto_compatible_with_other_user(self.bplayer, "chess", False)
self.assertTrue(result)

@unittest.skipIf(ONE_TEST_ONLY, "1 test only")
def test_find_matching_seek(self):
app_state = get_app_state(self.app)

seek0 = Seek("id0", self.aplayer, "chess")
seek1 = Seek("id1", self.bplayer, "zh")
seek2 = Seek("id2", self.cplayer, "koth")

app_state.seeks[seek0.id] = seek0
app_state.seeks[seek1.id] = seek1
app_state.seeks[seek2.id] = seek2

# There is no 960 in seeks
variant_tc = ("chess", True, 5, 3, 0) # chess960 5+3
add_to_auto_pairings(app_state, self.lplayer, DATA["chess960"])
result = find_matching_seek(app_state, self.lplayer, variant_tc)
self.assertIsNone(result)

self.lplayer.remove_from_auto_pairings()

# There is no rated seek in seeks
variant_tc = ("chess", False, 5, 3, 0) # chess 5+3
add_to_auto_pairings(app_state, self.lplayer, DATA["chess"])
result = find_matching_seek(app_state, self.lplayer, variant_tc)
self.assertIsNone(result)

self.lplayer.remove_from_auto_pairings()

seek3 = Seek("id3", self.dplayer, "chess", rated=True)
app_state.seeks[seek3.id] = seek3

variant_tc = ("chess", False, 5, 3, 0) # chess 5+3
add_to_auto_pairings(app_state, self.lplayer, DATA["chess"])
result = find_matching_seek(app_state, self.lplayer, variant_tc)
self.assertEqual(result, seek3)

@unittest.skipIf(ONE_TEST_ONLY, "1 test only")
def test_find_matching_user(self):
app_state = get_app_state(self.app)

add_to_auto_pairings(app_state, self.aplayer, DATA["chess"])
add_to_auto_pairings(app_state, self.bplayer, DATA["chess960"])
add_to_auto_pairings(app_state, self.cplayer, DATA["zh"])
add_to_auto_pairings(app_state, self.dplayer, DATA["zh960"])

variant_tc = ("kingofthehill", False, 5, 3, 0) # koth 5+3
add_to_auto_pairings(app_state, self.lplayer, DATA["koth"])
result = find_matching_user(app_state, self.lplayer, variant_tc)
self.assertIsNone(result)

self.lplayer.remove_from_auto_pairings()

variant_tc = ("chess", False, 5, 3, 0) # chess 5+3
add_to_auto_pairings(app_state, self.lplayer, DATA["chess"])
result = find_matching_user(app_state, self.lplayer, variant_tc)
self.assertEqual(result, self.aplayer)


if __name__ == "__main__":
unittest.main(verbosity=2)

0 comments on commit 9b84a00

Please sign in to comment.