def test_define_card_object(self):
header = [
'name', 'race_type', 'attribute', 'level', 'attack', 'defense',
'edition', 'set_number', 'pass_code', 'condition'
]
indexes = DataImport.init_indexes(header)
card_info = [
'Blue-Eyes White Dragon', 'Dragon', 'Light', 8, 3000, 2500, None,
'SDK-001', 89631139, 'LP'
]
card_object = DataImport.define_card_object(indexes, card_info)
predefined_set_object = SetInfo(set_number='SDK-001',
conditions={
'NM': 0,
'LP': 1,
'MP': 0,
'HP': 0,
'D': 0
})
predefined_card_object = Card(name='Blue-Eyes White Dragon',
race='Dragon',
monster_type='Normal Monster',
attribute='Light',
level=8,
attack=3000,
defense=2500,
pass_code=89631139,
set_info=[predefined_set_object])
self.assertEqual(card_object, predefined_card_object)
def main():
print("1. Import Data")
print("2. Search")
print("3. Cancel")
value = input("What would you like to do?\n")
if value == '1':
# Only allow fresh collection to import files.
num_records = DatabaseHandler.count_number_records()
if num_records == 0:
files: List[str] = DATA_FILES.split(',')
for file in files:
DataImport.import_data(DATA_PATH + file)
else:
logger.error('Number of records in database is ' +
str(num_records) + '. Should be 0 to import files.')
elif value == '2':
pass
elif value == '3':
pass
elif value == 'deletealldata':
DatabaseHandler.delete_all_records()
try:
shutil.rmtree(PATH_IMG_DOWNLOAD)
logger.info('Deleted path: ' + PATH_IMG_DOWNLOAD)
except OSError as ose:
logger.error(ose)
pass
def test_sets_partial_match(self):
set1: SetInfo = SetInfo(set_number='STON-ENSE1', edition='LE')
set2: SetInfo = SetInfo(set_number='STON-ENSE1', edition='FE')
result = DataImport.do_sets_match(set1, set2)
self.assertFalse(result)
set1: SetInfo = SetInfo(set_number='STON-ENSE1', edition='FE')
set2: SetInfo = SetInfo(set_number='DP04-EN012', edition='FE')
result = DataImport.do_sets_match(set1, set2)
self.assertFalse(result)
def main():
if action == '-s':
result = DatabaseHandler.execute_query({'name': arg})
print(result)
elif action == '-i':
# Only allow fresh collection to import files.
num_records = DatabaseHandler.count_number_records()
if num_records == 0:
files: List[str] = DATA_FILES.split(',')
for file in files:
DataImport.import_data(DATA_PATH + file)
else:
logger.error('Number of records in database is ' +
str(num_records) + '. Should be 0 to import files.')
def test_similar_sets(self):
test_hsi = TestHandleSetInfo
set_info1 = test_hsi._test_set_info
set_info1.set_conditions({'NM': 0, 'LP': 0, 'MP': 0, 'HP': 1, 'D': 0})
card1 = test_hsi._test_card
card1.set_set_info([set_info1])
set_info2 = test_hsi._test_set_info
set_info2.set_conditions({'NM': 1, 'LP': 1, 'MP': 0, 'HP': 0, 'D': 0})
card2 = test_hsi._test_card
card2.set_set_info([set_info2])
result_card = DataImport.handle_set_info(card1, card2)
expected_set_info = test_hsi._test_set_info
expected_set_info.set_conditions({
'NM': 1,
'LP': 1,
'MP': 0,
'HP': 1,
'D': 0
})
expected_card = test_hsi._test_card
expected_card.set_set_info([expected_set_info])
self.assertEqual(result_card, expected_card)
def test_card_not_exists(self):
new_card = Card(name='This is a card')
card_list: List[Card] = []
card_list.append(Card(name='This is another card'))
card_list.append(Card(name='This is yet another card'))
card_list.append(Card(name='This is a final card'))
result = DataImport.index_of_existing_data(new_card, card_list)
self.assertEqual(result, None)
def test_init_indexes_not_registered(self):
header = [
'name', 'race_type', 'attributes', 'level', 'attack', 'defense',
'pass_code', 'description', 'set_number', 'edition', 'rarity',
'condition'
]
result = DataImport.init_indexes(header)
self.assertEqual(result.get('attributes'), None)
self.assertEqual(result.get('rarity'), 10)
def test_race_and_no_type(self):
race_type = 'Machine'
result = DataImport.get_monster_race_and_type(race_type)
expected_result: dict = {
'race': 'Machine',
'monster_type': 'Normal Monster'
}
self.assertEqual(result, expected_result)
def test_race_and_multiple_types(self):
race_type = 'Machine / Union / Effect'
result = DataImport.get_monster_race_and_type(race_type)
expected_result: dict = {
'race': 'Machine',
'monster_type': 'Union Effect Monster'
}
self.assertEqual(result, expected_result)
def test_different_sets(self):
test_hsi = TestHandleSetInfo
test_set_info2 = test_hsi._test_set_info
test_set_info2.set_edition('FE')
test_card2 = test_hsi._test_card
test_card2.set_set_info([test_set_info2])
result_card = DataImport.handle_set_info(test_hsi._test_card,
test_card2)
expected_card = test_hsi._test_card
expected_card.set_set_info([test_set_info2, test_hsi._test_set_info])
self.assertEqual(result_card, expected_card)
def test_same_sets(self):
test_hsi = TestHandleSetInfo
result_card = DataImport.handle_set_info(test_hsi._test_card,
test_hsi._test_card)
expected_set_info = test_hsi._test_set_info
expected_set_info.set_conditions({
'NM': 2,
'LP': 0,
'MP': 0,
'HP': 0,
'D': 0
})
expected_card = test_hsi._test_card
expected_card.set_set_info([expected_set_info])
self.assertEqual(result_card, expected_card)
def test_same_set_different_editions(self):
test_hsi = TestHandleSetInfo
set_info1 = test_hsi._test_set_info
card1 = test_hsi._test_card
set_info2 = test_hsi._test_set_info
set_info2.set_conditions({'NM': 0, 'LP': 0, 'MP': 0, 'HP': 0, 'D': 1})
set_info2.set_edition(None)
card2 = test_hsi._test_card
card2.set_set_info([set_info2])
result_card = DataImport.handle_set_info(card1, card2)
expected_card = test_hsi._test_card
expected_card.set_set_info([set_info2, set_info1])
self.assertEqual(result_card, expected_card)
def test_sets_do_match(self):
set1: SetInfo = SetInfo(set_number='CDIP-EN035', edition='FE')
set2: SetInfo = SetInfo(set_number='CDIP-EN035', edition='FE')
result = DataImport.do_sets_match(set1, set2)
self.assertTrue(result)
def test_handle_conditions_not_exist(self):
conditions: dict = {'NM': 2, 'D': 0}
new_condition = 'RE'
updated_conditions = DataImport.handle_conditions(
conditions, new_condition)
self.assertEqual(updated_conditions, conditions)
def test_handle_conditions_add(self):
conditions: dict = {'NM': 2, 'D': 0}
new_condition = 'D'
updated_conditions = DataImport.handle_conditions(
conditions, new_condition)
self.assertEqual(updated_conditions, {'NM': 2, 'D': 1})
def test_key_is_not_found(self):
dictionary: dict = {'NM': 1, 'LP': 0, 'MP': 0, 'HP': 0, 'D': 0}
key = DataImport.get_key_from_value(dictionary, 3)
self.assertEqual(key, None)
