added tests for fitness function