def test_processing():
    inp = 'raw_data/test.json'
    outp = "raw_data/test.npz"
    process_creativebirds_json(in_path=inp, out_path=outp)
    a = np.load(outp, allow_pickle=True, encoding='latin1')
    drawing = Drawing.from_npz_data(a['train'][1])
    drawing.plot()
def test_one(idx=0, in_path='raw_data/test.json'):
    data = json.loads(open(in_path).read())
    # print(data)
    cc = CreativeItem.from_json_item(data[idx])
    cc.plot()
    ll = LineStringList.fromCreativeItem(cc)
    ll.simplify()
    ll.normalize()
    ll.plot()
    print(ll.linestring_list[:5])
    dd = DatasetItem.fromLineStringList(ll)
    print(dd.embedding[:5])
    dd.plot()
    print(dd.embedding.shape)
    print(dd.embedding.dtype)
    drawing = Drawing.from_npz_data(dd.embedding)
    drawing.plot()