31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
import sys
|
|
import os
|
|
|
|
# Přidání adresáře src do sys.path, aby bylo možné importovat moduly
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'src'))
|
|
|
|
import data_fetcher
|
|
|
|
def test_download_yearly_data():
|
|
"""Test stahování ročních dat pro rok 2020."""
|
|
print("Spouštím test stahování ročních dat pro rok 2020...")
|
|
# Ujistěme se, že adresář data existuje
|
|
os.makedirs("data", exist_ok=True)
|
|
|
|
# Volání funkce pro stažení dat
|
|
file_path = data_fetcher.download_yearly_data(2020, output_dir="data")
|
|
|
|
# Ověření výsledku
|
|
if file_path and os.path.exists(file_path):
|
|
print(f"Test úspěšný: Data byla stažena a uložena do {file_path}")
|
|
|
|
# Zkontrolujme, zda soubor není prázdný
|
|
if os.path.getsize(file_path) > 0:
|
|
print("Test úspěšný: Soubor není prázdný.")
|
|
else:
|
|
print("Test selhal: Soubor je prázdný.")
|
|
else:
|
|
print("Test selhal: Nepodařilo se stáhnout data.")
|
|
|
|
if __name__ == "__main__":
|
|
test_download_yearly_data() |