def argmax_test_4(self):
"""
Maximum argument with one variable
"""
for i, domain in enumerate(self.X.domain):
fac = Factor(self.X, [1, 2])
fac.values[i] = 10
event = Event([(self.X, domain)])
assert(fac.argmax() == event)
def argmax_test_5(self):
"""
Maximum argument with two variables
"""
for i, domainx in enumerate(self.X.domain):
for j, domainy in enumerate(self.Y.domain):
fac = Factor([self.X, self.Y], [1, 2, 3, 4])
fac.values[i + j*2] = 10
event = Event([(self.X, domainx), (self.Y, domainy)])
assert(fac.argmax() == event)
