mirror of
https://github.com/LazoCoder/Pokemon-Terminal
synced 2024-11-23 12:23:04 +00:00
from test_utils import expected_len
This commit is contained in:
parent
6058e1c433
commit
09dc66f488
1 changed files with 12 additions and 13 deletions
|
@ -3,30 +3,29 @@
|
|||
# To run the tests, use: python3 -m pytest --capture=sys
|
||||
|
||||
from database import Database
|
||||
from test_database import make_extra_counts
|
||||
|
||||
print('From https://en.wikipedia.org/wiki/Pok%C3%A9mon#Generation_1 ...')
|
||||
counts = {'kanto': 151, 'johto': 100, 'hoenn': 135, 'sinnoh': 107, 'all': 493}
|
||||
for region_name, extra_count in make_extra_counts().items():
|
||||
counts[region_name] += extra_count # add the extras to the wikipedia counts
|
||||
from test_utils import expected_len
|
||||
|
||||
|
||||
def test_extra_length(): # fails: 0 record
|
||||
assert Database().get_extra() # Fails: returns zero pokemon!'
|
||||
def test_extra_length(region_name='extra'):
|
||||
assert len(Database().get_extra()) == expected_len(region_name)
|
||||
|
||||
|
||||
def test_kanto_length(region_name='kanto'):
|
||||
region_name = 'kanto'
|
||||
assert len(Database().get_kanto()) == counts[region_name]
|
||||
assert len(Database().get_kanto()) == expected_len(region_name)
|
||||
|
||||
|
||||
def test_johto_length(region_name='johto'):
|
||||
assert len(Database().get_johto()) == counts[region_name]
|
||||
assert len(Database().get_johto()) == expected_len(region_name)
|
||||
|
||||
|
||||
def test_hoenn_length(region_name='hoenn'):
|
||||
assert len(Database().get_hoenn()) == counts[region_name]
|
||||
assert len(Database().get_hoenn()) == expected_len(region_name)
|
||||
|
||||
|
||||
def test_sinnoh_length(region_name='sinnoh'):
|
||||
assert len(Database().get_sinnoh()) == counts[region_name]
|
||||
assert len(Database().get_sinnoh()) == expected_len(region_name)
|
||||
|
||||
|
||||
def test_all_length(region_name='all'):
|
||||
expected = expected_len(region_name) + expected_len('extra')
|
||||
assert len(Database().get_all()) == expected
|
||||
|
|
Loading…
Reference in a new issue