def setUp(self):
super(UserGroupResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.lt = self.ts.create_arbitrary_location_type()
self.top_lvl_location = self.ts.create_arbitrary_location(
self.lt.id, location_code='Nigeria', location_name='Nigeria')
def setUp(self):
super(AllMetaResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.lt = self.ts.create_arbitrary_location_type()
self.not_allowed_to_see_location = self.ts.create_arbitrary_location(
self.lt.id)
def setUp(self):
super(CampaignDataPointResourceTest, self).setUp()
# Create a user.
self.username = '******'
self.password = '******'
self.user = User.objects.create_user(self.username,\
'*****@*****.**', self.password)
self.get_credentials()
self.ts = TestSetupHelpers()
## create a metadata and data for us to use to test ##
self.create_metadata()
self.country_lt = LocationType.objects\
.create(name='Country',admin_level = 0)
self.region_lt = LocationType.objects\
.create(name='Region',admin_level = 1)
self.province_lt = LocationType.objects\
.create(name='Province',admin_level = 2)
self.top_lvl_location = Location.objects.get(name='Nigeria')
ltc = LocationTreeCache()
ltc.main()
LocationPermission.objects.create(user_id = self.user.id,\
top_lvl_location_id = self.top_lvl_location.id)
def setUp(self):
super(SyncODKResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.lt = self.ts.create_arbitrary_location_type()
self.top_lvl_location = self.ts.create_arbitrary_location(
self.lt.id, location_code='Mars', location_name='Mars')
def setUp(self):
super(SourceSubmissionResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.ts.load_some_metadata()
ltr = LocationTreeCache()
ltr.main()
self.mapped_location_id = self.ts.locations[0].id
self.loc_map = SourceObjectMap.objects.create(
source_object_code='AF001039003000000000',
content_type='location',
mapped_by_id=self.ts.user.id,
master_object_id=self.mapped_location_id)
source_campaign_string = '2016 March NID OPV'
self.mapped_campaign_id = self.ts.campaigns[0].id
self.campaign_map = SourceObjectMap.objects.create(
source_object_code=source_campaign_string,
content_type='campaign',
mapped_by_id=self.ts.user.id,
master_object_id=self.mapped_campaign_id)
self.mapped_indicator_with_data = self.ts.locations[2].id
self.indicator_map = SourceObjectMap.objects.create(
source_object_code='Percent missed due to other reasons',
content_type='indicator',
mapped_by_id=self.ts.user.id,
master_object_id=self.mapped_indicator_with_data)
def setUp(self):
super(IndicatorTagResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.lt = self.ts.create_arbitrary_location_type()
self.top_lvl_location = self.ts.create_arbitrary_location(self.lt.id)
def setUp(self):
## instantiate the test client and all other methods ##
super(DocDataPointResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.lt = self.ts.create_arbitrary_location_type()
self.top_lvl_location = self.ts.create_arbitrary_location(
self.lt.id, location_code='Nigeria', location_name='Nigeria')
def setUp(self):
super(IndicatorToTagResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.lt = self.ts.create_arbitrary_location_type()
self.top_lvl_location = self.ts.create_arbitrary_location(self.lt.id)
self.ind_tag = IndicatorTag.objects.create(tag_name='a tag')
self.ind = self.ts.create_arbitrary_indicator()
def setUp(self):
super(IndicatorResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.lt = self.ts.create_arbitrary_location_type()
self.top_lvl_location = self.ts.create_arbitrary_location(
self.lt.id)
LocationPermission.objects.create(user_id=self.ts.user.id,
top_lvl_location_id=self.top_lvl_location.id)
self.ind = self.ts.create_arbitrary_indicator()
def setUp(self):
self.ts = TestSetupHelpers()
data_df = read_csv('rhizome/tests/_data/calc_data.csv')
self.create_metadata()
self.user = User.objects.get(username="******")
self.test_df = data_df[data_df['is_raw'] == 1]
self.target_df = data_df[data_df['is_raw'] == 0]
self.campaign_id = Campaign.objects.all()[0].id
self.top_lvl_location = Location.objects.filter(name='Nigeria')[0]
ltr = LocationTreeCache()
ltr.main()
def setUp(self):
super(GeoResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.lt = LocationType.objects.create(name='Region', admin_level=1)
self.distr, created = \
LocationType.objects.get_or_create(name='District',admin_level = 2)
self.planet_location_type = LocationType.objects\
.create(name = 'Planet', admin_level = 0)
self.ultimate_parent = Location.objects.create(
id=1,
name='Earth',
location_code='Earth',
location_type_id=self.planet_location_type.id)
location_df_from_csv = read_csv(
'rhizome/tests/_data/locations_nimroz.csv')
self.ts.model_df_to_data(location_df_from_csv, Location)
# make sure that the proper level is set for the
locs = Location.objects.filter(parent_location_id=6)
for loc in locs:
loc.location_type_id = self.distr.id
loc.save()
parent = Location.objects.get(id=6)
parent.location_type_id = self.lt.id
parent.save()
geo_json_df = read_csv('rhizome/tests/_data/geo_json_small.txt',
delimiter="|")
location_df = DataFrame(list(Location.objects.all()\
.values_list('id','location_code')),columns=['location_id','location_code'])
location_tree_df = DataFrame(list(Location.objects.all()\
.values_list('id','parent_location_id'))\
,columns=['location_id','parent_location_id'])
location_tree_df['parent_location_id'].fillna(self.ultimate_parent.id,\
inplace=True)
location_tree_df['lvl'] = Series(1, index=location_tree_df.index)
self.ts.model_df_to_data(location_tree_df, LocationTree)
merged_df = location_df.merge(geo_json_df)[['location_id', 'geo_json']]
self.ts.model_df_to_data(merged_df, LocationPolygon)
minify_geo_json()
LocationPermission.objects.create(user_id=self.ts.user.id,
top_lvl_location_id=1)
def setUp(self):
super(DateDataPointResourceTest, self).setUp()
# Create a user.
self.username = '******'
self.password = '******'
self.user = User.objects.create_user(self.username,\
'*****@*****.**', self.password)
self.lt = LocationType.objects.create(name='Country', admin_level=0)
self.province_lt = LocationType.objects.create(name='Province'\
,admin_level = 1)
self.district_lt = LocationType.objects.create(name='District'\
,admin_level = 2)
self.ind = Indicator.objects.create(name='Polio Cases',
short_name='Polio Cases',
data_format='date_int')
self.top_lvl_location = Location.objects.create(
name='Afghanistan',
location_code='Afghanistan',
id=1234,
location_type_id=self.lt.id,
)
self.some_province = Location.objects.create(
name='Province',
location_code='Province',
id=432,
parent_location_id=self.top_lvl_location.id,
location_type_id=self.province_lt.id)
self.some_district = Location.objects.create(
name='Achin',
location_code='Achin',
id=4321,
parent_location_id=self.some_province.id,
location_type_id=self.district_lt.id)
ltc = LocationTreeCache()
ltc.main()
LocationPermission.objects.create(user_id = self.user.id,\
top_lvl_location_id = self.top_lvl_location.id)
self.get_credentials()
self.create_polio_cases()
self.ts = TestSetupHelpers()
self.doc_id = Document.objects.create(doc_title='Data Entry').id
def setUp(self):
super(QueueProcessResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.lt = self.ts.create_arbitrary_location_type()
self.top_lvl_location = self.ts.create_arbitrary_location(
self.lt.id,
location_code='Nigeria',
location_name='Nigeria')
# create a document, and one record for the soruce_submission #
self.doc = Document.objects.create(doc_title='new doc')
self.ss_obj = SourceSubmission.objects.create(
document_id = self.doc.id,
process_status = 'PROCESSED',
row_number = 0
)
def setUp(self):
super(DocumentResourceTest, self).setUp()
self.ts = TestSetupHelpers()
self.create_metadata()
def setUp(self):
super(RefreshMasterAPIResourceTest, self).setUp()
self.create_metadata()
self.ts = TestSetupHelpers()
def setUp(self):
super(AggRefreshAPITestCase, self).setUp()
self.ts = TestSetupHelpers()
# create some metadata
user_id = User.objects.create_user('test', '*****@*****.**', 'test').id
self.location_type1 = LocationType.objects.create(admin_level=0,
name="country", id=1)
self.location_type2 = LocationType.objects.create(admin_level=1,
name="province", id=2)
location_df = read_csv('rhizome/tests/_data/locations.csv')
indicator_df = read_csv('rhizome/tests/_data/indicators.csv')
self.locations = self.ts.model_df_to_data(location_df, Location)
self.indicators = self.ts.model_df_to_data(indicator_df, Indicator)
campaign_type1 = CampaignType.objects.create(name='test')
ind_tag = IndicatorTag.objects.create(tag_name='Polio')
self.campaign_id = Campaign.objects.create(
start_date='2016-01-01',
end_date='2016-01-02',
campaign_type_id=campaign_type1.id
).id
document = Document.objects.create(
doc_title='test',
created_by_id=user_id,
guid='test')
self.ss_1 = SourceSubmission.objects.create(
document_id=document.id,
submission_json='',
row_number=0,
data_date='2016-01-01'
).id
# create a couple datapoints
dp_1 = DataPoint.objects.create(
location_id=self.locations[21].id,
indicator_id=self.indicators[0].id,
campaign_id=self.campaign_id,
value=1,
data_date='2016-01-01',
source_submission_id=self.ss_1,
unique_index=1
)
dp_2 = DataPoint.objects.create(
location_id=self.locations[22].id,
indicator_id=self.indicators[0].id,
campaign_id=self.campaign_id,
value=1,
data_date='2016-01-01',
source_submission_id=self.ss_1,
unique_index=2
)
