srsLTE/srsenb/test/mac/nr/sched_nr_prb_test.cc

131 lines
4.5 KiB
C++

/**
*
* \section COPYRIGHT
*
* Copyright 2013-2021 Software Radio Systems Limited
*
* By using this file, you agree to the terms and conditions set
* forth in the LICENSE file which can be found at the top level of
* the distribution.
*
*/
#include "srsenb/hdr/stack/mac/nr/sched_nr_rb.h"
#include "srsran/common/test_common.h"
using namespace srsenb;
using namespace srsenb::sched_nr_impl;
void test_bwp_prb_grant()
{
// TEST: default ctor
prb_grant grant;
TESTASSERT(grant.is_alloc_type1());
TESTASSERT(grant.prbs().length() == 0);
// TEST: assignment of RBGs
rbg_bitmap rbgs(18);
rbgs.set(1);
grant = rbgs;
TESTASSERT(grant.is_alloc_type0() and grant.rbgs().count() == 1);
// TEST: assignment of PRBs
prb_interval prb_interv(2, 5);
grant = prb_interv;
TESTASSERT(grant.is_alloc_type1() and grant.prbs().length() == 3);
// TEST: non-default ctor
prb_grant grant2(prb_interv), grant3(rbgs);
TESTASSERT(grant2.is_alloc_type1() and grant2.prbs().length() == 3);
TESTASSERT(grant3.is_alloc_type0() and grant3.rbgs().count() == 1);
// TEST: copy ctor
prb_grant grant4(grant2), grant5(grant3);
TESTASSERT(grant4.is_alloc_type1() and grant4.prbs().length() == 3);
TESTASSERT(grant5.is_alloc_type0() and grant5.rbgs().count() == 1);
// TEST: copy assignment
grant = grant3;
TESTASSERT(grant.is_alloc_type0() and grant.rbgs().count() == 1);
grant = grant2;
TESTASSERT(grant.is_alloc_type1() and grant.prbs().length() == 3);
}
void test_bwp_rb_bitmap()
{
bwp_rb_bitmap rb_bitmap(275, 0, true);
TESTASSERT(rb_bitmap.P() == 16);
TESTASSERT(rb_bitmap.rbgs().none());
TESTASSERT(rb_bitmap.prbs().none());
TESTASSERT(rb_bitmap.prbs().size() == 275 and rb_bitmap.nof_prbs() == 275);
TESTASSERT(rb_bitmap.rbgs().size() == 18 and rb_bitmap.nof_rbgs() == 18);
rb_bitmap.add(prb_interval{0, 1});
TESTASSERT(rb_bitmap.prbs().count() == 1 and rb_bitmap.prbs().test(0));
TESTASSERT(rb_bitmap.rbgs().count() == 1 and rb_bitmap.rbgs().test(0));
rb_bitmap.add(prb_interval{2, 4});
TESTASSERT(rb_bitmap.prbs().count() == 3 and rb_bitmap.prbs().test(2) and not rb_bitmap.prbs().test(1));
TESTASSERT(rb_bitmap.rbgs().count() == 1 and rb_bitmap.rbgs().test(0));
prb_bitmap prbs(rb_bitmap.nof_prbs());
prbs.set(1);
prbs.set(2);
prbs.set(15);
rb_bitmap.add(prbs);
TESTASSERT(rb_bitmap.prbs().count() == 5 and rb_bitmap.prbs().test(1) and rb_bitmap.prbs().test(15));
TESTASSERT(rb_bitmap.rbgs().count() == 1 and rb_bitmap.rbgs().test(0));
prbs.set(16);
rb_bitmap |= prbs;
TESTASSERT(rb_bitmap.prbs().count() == 6 and rb_bitmap.prbs().test(16));
TESTASSERT(rb_bitmap.rbgs().count() == 2 and rb_bitmap.rbgs().test(1));
rbg_bitmap rbgs(rb_bitmap.nof_rbgs());
rbgs.set(3);
rbgs.set(17);
rb_bitmap |= rbgs;
TESTASSERT(rb_bitmap.prbs().count() == (6 + 16 + 3) and rb_bitmap.prbs().test(rb_bitmap.nof_prbs() - 1));
TESTASSERT(rb_bitmap.rbgs().count() == 4 and rb_bitmap.rbgs().test(3) and rb_bitmap.rbgs().test(17));
rbgs.set(0);
rb_bitmap |= rbgs;
TESTASSERT(rb_bitmap.prbs().count() == (16 + 1 + 16 + 3) and rb_bitmap.prbs().test(rb_bitmap.nof_prbs() - 1));
TESTASSERT(rb_bitmap.rbgs().count() == 4 and rb_bitmap.rbgs().test(3) and rb_bitmap.rbgs().test(17));
// TEST: collides operator
TESTASSERT(rb_bitmap.collides(rbgs));
TESTASSERT(rb_bitmap.collides(prb_interval{0, 2}));
}
void test_bwp_rb_bitmap_search()
{
bwp_rb_bitmap rb_bitmap(275, 0, true);
prb_interval prbs = find_empty_interval_of_length(rb_bitmap.prbs(), 5);
TESTASSERT(prbs == prb_interval(0, 5));
prbs = find_empty_interval_of_length(rb_bitmap.prbs(), rb_bitmap.prbs().size());
TESTASSERT(prbs == prb_interval(0, rb_bitmap.prbs().size()));
rb_bitmap |= prb_interval{1, 5};
prbs = find_empty_interval_of_length(rb_bitmap.prbs(), rb_bitmap.prbs().size());
TESTASSERT(prbs == prb_interval(5, rb_bitmap.prbs().size()));
rb_bitmap |= prb_interval{16, 32};
prbs = find_empty_interval_of_length(rb_bitmap.prbs(), rb_bitmap.prbs().size());
TESTASSERT(prbs == prb_interval(32, rb_bitmap.prbs().size()));
rb_bitmap |= prb_interval{270, 275};
prbs = find_empty_interval_of_length(rb_bitmap.prbs(), rb_bitmap.prbs().size());
TESTASSERT(prbs == prb_interval(32, 270));
prbs = find_empty_interval_of_length(rb_bitmap.prbs(), 1);
TESTASSERT(prbs == prb_interval(0, 1));
prbs = find_empty_interval_of_length(rb_bitmap.prbs(), 5);
TESTASSERT(prbs == prb_interval(5, 10));
}
int main()
{
test_bwp_prb_grant();
test_bwp_rb_bitmap();
test_bwp_rb_bitmap_search();
}