def test_single_path_list_constructor(self):
path = Path([(0, 1), (1, 2), (2, 3), (3, 4)])
self.assertFalse(path.interface.multi)
self.assertEqual(path.dimension_values(0), np.array([
0, 1, 2, 3]))
self.assertEqual(path.dimension_values(1), np.array([
1, 2, 3, 4]))
def test_single_path_tuple_constructor(self):
path = Path(([0, 1, 2, 3], [1, 2, 3, 4]))
self.assertFalse(path.interface.multi)
self.assertEqual(path.dimension_values(0), np.array([
0, 1, 2, 3]))
self.assertEqual(path.dimension_values(1), np.array([
1, 2, 3, 4]))
def test_multi_path_unpack_single_paths(self):
path = Path([Path([(0, 1), (1, 2)]), Path([(2, 3), (3, 4)])])
self.assertTrue(path.interface.multi)
self.assertEqual(path.dimension_values(0), np.array([
0, 1, np.nan, 2, 3]))
self.assertEqual(path.dimension_values(1), np.array([
1, 2, np.nan, 3, 4]))
def test_multi_path_tuple(self):
path = Path(([0, 1], [[1, 3], [2, 4]]))
self.assertTrue(path.interface.multi)
self.assertEqual(path.dimension_values(0), np.array([
0, 1, np.nan, 0, 1]))
self.assertEqual(path.dimension_values(1), np.array([
1, 2, np.nan, 3, 4]))
def test_multi_path_list_constructor(self):
path = Path([[(0, 1), (1, 2)], [(2, 3), (3, 4)]])
self.assertTrue(path.interface.multi)
self.assertEqual(path.dimension_values(0), np.array([
0, 1, np.nan, 2, 3]))
self.assertEqual(path.dimension_values(1), np.array([
1, 2, np.nan, 3, 4]))
def test_multi_path_unpack_multi_paths(self):
path = Path([Path([[(0, 1), (1, 2)]]),
Path([[(2, 3), (3, 4)], [(4, 5), (5, 6)]])])
self.assertTrue(path.interface.multi)
self.assertEqual(path.dimension_values(0), np.array([
0, 1, np.nan, 2, 3, np.nan, 4, 5]))
self.assertEqual(path.dimension_values(1), np.array([
1, 2, np.nan, 3, 4, np.nan, 5, 6]))
def test_multi_path_cast_path(self):
path = Path([[(0, 1), (1, 2)], [(2, 3), (3, 4)]])
path2 = Path(path)
self.assertTrue(path2.interface.multi)
self.assertEqual(path2.dimension_values(0),
np.array([0, 1, np.nan, 2, 3]))
self.assertEqual(path2.dimension_values(1),
np.array([1, 2, np.nan, 3, 4]))