def test_plot():
ac = ArrowCalc(**dataset1['config'])
data = ac.calc([[6, 9, 0], [9, 6, 9]]) * 20
elements = [dict(polygon=pts) for pts in arrows.format_points(data)]
# print(data, file=sys.stderr)
# print(elements, file=sys.stderr)
kwargs = arrows.get_defaults()
kwargs['elements'] = elements
res = arrows.render_svg(**kwargs)
print(res)
def test_arrow_calc():
for ds in [dataset1, dataset2, dataset3]:
ac = ArrowCalc(**ds['config'])
# result = calc.prototype(400).astype(np.int)
results = ac.calc(ds['lengths'])
assert arr_eq(results, ds['results'], .001)
resultp = ac.prototype(ds['lengths'][0])
assert arr_eq(resultp, ds['results'][0], .001)
# assert 1 == 2
for l in np.arange(10, step=.01):
if l == 0:
continue
resultp = ac.prototype(l)
results = ac.calc([l])
assert arr_eq(results[0], resultp, .001)
assert arr_eq(results[:, 0, :], 0, .001)
assert arr_eq(results[:, 7, :], 0, .001)
assert arr_eq(results[:, 1, 1], ds['config']['height2'], .001)
assert arr_eq(results[:, 2, 1], results[:, 3, 1])
assert arr_eq(results[:, 2, 1], -results[:, 4, 1])
assert arr_eq(results[:, 2, 1], -results[:, 5, 1])
assert arr_eq(results[:, 6, 1], -ds['config']['height2'])
