# Stephanie Taylor
# CS152 Fall 2017
# Project 5
# This module tests the controlPopulation function in elephant.py
# Call it like this:
#    python test_controlPopulation.py
# Because I have seeded the random number generator, the output will be identical every time you run it. You should see:
# Test 1
# Out of 5144 adult female elephants, 2587 are darted, leading to a percentage of 0.503. 
# 
# Test 2
# 100 should have been culled and 100 were
# 
# Test 3
# Out of 5036 adult female elephants, 5036 are darted, leading to a percentage of 1.000. 
# 
# Test 4
# We needed to cull 1000 elephants and 1000 elephants were culled
# Out of 1000 calves before culling, approximately 909 should remain after culling. 913 remain.
# 
# Test 5
# We needed to cull 0 elephants and 0 elephants were culled
# Out of 0 calves before culling, approximately 0 should remain after culling. 0 remain.
# 
# Test 6
# We needed to cull 10000 elephants and 10000 elephants were culled
# Out of 10000 calves before culling, 5015 remain after culling.
# 
# Test 7
# We didn't need to cull any elephants because the carrying capacity is 10000 and there were only 9990 elephants. There are now 9990 elephants
#
# If you would like it to change from run to run, then comment out the call to random.seed.


import elephant
import random

def testDarting( pcnt_darted ):
    calving_interval = 3.1
    max_juvenile_age = 12
    max_adult_age = 60
    p_calf_survival = 0.85
    p_adult_survival = 0.996
    p_senior_survival = 0.2
    carrying_capacity = 10000
    num_over = 100

    params = [calving_interval, pcnt_darted, max_juvenile_age, max_adult_age, p_calf_survival, p_adult_survival, p_senior_survival, carrying_capacity]

    # Create a new, undarted population, and make it too big. It should remain too big if
    # the pcnt_darted > 0
    pop = []
    for i in range( carrying_capacity+num_over ):
        # Create a new adult elephant. 
        pop.append( [random.choice(['m','f']), random.randint(max_juvenile_age+1, max_adult_age ), 0, 0] )
    (pop, numCulled) = elephant.controlPopulation( params, pop )

    if pcnt_darted > 0.0:
        numF = 0
        numDarted = 0
        for e in pop:
            if e[0] == 'f':
                numF += 1
                if e[3] == 22:
                    numDarted += 1
        print "Out of %d adult female elephants, %d are darted, leading to a percentage of %0.3f." % (numF, numDarted, float(numDarted)/numF)
    else:
        print "%d should have been culled and %d were" % (num_over, numCulled)

def testCulling( population_overrun ):
    calving_interval = 3.1
    max_juvenile_age = 12
    max_adult_age = 60
    p_calf_survival = 0.85
    p_adult_survival = 0.996
    p_senior_survival = 0.2
    carrying_capacity = 10000
    pcnt_darted = 0.0

    params = [calving_interval, pcnt_darted, max_juvenile_age, max_adult_age, p_calf_survival, p_adult_survival, p_senior_survival, carrying_capacity]

    # Create a new, undarted population
    pop = []
    for i in range( carrying_capacity ):
        # Create a new adult elephant. 
        pop.append( [random.choice(['m','f']), random.randint(max_juvenile_age+1, max_adult_age ), 0, 0] )
    if population_overrun >= 0:
        for i in range( population_overrun ):
            # Create a new calf.
            pop.append( [random.choice(['m','f']), 1, 0, 0] )
        (pop,numCulled) = elephant.controlPopulation( params, pop )
        print "We needed to cull %d elephants and %d elephants were culled" % (population_overrun, numCulled)
    
        # If the population has been properly culled, there will still be calves.
        numCalves = 0
        for e in pop:
            if e[1] == 1:
                numCalves += 1
        print "Out of %d calves before culling, %d remain after culling." % (population_overrun, numCalves)
    elif population_overrun < 0:
        for i in range( -population_overrun ):
            pop.pop()
        (newpop,numCulled) = elephant.controlPopulation( params, pop )
        print "We didn't need to cull any elephants because the carrying capacity is %d and there were only %d elephants. There are now %d elephants" % (carrying_capacity, len(pop), len(newpop))

def main():
    random.seed( 0 )

    # Test the function with a darting percentage of 50%
    print "Test 1"
    testDarting( 0.5 )

    # Test the function with a darting percentage of 0%
    print "\nTest 2"
    testDarting( 0.0 )

    # Test the function with a darting percentage of 100%
    print "\nTest 3"
    testDarting( 1.0 )

    # Test the function with a population overrun of 1000
    print "\nTest 4"
    testCulling( 1000 )

    # Test the function with a population overrun of 0
    print "\nTest 5"
    testCulling( 0 )

    # Test the function with a population overrun of 10000
    print "\nTest 6"
    testCulling( 10000 )
    
    # Test the function with a population overrun of 10000
    print "\nTest 7"
    testCulling( -10 )

if __name__ == '__main__':
    main()