def setUp(self): super(ReaderTest, self).setUp() with absltest.mock.patch.object(example_parser, 'ExampleParser', testing.DummyParser): self.reader = tfrecords_reader.Reader(self.tmp_dir, 'some_spec') self.reader.read = functools.partial( self.reader.read, read_config=read_config_lib.ReadConfig(), shuffle_files=False, )
def _tfrecords_reader(self): return tfrecords_reader.Reader(self._data_dir, self._example_specs)
def setUp(self): super(ReaderTest, self).setUp() with absltest.mock.patch.object(example_parser, 'ExampleParser', testing.DummyParser): self.reader = tfrecords_reader.Reader(self.tmp_dir, 'some_spec')