From 15d2c860d3357ba4f81722e751e0b9528f3a5818 Mon Sep 17 00:00:00 2001
From: Dave Jones <dave@waveform.org.uk>
Date: Fri, 21 Oct 2016 22:30:40 +0100
Subject: [PATCH] Use with in a few more tests

---
 tests/test_devices.py | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/tests/test_devices.py b/tests/test_devices.py
index 01b714a..858b388 100644
--- a/tests/test_devices.py
+++ b/tests/test_devices.py
@@ -49,28 +49,28 @@ def test_device_init_twice_different_pin():
 
 def test_device_close():
     device = GPIODevice(2)
+    # Don't use "with" here; we're testing close explicitly
     device.close()
     assert device.closed
     assert device.pin is None
 
 def test_device_reopen_same_pin():
     pin = Device._pin_factory.pin(2)
-    device = GPIODevice(pin)
-    device.close()
-    device2 = GPIODevice(pin)
-    assert not device2.closed
-    assert device2.pin is pin
-    assert device.closed
-    assert device.pin is None
-    device2.close()
+    with GPIODevice(pin) as device:
+        pass
+    with GPIODevice(pin) as device2:
+        assert not device2.closed
+        assert device2.pin is pin
+        assert device.closed
+        assert device.pin is None
 
 def test_device_repr():
     with GPIODevice(2) as device:
         assert repr(device) == '<gpiozero.GPIODevice object on pin %s, is_active=False>' % device.pin
 
 def test_device_repr_after_close():
-    device = GPIODevice(2)
-    device.close()
+    with GPIODevice(2) as device:
+        pass
     assert repr(device) == '<gpiozero.GPIODevice object closed>'
 
 def test_device_unknown_attr():