class PlayerGenerator: """Generates players""" def __init__(self): """Initialize the generator""" self.name_generator = NameGenerator() self.position_generator = PositionGenerator() def generate(self): name = self.name_generator.generate() position = self.position_generator.generate() scoring_rate = normal(loc=0.085, scale=0.02) if scoring_rate < 0.0: scoring_rate = 0.0 shot_rate = normal(loc=6.0, scale=2.5) if shot_rate < 0.0: shot_rate = 0.0 return Player(name=name, position=position, scoring_rate=scoring_rate, shot_rate=shot_rate)
def test_creation(self): generator = NameGenerator() self.assertIsNotNone(generator) self.assertIsNotNone(generator.generate())