From 11a9c469044ef44683852f2a21e90ca615d45137 Mon Sep 17 00:00:00 2001
From: Bob Lantz <rlantz@cs.stanford.edu>
Date: Tue, 2 Dec 2014 23:22:56 -0800
Subject: [PATCH] Fix missing imports

---
 mininet/test/test_switchdpidassignment.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mininet/test/test_switchdpidassignment.py b/mininet/test/test_switchdpidassignment.py
index e401c60..61833e9 100644
--- a/mininet/test/test_switchdpidassignment.py
+++ b/mininet/test/test_switchdpidassignment.py
@@ -4,6 +4,7 @@
    Regression tests for switch dpid assignment."""
 
 import unittest
+import sys
 from functools import partial
 import re
 
@@ -13,7 +14,7 @@ from mininet.node import UserSwitch, OVSSwitch, OVSLegacyKernelSwitch, IVSSwitch
 from mininet.topo import Topo
 from mininet.log import setLogLevel
 from mininet.util import quietRun
-
+from mininet.clean import cleanup
 
 class testSwitchDpidAssignmentCommon ( object ):
     """Verify Switch dpid assignment."""