def test_get_ids(self):
ds = Dataset.from_array(
[[10, 20, 30], [20, 10, 50], [40, 10, 30]], # data
['i1', 'i2', 'i3'], # ids
static=True)
actual_ids = []
expected_ids = ['i1', 'i2', 'i3']
for row_id in ds.get_ids():
actual_ids.append(row_id)
self.assertEqual(expected_ids, actual_ids)
ds = Dataset.from_array(
[[10, 20, 30], [20, 10, 50], [40, 10, 30]], # data
['i1', 'i2', 'i3'], # ids
static=False)
self.assertRaises(RuntimeError, list, ds.get_ids())
def test_get_ids(self):
ds = Dataset.from_array(
[ [10, 20, 30], [20, 10, 50], [40, 10, 30]], # data
['i1', 'i2', 'i3'], # ids
static=True
)
actual_ids = []
expected_ids = ['i1', 'i2', 'i3']
for row_id in ds.get_ids():
actual_ids.append(row_id)
self.assertEqual(expected_ids, actual_ids)
ds = Dataset.from_array(
[ [10, 20, 30], [20, 10, 50], [40, 10, 30]], # data
['i1', 'i2', 'i3'], # ids
static=False
)
self.assertRaises(RuntimeError, list, ds.get_ids())
def _make_stub_dataset(self):
ids = ['id1', 'id2', 'id3', 'id4', 'id5']
X = [
[0, 0, 0],
[1, 1, 1],
[2, 2, 2],
[3, 3, 3],
[4, 4, 4]
]
dataset = Dataset.from_array(X, ids=ids)
return dataset
def test_from_array_without_ids(self):
ds = Dataset.from_array(
[[10, 20, 30], [20, 10, 50], [40, 10, 30]], # data
feature_names=['k1', 'k2', 'k3'] # feature names
)
expected_k1s = [10, 20, 40]
actual_k1s = []
actual_ids = []
for (idx, (row_id, d)) in ds:
actual_k1s.append(dict(d.num_values).get('k1', None))
actual_ids.append(row_id)
self.assertEqual(expected_k1s, actual_k1s)
self.assertEqual(len(actual_ids), 3)
def test_from_array_without_ids(self):
ds = Dataset.from_array(
[ [10, 20, 30], [20, 10, 50], [40, 10, 30]], # data
feature_names=['k1', 'k2', 'k3'] # feature names
)
expected_k1s = [10, 20, 40]
actual_k1s = []
actual_ids = []
for (idx, (row_id, d)) in ds:
actual_k1s.append(dict(d.num_values).get('k1', None))
actual_ids.append(row_id)
self.assertEqual(expected_k1s, actual_k1s)
self.assertEqual(len(actual_ids), 3)
def _make_stub_dataset(self):
ids = ['id1', 'id2', 'id3', 'id4', 'id5']
X = [[0, 0, 0], [1, 1, 1], [2, 2, 2], [3, 3, 3], [4, 4, 4]]
dataset = Dataset.from_array(X, ids=ids)
return dataset