<s> from django . test import TestCase <EOL> from django . contrib . formtools . wizard . storage import ( get_storage , <EOL> MissingStorageModule , <EOL> MissingStorageClass ) <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_module ( self ) : <EOL> self . assertRaises ( MissingStorageModule , get_storage , <EOL> '<STR_LIT>' , '<STR_LIT>' ) <EOL> def test_missing_class ( self ) : <EOL> self . assertRaises ( MissingStorageClass , get_storage , <EOL> '<STR_LIT>' , '<STR_LIT>' ) </s>
