fergalmoran/picard
1
0
Fork 0
mirror of https://github.com/fergalmoran/picard.git synced 2026-01-04 15:43:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add function to remove duplicate entries from multi-value variable

Browse Source
This commit is contained in:
Bob Swift
2021-03-27 11:35:07 -06:00
parent 663459a84b
commit 6604d38c60
2 changed files with 50 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
picard/script/functions.py
View File

@@ -1296,3 +1296,29 @@ Result: C; B; A
def func_reversemulti(parser, multi, separator=MULTI_VALUED_JOINER):
multi_value = MultiValue(parser, multi, separator)
return multi_value.separator.join(reversed(multi_value))
@script_function(eval_args=False, documentation=N_(
"""`$uniquemulti(name,case_sensitive="",separator="; ")`
Returns a copy of the multi-value tag `name` with no duplicate elements.
By default, a case-insensitive comparison of the elements is performed.
Example 1:
$uniquemulti(a; A; B; b; cd; Cd; cD; CD; a; A; b)
Result: A; b; CD
Example 2:
$uniquemulti(a; A; B; b; a; b; A; B, cd,True)
Result: a; A; B; b; cd
"""
))
def func_uniquemulti(parser, multi, case_sensitive="", separator=MULTI_VALUED_JOINER):
multi_value = MultiValue(parser, multi, separator)
if not case_sensitive:
multi_value._multi = list({v.lower(): v for v in multi_value}.values())
return multi_value.separator.join(sorted(list(set(multi_value))))

24
test/test_script.py
View File

@@ -1494,3 +1494,27 @@ class ScriptParserTest(PicardTestCase):
self.parser.eval("$reversemulti()")
with self.assertRaisesRegex(ScriptError, areg):
self.parser.eval("$reversemulti(B:AB; D:C; E:D; A:A; C:X,:,extra)")
def test_cmd_uniquemulti(self):
context = Metadata()
context["foo"] = ['a', 'A', 'B', 'b', 'cd', 'Cd', 'cD', 'CD', 'a', 'A', 'b']
context["bar"] = "a; A; B; b; cd; Cd; cD; CD; a; A; b"
# Tests with context
self.assertScriptResultEquals("$uniquemulti(%foo%)", "A; CD; b", context)
self.assertScriptResultEquals("$uniquemulti(%bar%)", "a; A; B; b; cd; Cd; cD; CD; a; A; b", context)
# Tests with static inputs
self.assertScriptResultEquals("$uniquemulti(a; A; B; b; cd; Cd; cD; CD; a; A; b)", "A; CD; b", context)
# Tests with separator override
self.assertScriptResultEquals("$uniquemulti(a: A: B: b: cd: Cd: cD: CD: a: A: b,,: )", "A: CD: b", context)
# Tests with case-sensitive comparison
self.assertScriptResultEquals("$uniquemulti(%foo%,1)", "A; B; CD; Cd; a; b; cD; cd", context)
# Tests with missing inputs
self.assertScriptResultEquals("$uniquemulti(,)", "", context)
self.assertScriptResultEquals("$uniquemulti(,,)", "", context)
self.assertScriptResultEquals("$uniquemulti(,:)", "", context)
# Tests with invalid number of arguments
areg = r"^\d+:\d+:\$uniquemulti: Wrong number of arguments for \$uniquemulti: Expected between 1 and 3, "
with self.assertRaisesRegex(ScriptError, areg):
self.parser.eval("$uniquemulti()")
with self.assertRaisesRegex(ScriptError, areg):
self.parser.eval("$uniquemulti(B:AB; D:C; E:D; A:A; C:X,1,:,extra)")