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