mirror of
https://github.com/LazoCoder/Pokemon-Terminal
synced 2024-11-27 14:21:06 +00:00
55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# To run the tests, use: python3 -m pytest --capture=sys
|
|
|
|
from database import Database
|
|
|
|
tuple_store = False
|
|
try:
|
|
Database.MAX_ID # Old Database makes db.__MAX_ID private
|
|
tuple_store = True
|
|
except AttributeError:
|
|
pass
|
|
|
|
print('From https://en.wikipedia.org/wiki/Pok%C3%A9mon#Generation_1 ...')
|
|
counts = {'kanto': 151, 'johto': 100, 'hoenn': 135, 'sinnoh': 107, 'all': 493}
|
|
|
|
|
|
def test_kanto_length(): # passes
|
|
db = Database()
|
|
if tuple_store:
|
|
assert len(db.get_region('kanto')) == 151 # Passes
|
|
else:
|
|
assert len(db.get_kanto()) == 151 # Passes
|
|
|
|
|
|
def test_johto_length(): # fails:
|
|
db = Database()
|
|
if tuple_store:
|
|
assert len(db.get_region('johto')) == 100 # Passes
|
|
else:
|
|
assert len(db.get_johto()) == 100 # Fails: assert 101 == 100
|
|
|
|
|
|
def test_hoenn_length(): # fails:
|
|
db = Database()
|
|
if tuple_store:
|
|
assert len(db.get_region('hoenn')) == 135 # Passes
|
|
else:
|
|
assert len(db.get_hoenn()) == 135 # Fails: assert 144 == 135
|
|
|
|
|
|
def test_sinnoh_length(): # fails:
|
|
db = Database()
|
|
if tuple_store:
|
|
assert len(db.get_region('sinnoh')) == 107 # Passes
|
|
else:
|
|
assert len(db.get_sinnoh()) == 107 # Fails: assert 121 == 107
|
|
|
|
|
|
def test_extra_length(): # fails: 0 record
|
|
db = Database()
|
|
if tuple_store:
|
|
assert db.get_region('extra') # Passes
|
|
else:
|
|
assert db.get_extra() # Fails: returns zero pokemon!'
|