spec/context_spec.rb in parsable-0.1.0 vs spec/context_spec.rb in parsable-0.1.1

@@ -11,17 +11,17 @@
   end
 
   describe '#custom_store' do
     it "stores custom variables in the custom object_key" do
       context.custom_store :email, "[e-mail]"
-      expect(context.instance_variable_get('@variables')[:custom][:email]).to eql("[e-mail]")
+      expect(context.instance_variable_get('@variables')[:custom].email).to eql("[e-mail]")
     end
   end
 
   describe '#system_store' do
     it "stores variables as a top level object_key" do
       context.system_store :test_object, 'test_attribute', 'test_value'
-      expect(context.instance_variable_get('@variables')[:test_object][:test_attribute]).to eql("test_value")
+      expect(context.instance_variable_get('@variables')[:test_object].test_attribute).to eql("test_value")
     end
   end
 
   describe '#read' do
     it "gets value for object.attribute" do