Exemplo n.º 1
0
class Zoo():
    """docstring for Zoo"""
    def __init__(self, name, capacity, budget):
        self.__animals = []
        self.__name = name
        self.__capacity = capacity
        self.__budget = budget
        self.__database = Database(name + ".db")

    def get_name(self):
        return self.__name

    def get_capacity(self):
        return self.__capacity

    def get_animals(self):
        return self.__animals

    def get_database(self):
        return self.__database

    def accommodate_animal(self, species, age, name, gender, weight):
        if self.__capacity <= len(self.__animals):
            return False
        new_animal = Animal(species, age, name, gender, weight)
        self.__animals.append(new_animal)
        self.__database.insert_animal(new_animal)
        return True

    # <name> : <species>, <age>, <weight>
    def see_animals(self):
        formated_list = []
        # animal_objects look like this: (id,species,age,name,gender,weight)
        for animal in self.__animals:
            species = animal.get_species()
            age = animal.get_age()
            name = animal.get_name()
            weight = animal.get_weight()

            animal_string = name + "\t   -   " + species + ", age: " + str(age)
            animal_string += " years, weight: " + str(weight) + " kg"
            formated_list.append(animal_string)
        output_string = '\n'.join(formated_list)
        return output_string

    def remove_animal(self, species, name):
        for animal in self.__animals:
            if animal.get_species() == species and animal.get_name() == name:
                self.__animals.remove(animal)
                break

    def move_animal(self, species, name):
        self.remove_animal(species, name)

    def daily_incomes(self):
        return 60 * len(self.__animals)

    def daily_expenses(self):
        money = 0
        for animal in self.__animals:
            animal_type = self.__database.get_food_type(animal.get_species())
            food_weight_ratio = self.__database.get_food_weight_ratio(animal.get_species())
            if animal_type == "carnivore":
                money += 4 * animal.feed(food_weight_ratio)
            else:
                money += 2 * animal.feed(food_weight_ratio)
        return money

    def flip_a_coin(self):
        gender = randint(0, 1)
        if gender == 0:
            gender = 'male'
        else:
            gender = 'female'
        return gender

    def generate_name(self, species, name, gender):
        name = self.__database.get_male_name(species)
        if gender == 'female':
            return name + 'ka'
        else:
            return name + str(randint(1, 100))

    def born_animal(self, species, name):
        self.__database.set_last_breed(species, name, 0)
        weight = self.__database.get_newborn_weight(species)
        gender = self.flip_a_coin()
        name = self.generate_name(species, name, gender)
        new_animal = Animal(species, 1, name, gender, weight)
        self.__animals.append(new_animal)
        self.__database.insert_animal(new_animal)

        if len(self.__animals) > self.__capacity:
            return False
        return True

    def ready_to_give_birth(self, species, name):
        breeding_period =\
            6 + self.database.get_gestation(species)
        if self.database.has_male(species):
            if breeding_period <= self.database.get_last_breed(species, name):
                return True
        return False

    def update_animals_from_database(self):
        self.__animals = []
        db = self.get_database()
        db_name = db.get_name()
        conn = sqlite3.connect(db_name)
        cursor = conn.cursor()

        animals_from_db = cursor.execute('''SELECT * FROM zoo''').fetchall()
        for animal in animals_from_db:
            # (1, 'lion', 10.0, 'Svetla', 'female', 160.0)
            species = animal[1]
            age = animal[2]
            name = animal[3]
            gender = animal[4]
            weight = animal[5]
            this_animal = Animal(species, age, name, gender, weight)
            self.__animals.append(this_animal)
        return self.__animals
Exemplo n.º 2
0
class DatabaseTest(unittest.TestCase):
    def setUp(self):
        self.db = Database("test_zoo.db")
        self.a = Animal("lion", 24, "niya", "female", 150)
        self.db.insert_animal(self.a)
        self.c = self.db.zoo_conn.cursor()

    def test_database_init(self):
        self.assertEqual("test_zoo.db", self.db.name)

    def test_insert_animal(self):
        animal_from_db = self.c.execute('''select * from zoo''').fetchall()[0]
        self.assertEqual((1, "lion", 24, "niya", "female", 150),
                    animal_from_db)
        last_breed_from_db = self.c.execute('''select id, last_breed
                        from breeding''').fetchall()[0]
        self.assertEqual((1, 0), last_breed_from_db)

    def test_remove_animal(self):
        self.db.remove_animal("lion", "niya")

        animal_from_db = self.c.execute("select * from zoo").fetchall()
        self.assertEqual(0, len(animal_from_db))

        breed_from_db = self.c.execute("select * from breeding").fetchall()
        self.assertEqual(0, len(breed_from_db))

    def test_get_males_with_no_males(self):
        self.assertEqual(0, len(self.db.get_males("lion")))

    def test_get_males_with_one_male(self):
        a2 = Animal("lion", 24, "sharik", "male", 150)
        self.db.insert_animal(a2)
        self.assertEqual([("sharik", )], self.db.get_males("lion"))

    def test_has_male(self):
        self.assertFalse(self.db.has_male("lion"))

    def test_get_females(self):
        self.assertEqual([("lion", "niya")], self.db.get_females())

    def test_get_life_expectancy(self):
        self.assertEqual(15, self.db.get_life_expectancy("lion"))

    def test_get_food_type(self):
        self.assertEqual("carnivore", self.db.get_food_type("lion"))

    def test_get_gestation(self):
        self.assertEqual(3, self.db.get_gestation("lion"))

    def test_get_newborn_weight(self):
        self.assertEqual(2.0, self.db.get_newborn_weight("lion"))

    def test_get_average_weight(self):
        self.assertEqual(200, self.db.get_average_weight("lion"))

    def test_get_weight_age_ratio(self):
        self.assertEqual(7.5, self.db.get_weight_age_ratio("lion"))

    def test_get_food_weight_ration(self):
        self.assertEqual(0.035, self.db.get_food_weight_ratio("lion"))

    def test_get_last_breed(self):
        self.assertEqual(0, self.db.get_last_breed("lion", "niya"))

    def test_set_last_breed(self):
        self.db.set_last_breed("lion", "niya", 3)
        result = self.db.get_last_breed("lion", "niya")
        self.assertEqual(result, 3)

    def tearDown(self):
        call("rm {}".format(self.db.name), shell=True)
Exemplo n.º 3
0
class DatabaseTest(unittest.TestCase):
    def setUp(self):
        self.db = Database("test_zoo.db")
        self.a = Animal("lion", 24, "Svetla", "female", 150)
        self.db.insert_animal(self.a)
        self.c = self.db.zoo_conn.cursor()

    def test_get_name(self):
        self.assertEqual("test_zoo.db", self.db.get_name())

    def test_insert_animal(self):
        animal_from_db = self.c.execute('''SELECT * FROM zoo''').fetchall()[0]
        self.assertEqual((1, "lion", 24, "Svetla", "female", 150),
            animal_from_db)
        last_breed_from_db = self.c.execute('''SELECT id, last_breed
                        FROM breeding''').fetchall()[0]
        self.assertEqual((1, 0), last_breed_from_db)

    def test_insert_second_animal(self):
        self.a2 = Animal("tiger", 24, "Tsveta", "female", 120)
        self.db.insert_animal(self.a2)
        animal_from_db = self.c.execute('''SELECT * FROM zoo''').fetchall()[1]
        self.assertEqual((2, "tiger", 24, "Tsveta", "female", 120),
            animal_from_db)
        last_breed_from_db = self.c.execute('''SELECT id, last_breed
                        FROM breeding''').fetchall()[1]
        self.assertEqual((2, 0), last_breed_from_db)

    def test_remove_animal(self):
        breed_from_db = self.c.execute("SELECT * from breeding").fetchall()
        self.assertEqual(1, len(breed_from_db))

        self.db.remove_animal("lion", "Svetla")

        animal_from_db = self.c.execute("SELECT * FROM zoo").fetchall()
        self.assertEqual(0, len(animal_from_db))

        breed_from_db = self.c.execute("SELECT * from breeding").fetchall()
        self.assertEqual(0, len(breed_from_db))

    def test_initial_fill_with_animals(self):
        self.db.remove_animal("lion", "Svetla")
        self.db.initial_fill_with_animals()

        animals_from_db = self.c.execute("SELECT * FROM zoo").fetchall()
        self.assertEqual(10, len(animals_from_db))

        breed_from_db = self.c.execute("SELECT * from breeding").fetchall()
        self.assertEqual(5, len(breed_from_db))

    def test_get_males_with_no_males(self):
        self.assertEqual(0, len(self.db.get_males("lion")))

    def test_get_males_with_one_male(self):
        a2 = Animal("lion", 24, "Pencho", "male", 150)
        self.db.insert_animal(a2)
        self.assertEqual([("Pencho", )], self.db.get_males("lion"))

    def test_has_a_male_species(self):
        self.assertFalse(self.db.has_a_male_species("lion"))

    def test_get_females(self):
        self.assertEqual([("lion", "Svetla")], self.db.get_females())

    def test_get_life_expectancy(self):
        self.assertEqual(15, self.db.get_life_expectancy("lion"))

    def test_get_food_type(self):
        self.assertEqual("carnivore", self.db.get_food_type("lion"))

    def test_get_gestation(self):
        self.assertEqual(3, self.db.get_gestation("lion"))

    def test_get_newborn_weight(self):
        self.assertEqual(2.0, self.db.get_newborn_weight("lion"))

    def test_get_average_weight(self):
        self.assertEqual(200, self.db.get_average_weight("lion"))

    def test_get_weight_age_ratio(self):
        self.assertEqual(7.5, self.db.get_weight_age_ratio("lion"))

    def test_get_food_weight_ration(self):
        self.assertEqual(0.035, self.db.get_food_weight_ratio("lion"))

    def test_get_last_breed(self):
        self.assertEqual(0, self.db.get_last_breed("lion", "Svetla"))

    def test_set_last_breed(self):
        self.db.set_last_breed("lion", "Svetla", 3)
        result = self.db.get_last_breed("lion", "Svetla")
        self.assertEqual(result, 3)

    def test_set_age(self):
        self.db.set_age("lion", "Svetla", 30)
        query = "SELECT age FROM zoo WHERE species = ? AND name = ?"
        result = self.c.execute(query, ("lion", "Svetla")).fetchall()
        self.assertEqual(30, result[0][0])

    def test_set_weight(self):
        self.db.set_weight("lion", "Svetla", 195)
        query = "SELECT weight FROM zoo WHERE species = ? AND name = ?"
        result = self.c.execute(query, ("lion", "Svetla")).fetchall()
        self.assertEqual(195, result[0][0])

    def tearDown(self):
        call("rm {}".format(self.db.name), shell=True)