fb9ef19c15
* proof of concept * add dict support, block top/game level merge * prevent key error when option being merged is new * update triggers guide * Add documentation about add/remove/replace * move to trailing name instead of proper tag * update docs * confirm types * Update Utils.py * Update Generate.py * pep8 * move to + syntax * forgot to support sets * specify received type of type error * Update Generate.py Co-authored-by: Fabian Dill <Berserker66@users.noreply.github.com> * Apply suggestion from review * add test for update weights * move test to new test case * Apply suggestions from code review Co-authored-by: black-sliver <59490463+black-sliver@users.noreply.github.com> --------- Co-authored-by: Fabian Dill <Berserker66@users.noreply.github.com> Co-authored-by: black-sliver <59490463+black-sliver@users.noreply.github.com>
40 lines
1.8 KiB
Python
40 lines
1.8 KiB
Python
import unittest
|
|
import Generate
|
|
|
|
|
|
class TestPlayerOptions(unittest.TestCase):
|
|
|
|
def test_update_weights(self):
|
|
original_weights = {
|
|
"scalar_1": 50,
|
|
"scalar_2": 25,
|
|
"list_1": ["string"],
|
|
"dict_1": {"option_a": 50, "option_b": 50},
|
|
"dict_2": {"option_f": 50},
|
|
"set_1": {"option_c"}
|
|
}
|
|
|
|
# test that we don't allow +merge syntax on scalar variables
|
|
with self.assertRaises(BaseException):
|
|
Generate.update_weights(original_weights, {"+scalar_1": 0}, "Tested", "")
|
|
|
|
new_weights = Generate.update_weights(original_weights, {"scalar_2": 0,
|
|
"+list_1": ["string_2"],
|
|
"+dict_1": {"option_b": 0, "option_c": 50},
|
|
"+set_1": {"option_c", "option_d"},
|
|
"dict_2": {"option_g": 50},
|
|
"+list_2": ["string_3"]},
|
|
"Tested", "")
|
|
|
|
self.assertEqual(new_weights["scalar_1"], 50)
|
|
self.assertEqual(new_weights["scalar_2"], 0)
|
|
self.assertEqual(new_weights["list_2"], ["string_3"])
|
|
self.assertEqual(new_weights["list_1"], ["string", "string_2"])
|
|
self.assertEqual(new_weights["dict_1"]["option_a"], 50)
|
|
self.assertEqual(new_weights["dict_1"]["option_b"], 0)
|
|
self.assertEqual(new_weights["dict_1"]["option_c"], 50)
|
|
self.assertNotIn("option_f", new_weights["dict_2"])
|
|
self.assertEqual(new_weights["dict_2"]["option_g"], 50)
|
|
self.assertEqual(len(new_weights["set_1"]), 2)
|
|
self.assertIn("option_d", new_weights["set_1"])
|