class BaseReaderTest(unittest.TestCase): def setUp(self): self.reader = BaseReader({}, meta()) def test_get_next_batch_not_implemented(self): with self.assertRaises(NotImplementedError): self.reader.get_next_batch() def test_set_last_position(self): self.reader.set_last_position(dict(position=5)) self.assertEqual(self.reader.last_position, dict(position=5))
def setUp(self): self.reader = BaseReader({}, meta())