def test_restore_assets(): assets = Assets() p = Specification() calc1 = Calculator(p, assets) calc1.store_assets() calc1.restore_assets() assert isinstance(calc1, Calculator)
def test_store_assets(): assets = Assets() p = Specification() dp = DepreciationParams() calc1 = Calculator(p, dp, assets) calc1.store_assets() assert isinstance(calc1, Calculator)