test_segment.py 文件源码

def test_unit_scaling(self):
        a = np.random.rand(1)
        self.assertRaises(ValueError, segment.unit_scaling, a, 2)
        size = (3, 2)
        a = np.random.rand(*size)
        rst = segment.unit_scaling(a, 2)
        self.assertEqual(size, rst.shape)
        b = np.random.rand(size[1])
        self.assertRaises(ValueError, segment.unit_scaling, a, 2, b)
        b = np.random.rand(size[0])
        rst = segment.unit_scaling(a, 2, b)
        self.assertTrue(np.may_share_memory(a, rst))
        rst = segment.unit_scaling(a, 2, b, True)
        self.assertFalse(np.may_share_memory(a, rst))