def test_strided_slice_descriptor_ctor(): desc = ann.StridedSliceDescriptor([1, 2, 3, 4, 5], [6, 7, 8, 9, 10], (10, 10)) desc.m_Begin = [1, 2, 3, 4, 5] desc.m_End = [6, 7, 8, 9, 10] desc.m_Stride = (10, 10) assert [1, 2, 3, 4, 5] == desc.m_Begin assert [6, 7, 8, 9, 10] == desc.m_End assert [10, 10] == desc.m_Stride
def test_strided_slice_descriptor_default_values(): desc = ann.StridedSliceDescriptor() desc.m_Begin = [1, 2, 3, 4, 5] desc.m_End = [6, 7, 8, 9, 10] desc.m_Stride = (10, 10) desc.m_BeginMask = 1 desc.m_EndMask = 2 desc.m_ShrinkAxisMask = 3 desc.m_EllipsisMask = 4 desc.m_NewAxisMask = 5 assert [1, 2, 3, 4, 5] == desc.m_Begin assert [6, 7, 8, 9, 10] == desc.m_End assert [10, 10] == desc.m_Stride assert 1 == desc.m_BeginMask assert 2 == desc.m_EndMask assert 3 == desc.m_ShrinkAxisMask assert 4 == desc.m_EllipsisMask assert 5 == desc.m_NewAxisMask