Add unit tests for column_space()

scripts/generate_field_test_vectors.py

@@ -505,9 +505,6 @@ def make_luts(field, sub_folder, seed, sparse=False):
  X.append(np.array([[I(e) for e in row] for row in x], dtype))
  Z.append(z)
 
- x = copy(x)
- x[j,:] = FIELD.random_element() * x[0,:]
-
  # Zero matrix
  x = copy(x)
  x[:] = F(0)
@@ -523,6 +520,54 @@ def make_luts(field, sub_folder, seed, sparse=False):
  d = {"X": X, "Z": Z}
  save_pickle(d, folder, "row_space.pkl")
 
+ set_seed(seed + 209)
+ shapes = [(2,2), (2,3), (2,4), (3,2), (4,2), (3,3)]
+ X = []
+ Z = []
+ for i in range(len(shapes)):
+ deg = shapes[i][0] # The degree of the vector space
+
+ # Random matrix
+ x = randint_matrix(0, order, shapes[i])
+ X.append(x)
+ x = matrix(FIELD, [[F(e) for e in row] for row in x])
+ z = x.column_space()
+ if z.dimension() == 0:
+ z = randint_matrix(0, 1, (0, deg))
+ else:
+ z = z.basis_matrix()
+ z = np.array([[I(e) for e in row] for row in z], dtype)
+ Z.append(z)
+
+ # Reduce the column space by 1 by copying the 0th column to the jth column
+ for j in range(1, shapes[i][1]):
+ x = copy(x)
+ x[:,j] = FIELD.random_element() * x[:,0]
+
+ z = x.column_space()
+ if z.dimension() == 0:
+ z = randint_matrix(0, 1, (0, deg))
+ else:
+ z = z.basis_matrix()
+ z = np.array([[I(e) for e in row] for row in z], dtype)
+ X.append(np.array([[I(e) for e in row] for row in x], dtype))
+ Z.append(z)
+
+ # Zero matrix
+ x = copy(x)
+ x[:] = F(0)
+ z = x.column_space()
+ if z.dimension() == 0:
+ z = randint_matrix(0, 1, (0, deg))
+ else:
+ z = z.basis_matrix()
+ z = np.array([[I(e) for e in row] for row in z], dtype)
+ X.append(np.array([[I(e) for e in row] for row in x], dtype))
+ Z.append(z)
+
+ d = {"X": X, "Z": Z}
+ save_pickle(d, folder, "column_space.pkl")
+
  ###############################################################################
  # Polynomial arithmetic
  ###############################################################################

tests/fields/conftest.py

@@ -335,6 +335,15 @@ def field_row_space(field_folder):
  return d
 
 
+@pytest.fixture(scope="session")
+def field_column_space(field_folder):
+ GF, d = read_pickle(field_folder, "column_space.pkl")
+ d["GF"] = GF
+ d["X"] = [GF(x) for x in d["X"]]
+ d["Z"] = [GF(z) for z in d["Z"]]
+ return d
+
+
 ###############################################################################
 # Fixtures for arithmetic methods over finite fields
 ###############################################################################

tests/fields/test_linalg.py

@@ -447,3 +447,24 @@ def test_row_space(field_row_space):
  z = x.row_space()
  assert np.array_equal(z, Z[i])
  assert type(z) is GF
+
+
+def test_column_space_exceptions():
+ GF = galois.GF(2**8)
+ with pytest.raises(ValueError):
+ A = GF.Random(5)
+ A.column_space()
+ with pytest.raises(ValueError):
+ A = GF.Random((2,2,2))
+ A.column_space()
+
+
+def test_column_space(field_column_space):
+ GF, X, Z = field_column_space["GF"], field_column_space["X"], field_column_space["Z"]
+
+ for i in range(len(X)):
+ dtype = random.choice(GF.dtypes)
+ x = X[i].astype(dtype)
+ z = x.column_space()
+ assert np.array_equal(z, Z[i])
+ assert type(z) is GF