def test_find_by_unit(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Combat Engineer")
test_soldier.save_soldier()
found_soldier = Soldier.find_by_unit("7 Rifles")
self.assertEqual(found_soldier.unit, test_soldier.unit)
def test_find_soldier_by_svcNum(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Combat Engineer")
test_soldier.save_soldier()
found_soldier = Soldier.find_by_svcNumber("201030")
self.assertEqual(found_soldier.svc_num, test_soldier.svc_num)
def test_find_soldier_by_corps(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Gunner")
test_soldier.save_soldier()
found_soldier = Soldier.find_by_corps("Gunner")
self.assertEqual(found_soldier.corps, test_soldier.corps)
def test_delete_soldier(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Combat Engineer")
test_soldier.save_soldier()
self.new_soldier.delete_soldier()
self.assertEqual(len(Soldier.soldiers_list), 1)
def __init__(self, name, health, unit_type, operators, clock):
self.clock = clock
self.name = name
self._health = health
self.unit_type = unit_type
self.experience = 0 # 0 - 50
self.recharge = randint(1000, 2000)
self.end_recharge_time = 0
self.operators = []
for operator in operators:
self.operators.append(
Soldier(operator["name"], operator["health"],
operator["unit_type"], self.clock))
def add_soldiers_definitions(self, new_soldiers):
self.dictionaries_list = new_soldiers
for soldier in new_soldiers:
s = Soldier()
s.set_stats(soldier)
s.convert_to_fractions()
self.soldiers_list.append(s)
return None
def __init__(self, name, units, clock):
self.clock = clock
self.name = name
self.units = []
self._health = None
for unit in units:
if unit["unit_type"] == "soldier":
self.units.append(
Soldier(unit["name"], unit["health"], unit["unit_type"],
self.clock))
elif unit["unit_type"] == "vehicle":
self.units.append(
Vehicle(unit["name"], unit["health"], unit["unit_type"],
unit["operators"], self.clock))
self.strategy = ""
white = (255,255,255) black = (0,0,0) #------Actual Code------ grid = Grid(width,height,square_size) player_size = square_size soldier_size = square_size player_pos= [2,2] soldier1_pos = [0,0] soldier1 = Soldier(soldier1_pos) print soldier1.pos[0] soldier1.move(soldier1_pos,"right") print soldier1.pos[0] soldier1_x = grid.getCoor(soldier1_pos)[0] soldier1_y = grid.getCoor(soldier1_pos)[1] player_x = grid.getCoor(player_pos)[0] player_y = grid.getCoor(player_pos)[1] the_square = Square([100,100],"regular",True,100) def moveSoldier(direction):
def test_copy_svcNumber(self):
self.new_soldier.save_soldier()
Soldier.copy_svcNumber("102030")
self.assertEqual(self.new_soldier.svc_num, pyperclip.paste())
def list_soldiers():
return Soldier.display_soldiers()
def setUp(self):
self.new_soldier = Soldier("102030", "Private", "Karl", "3 Rifles", "Infantry")
def find_soldier(number):
return Soldier.find_by_svcNumber(number)
def test_soldier_exists(self):
self.new_soldier.save_soldier()
found_soldier = Soldier.check_if_exists("102030")
self.assertTrue(found_soldier)
class TestSoldier(unittest.TestCase):
def setUp(self):
self.new_soldier = Soldier("102030", "Private", "Karl", "3 Rifles", "Infantry")
def tearDown(self):
Soldier.soldiers_list = []
def test_init(self):
self.assertEqual(self.new_soldier.svc_num, "102030")
self.assertEqual(self.new_soldier.rank, "Private")
self.assertEqual(self.new_soldier.name, "Karl")
self.assertEqual(self.new_soldier.unit, "3 Rifles")
self.assertEqual(self.new_soldier.corps, "Infantry")
def test_save_soldier(self):
self.new_soldier.save_soldier()
def test_save_multiple_soldiers(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Combat Engineer")
test_soldier.save_soldier()
self.assertEqual(len(Soldier.soldiers_list), 2)
def test_delete_soldier(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Combat Engineer")
test_soldier.save_soldier()
self.new_soldier.delete_soldier()
self.assertEqual(len(Soldier.soldiers_list), 1)
def test_list_all_soldiers(self):
self.assertEqual(Soldier.soldiers_list, Soldier.display_soldiers())
def test_soldier_exists(self):
self.new_soldier.save_soldier()
found_soldier = Soldier.check_if_exists("102030")
self.assertTrue(found_soldier)
def test_find_soldier_by_svcNum(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Combat Engineer")
test_soldier.save_soldier()
found_soldier = Soldier.find_by_svcNumber("201030")
self.assertEqual(found_soldier.svc_num, test_soldier.svc_num)
def test_find_by_rank(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Combat Engineer")
test_soldier.save_soldier()
found_soldier = Soldier.find_by_rank("Corporal")
self.assertEqual(found_soldier.rank, test_soldier.rank)
def test_find_by_name(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Combat Engineer")
test_soldier.save_soldier()
found_soldier = Soldier.find_by_name("Keys")
self.assertEqual(found_soldier.name, test_soldier.name)
def test_find_by_unit(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Combat Engineer")
test_soldier.save_soldier()
found_soldier = Soldier.find_by_unit("7 Rifles")
self.assertEqual(found_soldier.unit, test_soldier.unit)
def test_find_soldier_by_corps(self):
self.new_soldier.save_soldier()
test_soldier = Soldier("201030", "Corporal", "Keys", "7 Rifles", "Gunner")
test_soldier.save_soldier()
found_soldier = Soldier.find_by_corps("Gunner")
self.assertEqual(found_soldier.corps, test_soldier.corps)
def test_copy_svcNumber(self):
self.new_soldier.save_soldier()
Soldier.copy_svcNumber("102030")
self.assertEqual(self.new_soldier.svc_num, pyperclip.paste())
def test_list_all_soldiers(self): self.assertEqual(Soldier.soldiers_list, Soldier.display_soldiers())
def create_soldier(svc_num, rank, name, unit, corps):
new_soldier = Soldier(svc_num, rank, name, unit, corps)
return new_soldier
def main():
print("Hello, welcome to Atlas Private Corporation Systems v1.0.0")
print("Please enter your name:")
print('\n')
user_name = input().capitalize()
while True:
print('\n')
print(
f"Hello {user_name}. Use the following keys to interact with the system: \n 1 - add a new soldier \n 2 - delete a soldier \n 3 - list all soldiers \n 4 - find a soldier \n 5 - copy a soldier's service number \n 6 - exit the system"
)
try:
short_code = int(input())
if short_code == 1:
print('\n')
print("Creating a new soldier...")
print("Enter the service number...")
try:
s_num = int(input())
except ValueError:
print("Please use numbers")
print("Enter the rank...")
s_rank = input().capitalize()
print("Enter the name...")
s_name = input().capitalize()
print("Enter the unit...")
s_unit = input().upper()
print("Enter the corps...")
s_corps = input().upper()
print('\n')
print(
f"New soldier added:\n name:{s_name}, service number: {s_num}"
)
save_soldier(
create_soldier(s_num, s_rank, s_name, s_unit, s_corps))
print('\n')
elif short_code == 2:
print('\n')
print("Option 2")
print('\n')
elif short_code == 3:
print('\n')
if list_soldiers():
print("Here is a list of all the soldiers:")
print('\n')
for soldier in list_soldiers():
print(
f"{soldier.svc_num} {soldier.rank} {soldier.name} {soldier.unit} {soldier.corps}"
)
print('\n')
else:
print("Sorry, there are no soldiers in this list yet")
print('\n')
elif short_code == 4:
print('\n')
print(
"Enter the service number of the soldier you wish to find")
try:
search_number = int(input())
except ValueError:
print("Please enter a number")
if check_soldier_exists(search_number):
found_soldier = Soldier.find_by_svcNumber(search_number)
print("Here's the soldier you're looking for...")
print(
f"{found_soldier.svc_num} {found_soldier.rank} {found_soldier.name} {found_soldier.unit} {found_soldier.corps}"
)
print('\n')
else:
print("Sorry, that soldier is not in our list")
elif short_code == 5:
print('\n')
print("Option 5")
print('\n')
elif short_code == 6:
print('\n')
print("existing the system...have a nice day")
print('\n')
break
else:
print('\n')
print("Please use the numbers 1 - 6")
print('\n')
except ValueError:
print('\n')
print("Please use the numbers 1 - 6")
print('\n')
def copy_soldier_svcNumber(number):
return Soldier.copy_svcNumber(number)
def check_soldier_exists(number):
return Soldier.check_if_exists(number)
