# Copyright (c) 2007 Tim Freeman
# 
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
# 
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
# 
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#
# (This is the standard MIT License, copied from
# http://www.opensource.org/licenses/mit-license.php on 24 Apr 2007.)
from utility_combiners import Respectful_utility_combiner
from test_util import assert_equals

alice = "alice"
bob = "bob"
grocer = "grocer"
everybody_else = [alice, bob, grocer]

compassion_table = {}
for who in everybody_else:
    compassion_table[who] = 1

respect_table = {}
for who in everybody_else:
    respect_table[who] = 4

respect_combiner = Respectful_utility_combiner(
    compassion_table=compassion_table,
    respect_table=respect_table)

# Bob loses 2.  Alice gains 14.  The grocer loses 2.
# Respect level is 4.
# Gain from respect is bob + grocer = -4 * 4 = -16.
# Gain from compassion is 14.
# So the sum should be -2.
assert_equals(respect_combiner.ai_utility(
    {bob: 8, alice: 24, grocer: 6},
    {bob: 10, alice: 10, grocer: 8}),
              -2)