(dummy_dc_loader: load.DatasetCollectionLoader)
| 73 | |
| 74 | |
| 75 | def test_load_dataset(dummy_dc_loader: load.DatasetCollectionLoader): # pylint: disable=redefined-outer-name |
| 76 | with mock.patch.object(load, 'load', autospec=True) as mock_load: |
| 77 | examples = tf.data.Dataset.from_tensor_slices([1, 2, 3]) |
| 78 | expected = {'train': examples, 'test': examples} |
| 79 | mock_load.return_value = expected |
| 80 | loaded_dataset = dummy_dc_loader.load_dataset('c') |
| 81 | mock_load.assert_called_once_with(name='c/e:3.5.7', with_info=False) |
| 82 | assert loaded_dataset == expected |
| 83 | |
| 84 | |
| 85 | def test_load_dataset_split(dummy_dc_loader: load.DatasetCollectionLoader): # pylint: disable=redefined-outer-name |
nothing calls this directly
no test coverage detected