예제 #1
0
 def test_three_f(self):
     self.assertEqual(internal.get_contiguous_strides((1, 2, 3), 4, False),
                      [4, 4, 8])
예제 #2
0
 def test_one_f(self):
     self.assertEqual(internal.get_contiguous_strides((1,), 2, False), [2])
예제 #3
0
 def test_two_f(self):
     self.assertEqual(internal.get_contiguous_strides((1, 2), 3, False),
                      [3, 3])
예제 #4
0
 def test_three(self):
     self.assertEqual(internal.get_contiguous_strides((1, 2, 3), 4, True),
                      [24, 12, 4])
예제 #5
0
 def test_zero_f(self):
     self.assertEqual(internal.get_contiguous_strides((), 1, False), [])
예제 #6
0
 def test_three_f(self):
     self.assertEqual(internal.get_contiguous_strides((1, 2, 3), 4, False),
                      [4, 4, 8])
예제 #7
0
 def test_two(self):
     self.assertEqual(internal.get_contiguous_strides((1, 2), 3, True),
                      [6, 3])
예제 #8
0
 def test_two_f(self):
     self.assertEqual(internal.get_contiguous_strides((1, 2), 3, False),
                      [3, 3])
예제 #9
0
 def test_one_f(self):
     self.assertEqual(internal.get_contiguous_strides((1,), 2, False), [2])
예제 #10
0
 def test_zero_f(self):
     self.assertEqual(internal.get_contiguous_strides((), 1, False), [])
예제 #11
0
 def test_three(self):
     self.assertEqual(internal.get_contiguous_strides((1, 2, 3), 4, True),
                      [24, 12, 4])
예제 #12
0
 def test_two(self):
     self.assertEqual(internal.get_contiguous_strides((1, 2), 3, True),
                      [6, 3])