class PokemonTests(unittest.TestCase): def setUp(self): self.pokemon = Pokemon('pokedex.csv') def test_get_boss_url_existing(self): self.assertGreater(len(self.pokemon.get_boss_url('Heatran')), 0) self.assertGreater(len(self.pokemon.get_boss_url('golem')), 0) self.assertGreater(len(self.pokemon.get_boss_url('alolan raichu')), 0) def test_get_boss_url_nonexisting(self): self.assertEqual(len(self.pokemon.get_boss_url('afsdf')), 0) def test_get_boss_url_case_insensitive(self): self.assertGreater(len(self.pokemon.get_boss_url('pikachu')), 0) self.assertEqual(self.pokemon.get_boss_url('PikAchu'), self.pokemon.get_boss_url('pikachu')) def test_find_found(self): self.assertEqual(self.pokemon.find('venusaur'), 'venusaur') self.assertEqual(self.pokemon.find('Venusaur'), 'venusaur') self.assertEqual(self.pokemon.find('venasaur'), 'venusaur') self.assertEqual(self.pokemon.find('venusar'), 'venusaur') self.assertEqual(self.pokemon.find('veenusaur'), 'venusaur') self.assertEqual(self.pokemon.find('heatrn'), 'heatran') self.assertEqual(self.pokemon.find('mega charizard x'), 'mega charizard x') self.assertEqual(self.pokemon.find('maga charizard x'), 'mega charizard x') def test_find_not_found(self): self.assertEqual(self.pokemon.find('mega charizard'), '')