code/network-generator/tests/test_o_ran_spiral_radius_profile.py

   1 # Copyright 2023 highstreet technologies GmbH
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #     http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 from network_generation.model.python.hexagon import Hex
  16 from network_generation.model.python.o_ran_spiral_radius_profile import (
  17     SpiralRadiusProfile,
  18 )
  19
  20
  21 def test_o_ran_spiral_radius_profile() -> None:
  22     srp: SpiralRadiusProfile = SpiralRadiusProfile()
  23     assert srp.count == 7 * 7 * 7 * 7
  24     assert srp.id == "1111"
  25
  26     assert str(srp.oRanDuSpiral(Hex(0, 0, 0), 0)[0]) == "q: 0, r: 0, s: 0"
  27
  28     assert str(srp.oRanCuSpiral(Hex(1, 2, -3), 1)[1]) == "q: -6, r: 9, s: -3"
  29
  30     assert (
  31         str(srp.oRanNearRtRicSpiral(Hex(-1, 1, 0), 2)[2])
  32         == "q: 6, r: 15, s: -21"
  33     )