fzfpytest

fzfpytest.sh

#!/usr/bin/env sh

parse() {
    code='\
import re
import sys

files = sys.argv[1:]

tests = []
for file in files:
    with open(file, "r") as f:
        lines = f.readlines()

    lines = [
        re.sub("(class |def |:|\\(.*)", "", line).rstrip()
        for line in lines if "class" in line or "def" in line
    ]

    module_ = file
    tests.append(module_)
    for line in lines:
        # class
        if line.startswith("T"):
            class_ = line
            tests.append(f"{module_}::{class_}")
        # class function
        elif line.startswith("    "):
            function_ = line.strip()
            tests.append(f"{module_}::{class_}::{function_}")
        # module function
        elif line.startswith("t"):
            function_ = line
            tests.append(f"{module_}::{function_}")

print(" ".join(tests))
'
    python3 -c "$code" $@
}

gather() {
    res=$(ls tests/ | egrep test_.*.py)
    modules=""
    for module in $res; do
        modules+="tests/$module "
    done

    echo "$modules"
}

main() {
    res="$(gather)"
    res=$(parse $res)

    res=$(echo "$res" | sed -e 's/ /\n/g' | fzf)
    if [ -z "$res" ]; then
        return
    fi

    xdotool type "pt -s $res"
}
main $@