Exemple #1
0
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))
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())
Exemple #4
0
 def setUp(self):
     self.reader = BaseReader({}, meta())