Alek Ratzloff
b1461e24db
* Markov.Chain.add_weight/4 would return a fresh new markov chain, instead of using all of the previous values of the chain that was being modified. This would result in resetting the reply_chance value to the default of 0.01. * Add tests to make sure this doesn't happen to add_weights in the future. Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
103 lines
2.8 KiB
Elixir
103 lines
2.8 KiB
Elixir
defmodule MarkovChainTest do
|
|
use ExUnit.Case
|
|
alias Omnibot.Contrib.Markov.Chain
|
|
|
|
test "chain train works correctly" do
|
|
chain = %Chain {order: 2}
|
|
|> Chain.train(~w(foo bar baz))
|
|
assert chain.chain == %{
|
|
["bar", "baz"] => %{nil => 1},
|
|
["foo", "bar"] => %{"baz" => 1},
|
|
}
|
|
|
|
chain = chain |> Chain.train(~w(foo bar baz))
|
|
|
|
assert chain.chain == %{
|
|
["bar", "baz"] => %{nil => 2},
|
|
["foo", "bar"] => %{"baz" => 2},
|
|
}
|
|
|
|
chain = chain |> Chain.train(~w(baz bar foo))
|
|
|
|
assert chain.chain == %{
|
|
["bar", "foo"] => %{nil => 1},
|
|
["baz", "bar"] => %{"foo" => 1},
|
|
["bar", "baz"] => %{nil => 2},
|
|
["foo", "bar"] => %{"baz" => 2},
|
|
}
|
|
|
|
chain = chain |> Chain.train(~w(a b c))
|
|
assert chain.chain == %{
|
|
["b", "c"] => %{nil => 1},
|
|
["a", "b"] => %{"c" => 1},
|
|
["bar", "foo"] => %{nil => 1},
|
|
["baz", "bar"] => %{"foo" => 1},
|
|
["bar", "baz"] => %{nil => 2},
|
|
["foo", "bar"] => %{"baz" => 2},
|
|
}
|
|
end
|
|
|
|
test "chain add_weight works correctly" do
|
|
chain = %Chain {order: 2}
|
|
|> Chain.add_weight(["foo", "bar"], "baz")
|
|
assert chain.chain == %{
|
|
["foo", "bar"] => %{"baz" => 1},
|
|
}
|
|
|
|
chain = chain |> Chain.add_weight(["foo", "bar"], "baz", 2)
|
|
|
|
assert chain.chain == %{
|
|
["foo", "bar"] => %{"baz" => 3},
|
|
}
|
|
|
|
chain = chain |> Chain.add_weight(["foo", "bar"], "qux")
|
|
assert chain.chain == %{
|
|
["foo", "bar"] => %{"baz" => 3, "qux" => 1},
|
|
}
|
|
|
|
chain = chain |> Chain.add_weight(["bar", "baz"], "qux")
|
|
assert chain.chain == %{
|
|
["bar", "baz"] => %{"qux" => 1},
|
|
["foo", "bar"] => %{"baz" => 3, "qux" => 1},
|
|
}
|
|
|
|
chain = chain |> Chain.add_weight(["bar", "baz"], nil)
|
|
assert chain.chain == %{
|
|
["bar", "baz"] => %{"qux" => 1, nil => 1},
|
|
["foo", "bar"] => %{"baz" => 3, "qux" => 1},
|
|
}
|
|
end
|
|
|
|
test "chain add_weight does not reset reply_chance" do
|
|
chain = %Chain {order: 2, reply_chance: 0.0}
|
|
|> Chain.add_weight(["foo", "bar"], "baz")
|
|
|
|
chain = chain |> Chain.add_weight(["foo", "bar"], "baz", 2)
|
|
assert chain.reply_chance == 0.0
|
|
|
|
chain = chain |> Chain.add_weight(["foo", "bar"], "qux")
|
|
assert chain.reply_chance == 0.0
|
|
|
|
chain = chain |> Chain.add_weight(["bar", "baz"], "qux")
|
|
assert chain.reply_chance == 0.0
|
|
|
|
chain = chain |> Chain.add_weight(["bar", "baz"], nil)
|
|
assert chain.reply_chance == 0.0
|
|
end
|
|
|
|
test "chain merge works correctly" do
|
|
chain1 = %Chain {order: 2}
|
|
|> Chain.add_weight(["foo", "bar"], "baz")
|
|
|
|
chain2 = %Chain {order: 2}
|
|
|> Chain.add_weight(["foo", "bar"], "baz")
|
|
|> Chain.add_weight(["bar", "baz"], "qux")
|
|
|
|
merged = Chain.merge(chain1, chain2)
|
|
assert merged.chain == %{
|
|
["foo", "bar"] => %{"baz" => 2},
|
|
["bar", "baz"] => %{"qux" => 1},
|
|
}
|
|
end
|
|
end
|