def test_constructor(self):
# missing data arg
with pytest.raises(TypeError):
# noinspection PyArgumentList
GenotypeDaskArray()
# data has wrong dtype
data = 'foo bar'
with pytest.raises(TypeError):
GenotypeDaskArray(data)
# data has wrong dtype
data = np.array([4., 5., 3.7])
with pytest.raises(TypeError):
GenotypeDaskArray(data)
# data has wrong dimensions
data = np.array([1, 2, 3])
with pytest.raises(TypeError):
GenotypeDaskArray(data)
# data has wrong dimensions
data = np.array([[1, 2], [3, 4]]) # use HaplotypeDaskArray instead
with pytest.raises(TypeError):
GenotypeDaskArray(data)
# diploid data (typed)
gd = self.setup_instance(np.array(diploid_genotype_data, dtype='i1'))
aeq(diploid_genotype_data, gd)
assert np.int8 == gd.dtype
# polyploid data (typed)
gd = self.setup_instance(np.array(triploid_genotype_data, dtype='i1'))
aeq(triploid_genotype_data, gd)
assert np.int8 == gd.dtype
def setup_instance(self, data, dtype=None):
z = zarr.array(data, dtype=dtype, chunks=(2, 2, None))
return GenotypeDaskArray(z)
def setup_instance(self, data, dtype=None):
return GenotypeDaskArray(np.asarray(data, dtype=dtype),
chunks=(2, 2, None))