def test_init_from_args(argfilepath): prune = Prune.init_from_args("A,B,C") assert prune.opts.filename == None assert prune.opts.attribute == None assert prune.opts.values == None assert prune.opts.inverse == False prune = Prune.init_from_args("A,B,C --inverse") assert prune.opts.inverse == True taxa_abc = argfilepath('taxa_abc.txt') prune = Prune.init_from_args("--file {0}".format(taxa_abc)) assert prune.opts.filename == taxa_abc assert prune.opts.attribute == None assert prune.opts.values == None prune = Prune.init_from_args("--attribute foo --values bar") assert prune.opts.filename == None assert prune.opts.attribute == "foo" assert prune.opts.values == ["bar"]
def test_init_from_args(): prune = Prune.init_from_args("A,B,C") assert prune.taxa == set(("A","B","C")) assert prune.filename == None assert prune.attribute == None assert prune.value == None assert prune.inverse == False prune = Prune.init_from_args("A,B,C --inverse") assert prune.inverse == True prune = Prune.init_from_args("--file tests/argfiles/taxa_abc.txt") assert prune.taxa == set(("A","B","C")) assert prune.filename == "tests/argfiles/taxa_abc.txt" assert prune.attribute == None assert prune.value == None prune = Prune.init_from_args("--attribute foo --value bar") assert prune.taxa == [] assert prune.filename == None assert prune.attribute == "foo" assert prune.value == "bar"