def test_ba(self):
'''Test Ba'''
pop = 0
gov = 0
law = 0
tl = 0
planet = Planet()
planet.population = uwp.Population(pop)
planet.government = uwp.Government(gov)
planet.law_level = uwp.LawLevel(law)
planet.tech_level = uwp.TechLevel(tl)
self.assertTrue('Ba' in gen_trade_codes(planet))
def test_not_ba(self):
'''Test !Ba'''
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':
planet = Planet()
planet.starport = starport
planet.population = uwp.Population(pop)
planet.government = uwp.Government(gov)
planet.law_level = uwp.LawLevel(law)
planet.tech_level = uwp.TechLevel(tech)
self.assertFalse('Ba' in gen_trade_codes(planet))
def test_no_lt(self):
'''Test !Lt'''
for tech_level in '56789ABCDEF':
planet = Planet()
planet.tech_level = uwp.TechLevel(tech_level)
self.assertFalse('Lt' in gen_trade_codes(planet))
def test_no_ht(self):
'''Test !Ht'''
for tech_level in '0123456778AB':
planet = Planet()
planet.tech_level = uwp.TechLevel(tech_level)
self.assertFalse('Ht' in gen_trade_codes(planet))
def test_lt(self):
'''Test Lt'''
for tech_level in '01234':
planet = Planet()
planet.tech_level = uwp.TechLevel(tech_level)
self.assertTrue('Lt' in gen_trade_codes(planet))
def test_ht(self):
'''Test Ht'''
for tech_level in 'CDEF':
planet = Planet()
planet.tech_level = uwp.TechLevel(tech_level)
self.assertTrue('Ht' in gen_trade_codes(planet))
