def test_duplicate_by_transposing_and_into_letters():
from config import get_configs
from load_songs_tools import get_raw_data
configs = get_configs()
configs["augment_data"] = False
configs["use_letternames"] = False
seqs, syms = get_raw_data(configs)
seqs, translation_dict = duplicate_by_transposing_and_into_letters(seqs)
# want to store augmented_seqs
fname_base = "%s-letters-augmented" % (configs["corpus"])
print "...writing chord strings to file"
fname = "%s.txt" % fname_base
with open(fname, "w") as p:
for seq in seqs:
for ch in seq:
p.write(ch + " ")
p.write("\n")
print ".finished writing chord strings to file"
print "...pickling chord strings to file"
fname = "%s.pkl" % fname_base
with open(fname, "wb") as p:
pickle.dump(seqs, p)
print ".finished pickling chord strings to file"
print "... pickling translation dict to file"
fname = "%s-translation_dict.pkl" % fname_base
with open(fname, "wb") as p:
pickle.dump(translation_dict, p)
print ".finished pickling translation dict to file"
def test_PreprocessedSeq():
configs = get_configs()
seqs, syms = get_raw_data(configs)
window = configs["window"]
seq = seqs[0]
seq[1] = 'a'
seq_p = PreprocessedSeq(seq, syms, configs["window"])
data = PreprocessedData(seqs[:2], syms, window)
