<s> from datetime import date , timedelta <EOL> from kansha . cardextension . tests import CardExtensionTestCase <EOL> from . comp import DueDate <EOL> class CardDueDateTest ( CardExtensionTestCase ) : <EOL> extension_name = '<STR_LIT>' <EOL> extension_class = DueDate <EOL> def test_due_date ( self ) : <EOL> today = date . today ( ) <EOL> self . assertIsNone ( self . extension . due_date ) <EOL> self . assertIsNone ( self . extension . get_value ( ) ) <EOL> self . extension . set_value ( today ) <EOL> self . assertEqual ( self . extension . get_value ( ) , today ) <EOL> def test_days_count ( self ) : <EOL> today = date . today ( ) <EOL> self . extension . set_value ( today ) <EOL> self . assertEqual ( self . extension . get_days_count ( ) , <NUM_LIT:0> ) <EOL> self . extension . set_value ( today + timedelta ( days = <NUM_LIT:5> ) ) <EOL> self . assertEqual ( self . extension . get_days_count ( ) , - <NUM_LIT:5> ) <EOL> self . extension . set_value ( today - timedelta ( days = <NUM_LIT:5> ) ) <EOL> self . assertEqual ( self . extension . get_days_count ( ) , <NUM_LIT:5> ) <EOL> def test_class ( self ) : <EOL> today = date . today ( ) <EOL> self . extension . set_value ( today ) <EOL> self . assertEqual ( self . extension . get_class ( ) , '<STR_LIT>' ) <EOL> self . extension . set_value ( today + timedelta ( days = <NUM_LIT:1> ) ) <EOL> self . assertEqual ( self . extension . get_class ( ) , '<STR_LIT>' ) <EOL> self . extension . set_value ( today + timedelta ( days = <NUM_LIT:10> ) ) <EOL> self . assertEqual ( self . extension . get_class ( ) , '<STR_LIT>' ) <EOL> self . extension . set_value ( today - timedelta ( days = <NUM_LIT:1> ) ) <EOL> self . assertEqual ( self . extension . get_class ( ) , '<STR_LIT>' ) <EOL> self . extension . set_value ( today - timedelta ( days = <NUM_LIT:10> ) ) <EOL> self . assertEqual ( self . extension . get_class ( ) , '<STR_LIT>' ) </s>
