def test_determine_nsets2D_with_extra_dimension2(self):
b = np.array([[[2, 3], [4, 5]], [[6, 7], [8, 9]]])
nb = cost._determine_nsets(b, b)
return self.assertEqual(nb, 2)
def test_determine_nsets2D_simple2(self):
b = np.array([[[2, 3], [4, 5]], [[6, 7], [8, 9]]])
nb = cost._determine_nsets(b, b)
return self.assertEqual(nb, 2)
def test_determine_nsets1D_with_extra_dimension1(self):
a = np.array([[2, 3, 4, 5]])
na = cost._determine_nsets(a, None)
return self.assertEqual(na, 1)
def test_determine_nsets1D_with_extra_dimension2(self):
b = np.array([[2], [3], [4], [5]])
nb = cost._determine_nsets(b, None)
return self.assertEqual(nb, 1)
def test_determine_nsets1D_simple(self):
a = np.array([[2, 3], [4, 5]])
na = cost._determine_nsets(a, None)
return self.assertEqual(na, 2)