Fixup some tests

Author: jeromekelleher

bio2zarr/vcf.py

@@ -2056,8 +2056,7 @@ def get_max_encoding_memory(self):
                 "call_genotype_mask",
             ]
             gt_mem = sum(
-                self.schema.fields[col].variant_chunk_nbytes
-                for col in encoded_together
+                self.schema.fields[col].variant_chunk_nbytes for col in encoded_together
             )
         return max(max_encoding_mem, gt_mem)
 

tests/test_core.py

@@ -188,8 +188,8 @@ def test_5_chunk_1(self, n, expected):
         # It *might* work in CI, but it may well not either, as it's
         # probably dependent on a whole bunch of things. Expect to fail
         # at some point.
-        ("tests/data", 4630726),
-        ("tests/data/vcf", 4618589),
+        ("tests/data", 4960266),
+        ("tests/data/vcf", 4948129),
         ("tests/data/vcf/sample.vcf.gz", 1089),
     ],
 )

tests/test_vcf_examples.py

@@ -881,7 +881,7 @@ def test_split_explode(tmp_path):
         vcf.explode_partition(out, j)
     vcf.explode_finalise(out)
     pcvcf = vcf.IntermediateColumnarFormat(out)
-    assert pcvcf.columns["POS"].vcf_field.summary.asdict() == {
+    assert pcvcf.fields["POS"].vcf_field.summary.asdict() == {
         "num_chunks": 3,
         "compressed_size": 587,
         "uncompressed_size": 1008,