def test_load_mtgjson(sets_data):
isd_mtg_data = {k: v for k, v in sets_data.items() if k == "ISD"}
cdb = card_db.CardDb()
cdb.load_mtg_json(isd_mtg_data)
name_to_name = {n: c.name for n, c in cdb.name_to_card.items()}
assert name_to_name == {
"Abattoir Ghoul": "Abattoir Ghoul",
"Delver of Secrets": "Delver of Secrets",
"Insectile Aberration": "Insectile Aberration",
"Forest": "Forest",
}
code_to_setname = {c: s.name for c, s in cdb.code_to_card_set.items()}
assert code_to_setname == {"ISD": "Innistrad"}
setname_to_setcode = {
n: s.code
for n, s in cdb.setname_to_card_set.items()
}
assert setname_to_setcode == {"Innistrad": "ISD"}
id_to_setnum = {i: p.set_number for i, p in cdb.id_to_printing.items()}
assert id_to_setnum == {
"0b06d8d9e7662ada82bd29e1138d959978ba2280": "51a",
"e5c4aa9a443c346ccbf8d99c9320138827065e05": "51b",
"958ae1416f8f6287115ccd7c5c61f2415a313546": "85",
"e9abef8533c9ce6549147232c5fceff75ffb460a": "262",
"3c509643d7f7827b2debf968c05cb800cb772360": "263",
"d5dbd9b201a515d119b424b3d7b06dcf30a5c675": "264",
}
def test_index_sorting(sets_data):
cdb = card_db.CardDb()
cdb.load_mtg_json(sets_data)
cdb.rebuild_indexes()
forest_set_and_mvid = [(p.set_code, p.multiverseid)
for p in cdb.card_name_to_printings["Forest"]]
assert forest_set_and_mvid == [
("ICE", 2746),
("ICE", 2747),
("ICE", 2748),
("ISD", 245247),
("ISD", 245248),
("ISD", 245246),
("LEA", 288),
("LEA", 289),
]
isd_set_numbers = [p.set_number for p in cdb.set_code_to_printings["ISD"]]
assert isd_set_numbers == ["51a", "51b", "85", "262", "263", "264"]
assert cdb.cards[0].name == "Abattoir Ghoul"
assert cdb.cards[-1].name == "Wastes"
assert cdb.card_sets[0].code == "LEA"
assert cdb.card_sets[-1].code == "W16"
isd_dos = cdb.id_to_printing["0b06d8d9e7662ada82bd29e1138d959978ba2280"]
assert cdb.set_code_to_printing_to_row["ISD"][isd_dos] == 0
isd_ag = cdb.id_to_printing["958ae1416f8f6287115ccd7c5c61f2415a313546"]
assert cdb.set_code_to_printing_to_row["ISD"][isd_ag] == 2
def test_rebuild_indexes(sets_data):
cdb = card_db.CardDb()
cdb.load_mtg_json(sets_data)
cdb.rebuild_indexes()
isd_ids = {p.id_ for p in cdb.set_code_to_printings["ISD"]}
assert isd_ids == {
"0b06d8d9e7662ada82bd29e1138d959978ba2280",
"e5c4aa9a443c346ccbf8d99c9320138827065e05",
"958ae1416f8f6287115ccd7c5c61f2415a313546",
"e9abef8533c9ce6549147232c5fceff75ffb460a",
"3c509643d7f7827b2debf968c05cb800cb772360",
"d5dbd9b201a515d119b424b3d7b06dcf30a5c675",
}
darkrit_ids = {p.id_ for p in cdb.card_name_to_printings["Dark Ritual"]}
assert darkrit_ids == {
"fff0b8e8fea06ee1ac5c35f048a0a459b1222673",
"2fab0ea29e3bbe8bfbc981a4c8163f3e7d267853",
"19c38ff78c0e98b38f3bd8184478e22152d9a624",
}
snnm_keys = [
("ISD", "Abattoir Ghoul", "85", 222911),
("ISD", "Abattoir Ghoul", None, 222911),
("ISD", "Abattoir Ghoul", "85", None),
("ISD", "Abattoir Ghoul", None, None),
]
for snnm_key in snnm_keys:
found_print_ids = {
p.id_
for p in cdb.set_name_num_mv_to_printings[snnm_key]
}
assert found_print_ids == {"958ae1416f8f6287115ccd7c5c61f2415a313546"}
def build_card_db(data_path, include_online_only):
"""Get a card_db with current mtgjson data."""
try:
mtgjson.fetch_mtgjson(data_path)
except mtgjson.DownloadError:
print("Failed to fetch mtgjson data, attempting to use cached data.")
print("Reading mtgjson data.")
mtgjsondata = mtgjson.read_mtgjson(data_path)
return card_db.CardDb(mtgjsondata, include_online_only=include_online_only)
def test_sets_loaded(sets_data, online_only, expected):
cdb = card_db.CardDb()
cdb.load_mtg_json(sets_data, include_online_only=online_only)
assert set(cdb.code_to_card_set.keys()) == expected
def cdb(sets_data):
"""Fixture card_db with mtg json data."""
return card_db.CardDb(sets_data)
def cdb(sets_data):
"""card_db fixture for testing."""
return card_db.CardDb(sets_data)
def cdb(sets_data):
"""card_db fixture for testing."""
sets_data = {
k: v for k, v in sets_data.items() if k in {"LEA", "FEM", "S00", "ICE", "HOP"}
}
return card_db.CardDb(sets_data)
def cdb(sets_data):
"""card_db fixture for testing."""
sets_data = {k: v for k, v in sets_data.items() if k in {"MMA", "pMGD"}}
return card_db.CardDb(sets_data)
