Skip to content

Commit

Permalink
Update default fields for farm group (#169)
Browse files Browse the repository at this point in the history
  • Loading branch information
meomancer authored Oct 2, 2024
1 parent be2d41d commit 4993ebc
Show file tree
Hide file tree
Showing 2 changed files with 63 additions and 18 deletions.
6 changes: 5 additions & 1 deletion django_project/gap/models/crop_insight.py
Original file line number Diff line number Diff line change
Expand Up @@ -328,7 +328,11 @@ def default_fields_used():
'latitude',
'longitude',
'SPWTopMessage',
'SPWDescription'
'SPWDescription',
'TooWet',
'last_4_days_mm',
'last_2_days_mm',
'today_tomorrow_mm'
]

def __init__(
Expand Down
75 changes: 58 additions & 17 deletions django_project/spw/tests/test_crop_insight_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,10 @@ class TestCropInsightGenerator(TestCase):
csv_headers = [
'farmID', 'phoneNumber', 'latitude', 'longitude', 'SPWTopMessage',
'SPWDescription',
'TooWet',
'last_4_days_mm',
'last_2_days_mm',
'today_tomorrow_mm',
'day1_mm', 'day1_Chance', 'day1_Type',
'day2_mm', 'day2_Chance', 'day2_Type',
'day3_mm', 'day3_Chance', 'day3_Type',
Expand Down Expand Up @@ -333,9 +337,18 @@ def mock_send_fn(self, fail_silently=False):
'Both current forecast '
'historical rains have good signal to plant.'
)
self.assertEqual(row[6], '10.0') # Precip (daily)
self.assertEqual(row[7], '50.0') # Precip % chance
self.assertEqual(row[8], 'Light rain') # Precip Type
# Too wet column
self.assertEqual(
row[6], 'Likely too wet to plant'
) # tooWet
self.assertEqual(row[7], '80.0') # last4Days
self.assertEqual(row[8], '60.0') # last2Days
self.assertEqual(row[9], '40.0') # todayTomorrow

# First day forecast
self.assertEqual(row[10], '10.0') # Precip (daily)
self.assertEqual(row[11], '50.0') # Precip % chance
self.assertEqual(row[12], 'Light rain') # Precip Type

# Farm 2
elif row_num == 3:
Expand All @@ -349,9 +362,16 @@ def mock_send_fn(self, fail_silently=False):
self.assertEqual(
row[5], 'Wait for more positive forecast.'
)
self.assertEqual(row[6], '0.5') # Precip (daily)
self.assertEqual(row[7], '10.0') # Precip % chance
self.assertEqual(row[8], 'No Rain') # Precip Type
# Too wet column
self.assertEqual(row[6], 'Too wet to plant') # tooWet
self.assertEqual(row[7], '100.0') # last4Days
self.assertEqual(row[8], '80.0') # last2Days
self.assertEqual(row[9], '80.0') # todayTomorrow

# First day forecast
self.assertEqual(row[10], '0.5') # Precip (daily)
self.assertEqual(row[11], '10.0') # Precip % chance
self.assertEqual(row[12], 'No Rain') # Precip Type

# Farm 3
elif row_num == 4:
Expand All @@ -363,9 +383,17 @@ def mock_send_fn(self, fail_silently=False):
self.assertEqual(row[3], '50.2222') # Longitude
self.assertEqual(row[4], '')
self.assertEqual(row[5], '')
self.assertEqual(row[6], '') # Precip (daily)
self.assertEqual(row[7], '') # Precip % chance
self.assertEqual(row[8], '') # Precip Type

# Too wet column
self.assertEqual(row[6], '') # tooWet
self.assertEqual(row[7], '') # last4Days
self.assertEqual(row[8], '') # last2Days
self.assertEqual(row[9], '') # todayTomorrow

# First day forecast
self.assertEqual(row[10], '') # Precip (daily)
self.assertEqual(row[11], '') # Precip % chance
self.assertEqual(row[12], '') # Precip Type

# Farm 4 has same grid with farm 2
elif row_num == 5:
Expand All @@ -379,9 +407,16 @@ def mock_send_fn(self, fail_silently=False):
self.assertEqual(
row[5], 'Wait for more positive forecast.'
)
self.assertEqual(row[6], '0.5') # Precip (daily)
self.assertEqual(row[7], '10.0') # Precip % chance
self.assertEqual(row[8], 'No Rain') # Precip Type
# Too wet column
self.assertEqual(row[6], 'Too wet to plant') # tooWet
self.assertEqual(row[7], '100.0') # last4Days
self.assertEqual(row[8], '80.0') # last2Days
self.assertEqual(row[9], '80.0') # todayTomorrow

# First day forecast
self.assertEqual(row[10], '0.5') # Precip (daily)
self.assertEqual(row[11], '10.0') # Precip % chance
self.assertEqual(row[12], 'No Rain') # Precip Type

# Farm 5 has same grid with farm 3
elif row_num == 6:
Expand All @@ -391,11 +426,17 @@ def mock_send_fn(self, fail_silently=False):
self.assertEqual(row[1], self.farm_5.phone_number)
self.assertEqual(row[2], '50.0') # Latitude
self.assertEqual(row[3], '50.2222') # Longitude
self.assertEqual(row[4], '')
self.assertEqual(row[5], '')
self.assertEqual(row[6], '') # Precip (daily)
self.assertEqual(row[7], '') # Precip % chance
self.assertEqual(row[8], '') # Precip Type

# Too wet column
self.assertEqual(row[6], '') # tooWet
self.assertEqual(row[7], '') # last4Days
self.assertEqual(row[8], '') # last2Days
self.assertEqual(row[9], '') # todayTomorrow

# First day forecast
self.assertEqual(row[10], '') # Precip (daily)
self.assertEqual(row[11], '') # Precip % chance
self.assertEqual(row[12], '') # Precip Type
row_num += 1

@patch('spw.generator.crop_insight.CropInsightFarmGenerator.generate_spw')
Expand Down

0 comments on commit 4993ebc

Please sign in to comment.