test/test_dump: cleanup

This commit is contained in:
Florent Kermarrec 2017-01-19 05:30:25 +01:00
parent 687d09f331
commit a80d890fb9
1 changed files with 31 additions and 20 deletions

View File

@ -1,28 +1,39 @@
import unittest import unittest
import os
from math import cos, sin
from litescope.software.dump import * from litescope.software.dump import *
class TestDump(unittest.TestCase): #TODO:
def test(self): # - find a way to check if files are generated corectly
print("creating dump...")
dump = Dump() dump = Dump()
for i in range(4): for i in range(4):
dump.add(DumpVariable("foo"+str(i), 2**i, [j for j in range(256)])) dump.add(DumpVariable("ramp"+str(i), 2**i, [j for j in range(256)]))
pi = 3.1415 pi = 3.1415
from math import cos, sin dump.add(DumpVariable("sin", 8, [128+128*sin(j/(2*pi*16)) for j in range(1024)]))
dump.add(DumpVariable("sinus", 8, [128+128*sin(j/(2*pi*16)) for j in range(1024)])) dump.add(DumpVariable("cos", 8, [128+128*cos(j/(2*pi*16)) for j in range(1024)]))
dump.add(DumpVariable("cosinus", 8, [128+128*cos(j/(2*pi*16)) for j in range(1024)]))
print("csv export test")
CSVDump(dump).write("dump.csv")
print("python export test...") class TestDump(unittest.TestCase):
PythonDump(dump).write("dump.py") def test_csv(self):
filename = "dump.csv"
CSVDump(dump).write(filename)
os.remove(filename)
print("sigrok export/import test...") def test_py(self):
SigrokDump(dump).write("dump.sr") filename = "dump.py"
SigrokDump(dump).read("dump.sr") PythonDump(dump).write(filename)
SigrokDump(dump).write("dump.sr") os.remove(filename)
print("vcd export test...") def test_sigrok(self):
VCDDump(dump).write("dump.vcd") filename = "dump.sr"
SigrokDump(dump).write(filename)
SigrokDump(dump).read(filename)
SigrokDump(dump).write(filename)
os.remove(filename)
def test_vcd(self):
filename = "dump.vcd"
VCDDump(dump).write(filename)
os.remove(filename)