from pypower.api import ppoption, runpf, printpf
import numpy as np


def banshee_f1(s_base, bus, branch, gen, gencost):
    ppc = {"version": '2'}
    ppc["baseMVA"] = s_base
    ppc["bus"] = bus
    ppc["branch"] = branch
    ppc["gen"] = gen
    ppc["gencost"] = gencost

    return ppc


name = 'banshee_f1'
s_base = 3.75   # MVA
bus = np.load(name + '_bus.npy')
branch = np.load(name + '_branch.npy')
gen = np.load(name + '_gen.npy')
gencost = np.load(name + '_gencost.npy')

ppc = banshee_f1(s_base, bus, branch, gen, gencost)
ppopt = ppoption()
r = runpf(ppc, ppopt)
printpf(r)