def test_query_dataset_io_with_fn(self):
ds_ios = query_dataset_io(lambda ds_io: ds_io.name == 'mem')
self.assertEqual(1, len(ds_ios))
self.assertIsInstance(ds_ios[0], MemDatasetIO)
def test_query_dataset_io(self):
ds_ios = query_dataset_io()
self.assertEqual(4, len(ds_ios))