add: compute_columns_for_custody_group tests

AllTests-mainnet.md

@@ -513,6 +513,11 @@ OK: 253/253 Fail: 0/253 Skip: 0/253
 OK: 56/56 Fail: 0/56 Skip: 0/56
 ## EF - PeerDAS - Networking [Preset: mainnet]
 ```diff
++ Networking - Compute Columns for Custody Group - mainnet/fulu/networking/compute_columns_f OK
++ Networking - Compute Columns for Custody Group - mainnet/fulu/networking/compute_columns_f OK
++ Networking - Compute Columns for Custody Group - mainnet/fulu/networking/compute_columns_f OK
++ Networking - Compute Columns for Custody Group - mainnet/fulu/networking/compute_columns_f OK
++ Networking - Compute Columns for Custody Group - mainnet/fulu/networking/compute_columns_f OK
 + Networking - Get Custody Groups - mainnet/fulu/networking/get_custody_groups/pyspec_tests/ OK
 + Networking - Get Custody Groups - mainnet/fulu/networking/get_custody_groups/pyspec_tests/ OK
 + Networking - Get Custody Groups - mainnet/fulu/networking/get_custody_groups/pyspec_tests/ OK
@@ -523,7 +528,7 @@ OK: 56/56 Fail: 0/56 Skip: 0/56
 + Networking - Get Custody Groups - mainnet/fulu/networking/get_custody_groups/pyspec_tests/ OK
 + Networking - Get Custody Groups - mainnet/fulu/networking/get_custody_groups/pyspec_tests/ OK
 ```
-OK: 9/9 Fail: 0/9 Skip: 0/9
+OK: 14/14 Fail: 0/14 Skip: 0/14
 ## EF - SSZ generic types
 ```diff
   Testing basic_vector inputs - invalid                                                      Skip
@@ -1144,4 +1149,4 @@ OK: 2/2 Fail: 0/2 Skip: 0/2
 OK: 9/9 Fail: 0/9 Skip: 0/9
 
 ---TOTAL---
-OK: 777/782 Fail: 0/782 Skip: 5/782
+OK: 782/787 Fail: 0/787 Skip: 5/787

beacon_chain/spec/peerdas_helpers.nim

@@ -25,7 +25,7 @@ type
   ProofBytes = array[fulu.CELLS_PER_EXT_BLOB, KzgProof]
 
 # https://github.com/ethereum/consensus-specs/blob/v1.5.0-alpha.10/specs/fulu/das-core.md#compute_columns_for_custody_group
-iterator compute_columns_for_custody_group(custody_group: CustodyIndex):
+iterator compute_columns_for_custody_group*(custody_group: CustodyIndex):
                                            ColumnIndex =
   for i in 0'u64 ..< COLUMNS_PER_GROUP:
     yield ColumnIndex(NUMBER_OF_CUSTODY_GROUPS * i + custody_group)

tests/consensus_spec/test_fixture_networking.nim

@@ -9,7 +9,7 @@
 {.used.}
 
 import
-  std/[json, streams],
+  std/[json, streams, sequtils],
   yaml,
   kzg4844/[kzg, kzg_abi],
   stint,
@@ -20,7 +20,28 @@ import
 
 from std/sequtils import mapIt
 
-proc runGetCustodyColumns(suiteName, path: string) =
+proc runComputeForCustodyGroup(suiteName, path: string) =
+  let relativeTestPathComponent = path.relativeTestPathComponent()
+  test "Networking - Compute Columns for Custody Group - " &
+      relativeTestPathComponent:
+    type TestMetaYaml = object
+      custody_group: uint64
+      result: seq[uint64]
+    let
+      meta = block:
+        var s = openFileStream(path/"meta.yaml")
+        defer: close(s)
+        var res: TestMetaYaml
+        yaml.load(s, res)
+        res
+      custody_group = meta.custody_group
+
+    var counter = 0
+    for column in compute_columns_for_custody_group(custody_group):
+      check column == meta.result[counter]
+      inc counter
+
+proc runGetCustodyGroups(suiteName, path: string) =
   let relativePathComponent = path.relativeTestPathComponent()
   test "Networking - Get Custody Groups - " & relativePathComponent:
     type TestMetaYaml = object
@@ -46,7 +67,13 @@ proc runGetCustodyColumns(suiteName, path: string) =
 suite "EF - PeerDAS - Networking" & preset():
   const presetPath = SszTestsDir/const_preset
   # foldering to be resolved in alpha 11 release of consensus spec tests
-  let basePath =
-    presetPath/"fulu"/"networking"/"get_custody_groups"/"pyspec_tests"
-  for kind, path in walkDir(basePath, relative = true, checkDir = true):
-    runGetCustodyColumns(suiteName, basePath/path)
+  block:
+    let basePath =
+      presetPath/"fulu"/"networking"/"get_custody_groups"/"pyspec_tests"
+    for kind, path in walkDir(basePath, relative = true, checkDir = true):
+      runGetCustodyGroups(suiteName, basePath/path)
+  block:
+    let basePath =
+      presetPath/"fulu"/"networking"/"compute_columns_for_custody_group"/"pyspec_tests"
+    for kind, path in walkDir(basePath, relative = true, checkDir = true):
+      runComputeForCustodyGroup(suiteName, basePath/path)