diff --git a/tests/test_compare_vcfs.py b/tests/test_compare_vcfs.py index dd428f3..db5084d 100644 --- a/tests/test_compare_vcfs.py +++ b/tests/test_compare_vcfs.py @@ -185,6 +185,25 @@ def test_both_monoallelic(): assert np.array_equal(actual, expected) +def test_actg_alleles_true(): + ds1 = make_test_case() + ds2 = ds1.copy(deep=True) + actual_alleles, actual_genotypes = compare_vcfs.remap_genotypes(ds1, ds2, actg_alleles=True) + expected_alleles = np.array([ + [b'A', b'C', b'G', b'T'], + [b'A', b'C', b'G', b'T'], + ]) + expected_genotypes = xr.DataArray( + [ + [[0, 1], [1, 0]], + [[3, 2], [2, 3]], + ], + dims=["variants", "samples", "ploidy"] + ) + assert np.array_equal(actual_alleles, expected_alleles) + assert np.array_equal(actual_genotypes, expected_genotypes) + + @pytest.mark.parametrize( "arr1, arr2, expected", [