import unittest
import os
from math import cos, sin

from litescope.software.dump import *

#TODO:
# - find a way to check if files are generated corectly

dump = Dump()
for i in range(4):
    dump.add(DumpVariable("ramp"+str(i), 2**i, [j for j in range(256)]))
pi = 3.1415
dump.add(DumpVariable("sin", 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)]))


class TestDump(unittest.TestCase):
    def test_csv(self):
        filename = "dump.csv"
        CSVDump(dump).write(filename)
        os.remove(filename)

    def test_py(self):
        filename = "dump.py"
        PythonDump(dump).write(filename)
        os.remove(filename)

    def test_sigrok(self):
        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)