fix nrPDSCH()

catkira · Oct 29, 2023 · 3c68b18 · 3c68b18
1 parent f94905e
commit 3c68b18
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/py3gpp/nrPDSCH.py b/py3gpp/nrPDSCH.py
@@ -26,6 +26,6 @@ def nrPDSCH(cws, modulation, nlayers, nid, nrnti):
 
     # TODO
     # res_sym = nrLayerMap(res_sym, nlayers)
-    res_sym = np.hstack(res_sym)
+    # res_sym = np.hstack(res_sym)
 
     return np.array(res_sym)
diff --git a/tests/test_nrPDSCH.py b/tests/test_nrPDSCH.py
@@ -14,7 +14,10 @@ def run_nr_pdsch(cws, modulation, nlayers, nid, nrnti):
     pdsch_syms = nrPDSCH(cws, modulation, nlayers, nid, nrnti)
     pdsch_syms = np.around(pdsch_syms, 4)
 
-    assert np.array_equal(pdsch_syms, pdsch_symbols_ref)
+    assert np.array_equal(pdsch_syms[0], pdsch_symbols_ref)
 
 def test_run_nr_pdsch():
     run_nr_pdsch([pdsch_bits_ref], ["64QAM"], 1, 0, 0)
+
+if __name__ == '__main__':
+    run_nr_pdsch([pdsch_bits_ref], ["64QAM"], 1, 0, 0)