def test_batchtospacend_descriptor_rubbish_assignment_crops(
input_crops, value, vtype):
desc = ann.BatchToSpaceNdDescriptor()
with pytest.raises(TypeError) as err:
desc.m_Crops = input_crops
assert "Failed to convert python input value {} of type '{}' to C type".format(
value, vtype) in str(err.value)
def test_batchtospacend_descriptor_assignment():
desc = ann.BatchToSpaceNdDescriptor()
desc.m_BlockShape = (1, 2, 3)
ololo = [(1, 2), (3, 4)]
size_1 = len(ololo)
desc.m_Crops = ololo
assert size_1 == len(ololo)
desc.m_DataLayout = ann.DataLayout_NHWC
assert ann.DataLayout_NHWC == desc.m_DataLayout
assert [1, 2, 3] == desc.m_BlockShape
assert [(1, 2), (3, 4)] == desc.m_Crops
def test_batchtospacend_descriptor_ctor():
desc = ann.BatchToSpaceNdDescriptor([1, 2, 3], [(4, 5), (6, 7)])
assert desc.m_DataLayout == ann.DataLayout_NCHW
assert [1, 2, 3] == desc.m_BlockShape
assert [(4, 5), (6, 7)] == desc.m_Crops
def test_batchtospacend_descriptor_empty_assignment():
desc = ann.BatchToSpaceNdDescriptor()
desc.m_BlockShape = []
assert [] == desc.m_BlockShape
def test_batchtospacend_descriptor_default_values():
desc = ann.BatchToSpaceNdDescriptor()
assert desc.m_DataLayout == ann.DataLayout_NCHW
assert [1, 1] == desc.m_BlockShape
assert [(0, 0), (0, 0)] == desc.m_Crops
