Skip to content

Commit b9259f3

Browse files
dmalzldmalzl
authored andcommitted
add test for dtype propagation
1 parent bf0a842 commit b9259f3

2 files changed

Lines changed: 14 additions & 1 deletion

File tree

tests/data/float_counts.cool

3.92 MB
Binary file not shown.

tests/test_rearrange_cooler.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def test_rearrange_cooler(request):
6262

6363
# III.
6464
# Check that when taking two chromosomes in a different order and inverting one,
65-
# all is ax espected
65+
# all is as espected
6666

6767
reorder_invert_view = (
6868
orig_view.iloc[1::-1].assign(strand=["+", "-"]).reset_index(drop=True)
@@ -136,3 +136,16 @@ def test_rearrange_cooler(request):
136136
::-1, :
137137
]
138138
assert np.array_equal(old_trans_m, reordered_inverted_trans_m, equal_nan=True)
139+
140+
# IV.
141+
# Check that pixel datatypes are propagted to output cooler
142+
float_clr = cooler.Cooler(op.join(request.fspath.dirname, "data/float_counts.cool"))
143+
new_view = (
144+
bioframe
145+
.make_viewframe(float_clr.chromsizes)
146+
.sort_values(['chrom', 'start', 'end'])
147+
)
148+
rearrange_cooler(float_clr, new_view, "test_propagated_datatypes.cool")
149+
150+
propagate_clr = cooler.Cooler("test_propagated_datatypes.cool")
151+
assert_frame_equal(float_clr.pixels()[:], propagate_clr.pixels()[:])

0 commit comments

Comments
 (0)