def test_space_to_batch_nd_descriptor_assigned_values(): desc = ann.SpaceToBatchNdDescriptor() desc.m_BlockShape = (90, 100) desc.m_PadList = [(1, 2), (3, 4)] assert [90, 100] == desc.m_BlockShape assert [(1, 2), (3, 4)] == desc.m_PadList assert ann.DataLayout_NCHW == desc.m_DataLayout
def test_space_to_batch_nd_descriptor_ctor(): desc = ann.SpaceToBatchNdDescriptor((1, 2, 3), [(1, 2), (3, 4)]) assert [1, 2, 3] == desc.m_BlockShape assert [(1, 2), (3, 4)] == desc.m_PadList assert ann.DataLayout_NCHW == desc.m_DataLayout
def test_space_to_batch_nd_descriptor_default_values(): desc = ann.SpaceToBatchNdDescriptor() assert [1, 1] == desc.m_BlockShape assert [(0, 0), (0, 0)] == desc.m_PadList assert ann.DataLayout_NCHW == desc.m_DataLayout
def test_spacetobatchnd_descriptor_default_values(): desc = ann.SpaceToBatchNdDescriptor() assert desc.m_DataLayout == ann.DataLayout_NCHW