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'])