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