def test_not_ri(self):
'''Test trade code not Ri'''
uwp = Uwp('C700055-5')
for atm in '01234579ABC':
for pop in '0123459ABCDEF':
for gov in '012ABCDEF':
uwp.atmosphere = atm
uwp.population = pop
uwp.government = gov
tcs = TradeCodes(uwp)
self.assertFalse('Ri' in tcs.trade_codes)
def test_ri(self):
'''Test trade code Ri'''
uwp = Uwp('C700055-5')
for atm in '68':
for pop in '678':
for gov in '456789':
uwp.atmosphere = atm
uwp.population = pop
uwp.government = gov
tcs = TradeCodes(uwp)
self.assertTrue('Ri' in tcs.trade_codes)
def test_not_di(self):
'''Test trade code not Di'''
uwp = Uwp('X766000-0')
for pop in '123456789ABCDEF':
for gov in '123456789ABCDEF':
for law in '123456789ABCDEFGHJ':
uwp.population = pop
uwp.government = gov
uwp.law_level = law
tcs = TradeCodes(uwp)
self.assertFalse('Di' in tcs.trade_codes)
def test_not_ba(self):
'''Test trade code not Ba'''
uwp = Uwp('X755000-0')
for pop in '123456789A':
for gov in '123456789AB':
for law in '123456789A': # Assume the rest are OK
for tech in '123456': # Assume the rest are OK
for starport in 'ABCD':
uwp.starport = starport
uwp.population = pop
uwp.government = gov
uwp.law_level = law
uwp.tech_level = tech
tcs = TradeCodes(uwp)
self.assertFalse('Ba' in tcs.trade_codes)
