Add weeks to instant check

openfisca · Mar 11, 2021 · c557909 · c557909
1 parent 8f580d4
commit c557909
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openfisca_core/periods.py b/openfisca_core/periods.py
@@ -203,7 +203,7 @@ def offset(self, offset, unit):
         Instant((2014, 12, 31))
         """
         year, month, day = self
-        assert unit in (DAY, MONTH, YEAR), 'Invalid unit: {} of type {}'.format(unit, type(unit))
+        assert unit in (WEEKDAY, WEEK, DAY, MONTH, YEAR), 'Invalid unit: {} of type {}'.format(unit, type(unit))
         if offset == 'first-of':
             if unit == MONTH:
                 day = 1