<s> from django . test import TestCase <EOL> from django . contrib . formtools . wizard . storage import get_storage , MissingStorage <EOL> from django . contrib . formtools . wizard . storage . base import BaseStorage <EOL> class TestLoadStorage ( TestCase ) : <EOL> def test_load_storage ( self ) : <EOL> self . assertEqual ( <EOL> type ( get_storage ( '<STR_LIT>' , '<STR_LIT>' ) ) , <EOL> BaseStorage ) <EOL> def test_missing_storage ( self ) : <EOL> self . assertRaises ( MissingStorage , get_storage , <EOL> '<STR_LIT>' , '<STR_LIT>' ) <EOL> self . assertRaises ( MissingStorage , get_storage , <EOL> '<STR_LIT>' , '<STR_LIT>' ) </s>
