def test_sub_fields(self):
ca = Argument("base", dict, [
Argument("sub1", int),
Argument("sub2", [str, dict], [
Argument("subsub1", int),
Argument("subsub2", dict, [Argument("subsubsub1", int)])
])
])
# short one
test_dict1 = {"base": {"sub1": 1, "sub2": "hello"}}
self.assertSetEqual(set(ca.flatten_sub(test_dict1["base"]).keys()),
set(test_dict1["base"].keys()))
ca.check(test_dict1)
ca.check_value(test_dict1["base"])
# long one
test_dict2 = {
"base": {
"sub1": 1,
"sub2": {
"subsub1": 11,
"subsub2": {
"subsubsub1": 111
}
}
}
}
ca.check(test_dict2)
ca.check_value(test_dict2["base"])
# expect error
err_dict1 = {
"base": {
"sub1": 1,
"sub2": {
"subsub1": 11,
"subsub2": {
"subsubsub2": 111
}
}
}
} # different name here
with self.assertRaises(ArgumentKeyError):
ca.check(err_dict1)
err_dict1["base"]["sub2"]["subsub2"]["subsubsub1"] = 111
ca.check(err_dict1) # now should pass
with self.assertRaises(ArgumentKeyError):
ca.check(err_dict1, strict=True) # but should fail when strict
err_dict1["base"]["sub2"] = None
with self.assertRaises(ArgumentTypeError):
ca.check(err_dict1)
# make sure no dup keys is allowed
with self.assertRaises(ValueError):
Argument("base", dict,
[Argument("sub1", int),
Argument("sub1", int)])
def test_sub_variants(self):
ca = Argument(
"base",
dict,
[Argument("sub1", int),
Argument("sub2", str)],
[
Variant(
"vnt_flag",
[
Argument("type1", dict, [
Argument("shared", int),
Argument("vnt1_1", int),
Argument("vnt1_2", dict,
[Argument("vnt1_1_1", int)])
]),
Argument("type2",
dict, [
Argument("shared", int),
Argument("vnt2_1", int),
],
alias=['type2a']),
Argument(
"type3",
dict,
[Argument("vnt3_1", int)],
[ # testing cascade variants here
Variant("vnt3_flag1", [
Argument("v3f1t1", dict, [
Argument('v3f1t1_1', int),
Argument('v3f1t1_2', int)
]),
Argument("v3f1t2", dict,
[Argument('v3f1t2_1', int)])
]),
Variant("vnt3_flag2", [
Argument("v3f2t1", dict, [
Argument('v3f2t1_1', int),
Argument('v3f2t1_2', int)
]),
Argument("v3f2t2", dict,
[Argument('v3f2t2_1', int)])
])
])
])
])
test_dict1 = {
"base": {
"sub1": 1,
"sub2": "a",
"vnt_flag": "type1",
"shared": 10,
"vnt1_1": 11,
"vnt1_2": {
"vnt1_1_1": 111
}
}
}
self.assertSetEqual(set(ca.flatten_sub(test_dict1["base"]).keys()),
set(test_dict1["base"].keys()))
ca.check(test_dict1)
test_dict2 = {
"base": {
"sub1": 1,
"sub2": "a",
"vnt_flag": "type2",
"shared": 20,
"vnt2_1": 21
}
}
self.assertSetEqual(set(ca.flatten_sub(test_dict2["base"]).keys()),
set(test_dict2["base"].keys()))
ca.check(test_dict2, strict=True)
test_dict2["base"]["vnt_flag"] = "type2a"
ca.check(test_dict2, strict=True)
err_dict1 = {
"base": {
"sub1": 1,
"sub2": "a",
"vnt_flag": "type2", # here is wrong
"shared": 10,
"vnt1_1": 11,
"vnt1_2": {
"vnt1_1_1": 111
}
}
}
with self.assertRaises(ArgumentKeyError):
ca.check(err_dict1)
err_dict1["base"]["vnt_flag"] = "type1"
ca.check(err_dict1, strict=True) # no additional should pass
err_dict1["base"]["additional"] = "hahaha"
ca.check(err_dict1) # without strict should pass
with self.assertRaises(ArgumentKeyError):
ca.check(err_dict1, strict=True) # but should fail when strict
err_dict2 = {
"base": {
"sub1": 1,
"sub2": "a",
"vnt_flag": "badtype", # here is wrong
"shared": 20,
"vnt2_1": 21
}
}
with self.assertRaises(ArgumentValueError):
ca.check(err_dict2)
# test optional choice
test_dict1["base"].pop("vnt_flag")
with self.assertRaises(ArgumentKeyError):
ca.check(test_dict1)
ca.sub_variants["vnt_flag"].optional = True
ca.sub_variants["vnt_flag"].default_tag = "type1"
ca.check(test_dict1)
# test cascade variants
test_dict3 = {
"base": {
"sub1": 1,
"sub2": "a",
"vnt_flag": "type3",
"vnt3_1": 31,
"vnt3_flag1": "v3f1t1",
"vnt3_flag2": "v3f2t2",
"v3f1t1_1": 3111,
"v3f1t1_2": 3112,
"v3f2t2_1": 3221
}
}
self.assertSetEqual(set(ca.flatten_sub(test_dict3["base"]).keys()),
set(test_dict3["base"].keys()))
ca.check(test_dict3, strict=True)
test_dict3["base"].pop("vnt3_flag2")
with self.assertRaises(ArgumentKeyError):
ca.check(test_dict3)
ca.sub_variants['vnt_flag'].choice_dict["type3"].sub_variants[
"vnt3_flag2"].optional = True
ca.sub_variants['vnt_flag'].choice_dict["type3"].sub_variants[
"vnt3_flag2"].default_tag = 'v3f2t2'
ca.check(test_dict3, strict=True)
# make sure duplicate tag is not allowed
with self.assertRaises(ValueError):
Argument("base", dict, [], [
Variant("flag",
[Argument("type1", dict),
Argument("type1", dict)])
])
with self.assertRaises(ValueError):
Argument("base", dict, [], [
Variant("flag", [Argument("type1", dict)]),
Variant("flag", [Argument("type2", dict)])
])
