mirror of
https://github.com/bellingcat/geoclustering.git
synced 2026-06-07 19:18:30 +03:00
26 lines
577 B
Python
26 lines
577 B
Python
from pathlib import Path
|
|
import shutil
|
|
from geoclustering.io import write_output_file
|
|
from tests.helpers import read_fixture_csv
|
|
|
|
|
|
def test_csv_filters():
|
|
df = read_fixture_csv("io.csv")
|
|
# entries 2 & 5 in fixture are valid.
|
|
assert len(df) == 2
|
|
assert df.iloc[0]["name"] == None
|
|
assert df.iloc[1]["name"] == "Bob"
|
|
|
|
|
|
def test_write_output_file():
|
|
p = "./this/dir/does/not/exist"
|
|
f = "test.txt"
|
|
write_output_file(p, f, "test")
|
|
|
|
path = Path(p) / f
|
|
|
|
with open(path) as f:
|
|
assert f.read() == "test"
|
|
|
|
shutil.rmtree(Path("./this"))
|