sim/cocotb/dual_parallel/test.py

   1 import cocotb
   2 from cocotb.triggers import Timer
   3 from cocotb.utils import get_sim_steps
   4 from cocotb.binary import BinaryValue
   5
   6 from c4m_jtag import JTAG_Master
   7
   8 @cocotb.test()
   9 def test01_dual(dut):
  10     """
  11     Test the IDCODE command
  12     """
  13
  14     # TODO: Allow parallel operation of the JTAG chains
  15
  16     # Run @ 1MHz
  17     clk_period = get_sim_steps(1, "us")
  18     master1 = JTAG_Master(dut.i1_tck, dut.i1_tms, dut.i1_tdi, dut.i1_tdo, dut.i1_trst_n, clk_period)
  19     master2 = JTAG_Master(dut.i2_tck, dut.i2_tms, dut.i2_tdi, dut.i2_tdo, dut.i2_trst_n, clk_period)
  20
  21     dut._log.info("Set command to SAMPLEPRELOAD")
  22     yield master1.load_ir(master1.SAMPLEPRELOAD)
  23     yield master2.load_ir(master2.SAMPLEPRELOAD)
  24
  25     dut._log.info("Load data, scan out first sample")
  26     yield master1.shift_data([0, 0, 0])
  27     dut._log.info("  master1 scan_out: {}".format(master1.result.binstr))
  28     assert(master1.result.binstr == "011")
  29     yield master2.shift_data([1, 1, 1])
  30     dut._log.info("  master2 scan_out: {}".format(master2.result.binstr))
  31     assert(master2.result.binstr == "101")
  32
  33     dut._log.info("Set command to EXTEST")
  34     yield master1.load_ir(master1.EXTEST)
  35     yield master2.load_ir(master2.EXTEST)
  36
  37     dut._log.info("Second scan")
  38     yield master1.shift_data([0, 0, 0])
  39     dut._log.info("  master1 scan_out: {}".format(master1.result.binstr))
  40     assert(master1.result.binstr == "111")
  41     yield master2.shift_data([1, 1, 1])
  42     dut._log.info("  master2 scan_out: {}".format(master2.result.binstr))
  43     assert(master2.result.binstr == "Z01")