forked from weecology/EvergladesTools
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_nest_prediction.py
43 lines (33 loc) · 1.34 KB
/
test_nest_prediction.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#test nest_detection
import os
import sys
sys.path.append(os.path.dirname(os.getcwd()))
import pytest
import nest_detection
import geopandas
def test_load_files():
df = nest_detection.load_files("data/predictions/")
assert not df.empty
assert df.Site.unique() == ["Joule"]
def test_compare_site():
df = nest_detection.load_files("data/predictions/")
results = nest_detection.compare_site(df)
assert not results.empty
def test_detect_nests():
filename = nest_detection.detect_nests("data/predictions/",savedir="output/")
assert os.path.exists(filename)
#only run on hipergator with access to the rgb data
is_travis = 'TRAVIS' in os.environ
@pytest.mark.skipif(is_travis, reason="Cannot load comet on TRAVIS")
def test_detect_nests():
filename = nest_detection.detect_nests("data/predictions/",savedir="output/")
rgb_pool = nest_detection.find_files()
nest_detection.extract_nests(filename, rgb_pool=rgb_pool, savedir="output", upload=True)
gdf = geopandas.read_file(filename)
grouped = gdf.groupby("target_ind")
for name, group in grouped:
if group.shape[0] < 3:
continue
for index,row in group.iterrows():
filename = "output/{}/{}_{}_{}.png".format(index, row["target_ind"],row["Site"],row["Date"])
os.path.exists(filename)