Coverage for src / quber / core / extractors / mock.py: 100%

1import asyncio 

2from pathlib import Path 

3from typing import List, Optional 

4 

5from quber.core.extractors.base import ExtractedTable 

6 

7 

8class MockTableExtractor: 

9 def __init__(self, tables: Optional[List[ExtractedTable]] = None) -> None: 

10 self._tables = tables or [] 

11 

12 async def extract_tables(self, source: Path) -> List[ExtractedTable]: 

13 _ = source 

14 return list(self._tables) 

15 

16 def extract_tables_sync(self, source: Path) -> List[ExtractedTable]: 

17 return asyncio.run(self.extract_tables(source))