mirror of
https://github.com/jhshi/openofdm.git
synced 2024-12-26 00:41:21 +00:00
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
|
|
# Authors: Veeresh Taranalli <veeresht@gmail.com>
|
|
# License: BSD 3 clause
|
|
|
|
from numpy import array, ones_like, arange
|
|
from numpy.testing import assert_array_almost_equal, assert_array_equal, assert_, assert_equal
|
|
from commpy.channelcoding.gfields import GF
|
|
|
|
|
|
class TestGaloisFields(object):
|
|
|
|
def test_closure(self):
|
|
for m in arange(1, 9):
|
|
x = GF(arange(2**m), m)
|
|
for a in x.elements:
|
|
for b in x.elements:
|
|
assert_((GF(array([a]), m) + GF(array([b]), m)).elements[0] in x.elements)
|
|
assert_((GF(array([a]), m) * GF(array([b]), m)).elements[0] in x.elements)
|
|
|
|
def test_addition(self):
|
|
m = 3
|
|
x = GF(arange(2**m), m)
|
|
y = GF(array([6, 4, 3, 1, 2, 0, 5, 7]), m)
|
|
z = GF(array([6, 5, 1, 2, 6, 5, 3, 0]), m)
|
|
assert_array_equal((x+y).elements, z.elements)
|
|
|
|
def test_multiplication(self):
|
|
m = 3
|
|
x = GF(array([7, 6, 5, 4, 3, 2, 1, 0]), m)
|
|
y = GF(array([6, 4, 3, 1, 2, 0, 5, 7]), m)
|
|
z = GF(array([4, 5, 4, 4, 6, 0, 5, 0]), m)
|
|
assert_array_equal((x*y).elements, z.elements)
|
|
|
|
def test_tuple_form(self):
|
|
m = 3
|
|
x = GF(arange(0, 2**m-1), m)
|
|
y = x.power_to_tuple()
|
|
z = GF(array([1, 2, 4, 3, 6, 7, 5]), m)
|
|
assert_array_equal(y.elements, z.elements)
|
|
|
|
def test_power_form(self):
|
|
m = 3
|
|
x = GF(arange(1, 2**m), m)
|
|
y = x.tuple_to_power()
|
|
z = GF(array([0, 1, 3, 2, 6, 4, 5]), m)
|
|
assert_array_equal(y.elements, z.elements)
|
|
m = 4
|
|
x = GF(arange(1, 2**m), m)
|
|
y = x.tuple_to_power()
|
|
z = GF(array([0, 1, 4, 2, 8, 5, 10, 3, 14, 9, 7, 6, 13, 11, 12]), m)
|
|
assert_array_equal(y.elements, z.elements)
|
|
|
|
def test_order(self):
|
|
m = 4
|
|
x = GF(arange(1, 2**m), m)
|
|
y = x.order()
|
|
z = array([1, 15, 15, 15, 15, 3, 3, 5, 15, 5, 15, 5, 15, 15, 5])
|
|
assert_array_equal(y, z)
|
|
|
|
def test_minpols(self):
|
|
m = 4
|
|
x = GF(arange(2**m), m)
|
|
z = array([2, 3, 19, 19, 19, 19, 7, 7, 31, 25, 31, 25, 31, 25, 25, 31])
|
|
assert_array_equal(x.minpolys(), z)
|
|
m = 6
|
|
x = GF(array([2, 8, 32, 6, 24, 35, 10, 40, 59, 41, 14, 37]), m)
|
|
z = array([67, 87, 103, 73, 13, 109, 91, 117, 7, 115, 11, 97])
|
|
assert_array_equal(x.minpolys(), z)
|