Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Resolves #73 #82

Merged
merged 11 commits into from
Jul 17, 2024
Merged

Resolves #73 #82

Show file tree
Hide file tree
Changes from 7 commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
51cd0c4
chore(issue-73): Added initial Terran State class
RickFqt Mar 19, 2024
e90542f
refactor: Adjusted to reviewer's suggestions
RickFqt Apr 18, 2024
cbe7fc7
refactor: Changed TerranState to work with raw units
RickFqt Jun 21, 2024
e1c2dd6
test: Added test for sc2 terran state
RickFqt Jul 2, 2024
b1d21e4
feat: Added protoss and zerg states
RickFqt Jul 2, 2024
373aede
refactor: Added constants to improve readability
RickFqt Jul 5, 2024
98166c9
refactor: Added constants file
RickFqt Jul 11, 2024
94263bc
refactor: Changed files organization
RickFqt Jul 12, 2024
4d2f055
refactor: Changed units amount functions
RickFqt Jul 12, 2024
9858508
feat: Added starcraft2 general state class
RickFqt Jul 16, 2024
ca80acf
refactor: Removed magic numbers from terran state
RickFqt Jul 16, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Empty file added tests/units/sc2/states/__init__.py
View file
Open in desktop
Empty file.
218 changes: 218 additions & 0 deletions tests/units/sc2/states/test_aux_methods_sc2_state.py
View file
Open in desktop
RickFqt marked this conversation as resolved.
Show resolved Hide resolved
Original file line number Diff line number Diff line change
@@ -0,0 +1,218 @@
import unittest

from pysc2.lib.named_array import NamedDict

from urnai.sc2.states.states_utils import (
append_player_and_enemy_grids,
get_my_raw_units_amount,
get_raw_units_by_type,
)


class TestAuxSC2State(unittest.TestCase):

def test_append_player_and_enemy_grids(self):
# GIVEN
obs = NamedDict({
'raw_units': [
NamedDict({
'unit_type': 1,
'alliance': 1,
'build_progress': 100,
'x': 1,
'y': 1,
}),
NamedDict({
'unit_type': 2,
'alliance': 1,
'build_progress': 100,
'x': 2,
'y': 2,
}),
NamedDict({
'unit_type': 2,
'alliance': 4,
'build_progress': 100,
'x': 63,
'y': 63,
}),
]
})
new_state = []
# WHEN
new_state = append_player_and_enemy_grids(obs, new_state, 3, 64)
# THEN
print(new_state)
assert len(new_state) == (18)
assert new_state[8] == 0.005
assert new_state[9] == 0.01

def test_get_my_raw_units_amount(self):
# GIVEN
obs = NamedDict({
'raw_units': [
NamedDict({
'unit_type': 1,
'alliance': 1,
'build_progress': 100,
'x': 1,
'y': 1,
}),
NamedDict({
'unit_type': 2,
'alliance': 1,
'build_progress': 100,
'x': 2,
'y': 2,
}),
]
})
# WHEN
amount = get_my_raw_units_amount(obs, 1)
# THEN
assert amount == 1

def test_get_my_raw_units_amount_no_units(self):
# GIVEN
obs = NamedDict({
'raw_units': []
})
# WHEN
amount = get_my_raw_units_amount(obs, 1)
# THEN
assert amount == 0

def test_get_my_raw_units_amount_no_units_of_type(self):
# GIVEN
obs = NamedDict({
'raw_units': [
NamedDict({
'unit_type': 2,
'alliance': 1,
'build_progress': 100,
'x': 1,
'y': 1,
}),
NamedDict({
'unit_type': 2,
'alliance': 1,
'build_progress': 100,
'x': 2,
'y': 2,
}),
]
})
# WHEN
amount = get_my_raw_units_amount(obs, 1)
# THEN
assert amount == 0

def test_get_my_raw_units_amount_no_units_of_alliance(self):
# GIVEN
obs = NamedDict({
'raw_units': [
NamedDict({
'unit_type': 1,
'alliance': 2,
'build_progress': 100,
'x': 1,
'y': 1,
}),
NamedDict({
'unit_type': 1,
'alliance': 2,
'build_progress': 100,
'x': 2,
'y': 2,
}),
]
})
# WHEN
amount = get_my_raw_units_amount(obs, 1)
# THEN
assert amount == 0

def test_get_my_raw_units_amount_no_units_of_build_progress(self):
# GIVEN
obs = NamedDict({
'raw_units': [
NamedDict({
'unit_type': 1,
'alliance': 1,
'build_progress': 50,
'x': 1,
'y': 1,
}),
NamedDict({
'unit_type': 1,
'alliance': 1,
'build_progress': 50,
'x': 2,
'y': 2,
}),
]
})
# WHEN
amount = get_my_raw_units_amount(obs, 1)
# THEN
assert amount == 0

def test_get_raw_units_by_type(self):
# GIVEN
obs = NamedDict({
'raw_units': [
NamedDict({
'unit_type': 1,
'alliance': 1,
'build_progress': 100,
'x': 1,
'y': 1,
}),
NamedDict({
'unit_type': 2,
'alliance': 1,
'build_progress': 100,
'x': 2,
'y': 2,
}),
]
})
# WHEN
units = get_raw_units_by_type(obs, 1)
# THEN
assert len(units) == 1

def test_get_raw_units_by_type_no_units(self):
# GIVEN
obs = NamedDict({
'raw_units': []
})
# WHEN
units = get_raw_units_by_type(obs, 1)
# THEN
assert len(units) == 0

def test_get_raw_units_by_type_no_units_of_type(self):
# GIVEN
obs = NamedDict({
'raw_units': [
NamedDict({
'unit_type': 2,
'alliance': 1,
'build_progress': 100,
'x': 1,
'y': 1,
}),
NamedDict({
'unit_type': 2,
'alliance': 1,
'build_progress': 100,
'x': 2,
'y': 2,
}),
]
})
# WHEN
units = get_raw_units_by_type(obs, 1)
# THEN
assert len(units) == 0
Loading