def test_next_batch_order(): with tempdir(): os.mkdir(Constants.DEFAULT_LDIR) db = CsvDatabase() batch_id = db.next_batch_id() batch_order = db.next_batch_order(batch_id) assert batch_order == 1 batch_order = db.next_batch_order(batch_id) assert batch_order == 2