def test_github_dataset():
cardset = forecastcards.Cardset(data_loc = gh_data)
file_to_project_id= cardset.file_to_project_id
card_locs_by_type = cardset.card_locs_by_type
dataset = forecastcards.Dataset(card_locs_by_type = card_locs_by_type, file_to_project_id= file_to_project_id )
#print(dataset.df.shape)
assert(dataset.df.shape[0] == 16 )
def check_project_id(self, compatable_cardsets):
'''
check to see if project should be added
:returns: True if project should be added; False if it shouldn't
'''
cs_with_project_id = []
for cs in compatable_cardsets:
if isinstance(cs, forecastcards.Cardset):
if self.project_id in cs.projects:
cs_with_project_id.append(cs)
continue
elif cs.find('github') >= 0:
cs = forecastcards.github_url_to_dict(cs)
cs = forecastcards.Cardset(data_loc=cs_loc, validate=False)
if self.project_id in cs.projects:
cs_with_project_id.append(cs_loc_orig)
if cs_with_project_id:
print("Duplicate Project ID found in following cardsets:",
"\n".join(cs_with_project_id))
return False
else:
return True
#!/usr/bin/env python
import os
import sys
import forecastcards
USAGE = """
python forecastcards/scripts/validate_cardset.py "/Users/elizabeth/Documents/urbanlabs/NCHRP 08-110/working/forecastcards/forecastcards/examples"
"""
if __name__ == "__main__":
if len(sys.argv) < 2:
data_loc = "."
else:
data_loc = sys.argv[1]
cardset = forecastcards.Cardset(data_loc=data_loc)
if not cardset.invalid_projects:
print("Cardset Valid!")
else:
print("Invalid Project: ", cardset.invalid_projects)
print(cardset.failed_reports)
def test_local_add_github_cardset():
cardset = forecastcards.Cardset(data_loc=ex_data,
exclude_projects=['lu123'])
cardset.add_projects(data_loc=gh_data, select_projects=['lu123'])
assert (valid_project_control == set(cardset.validated_projects))
def test_github_cardset():
cardset = forecastcards.Cardset(data_loc=gh_data)
assert (valid_project_control == set(cardset.validated_projects))
def test_local_cardset():
#forecastcards.Cardset(data_loc = ex_data)
cardset = forecastcards.Cardset()
assert (valid_project_control == set(cardset.validated_projects))
