add repr (#48)

dask · Apr 2, 2018 · 6dd1a61 · 6dd1a61
1 parent 0695c24
commit 6dd1a61
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/dask_kubernetes/core.py b/dask_kubernetes/core.py
@@ -253,6 +253,10 @@ def namespace(self):
     def name(self):
         return self.pod_template.metadata.generate_name
 
+    def __repr__(self):
+        return 'KubeCluster("%s", workers=%d)' % (self.scheduler.address,
+                                                  len(self.pods()))
+
     @property
     def scheduler(self):
         return self.cluster.scheduler

diff --git a/dask_kubernetes/tests/test_core.py b/dask_kubernetes/tests/test_core.py
@@ -335,3 +335,10 @@ def test_automatic_startup(image_name, loop, ns):
         with set_config(**{'kubernetes-worker-template-path': fn}):
             with KubeCluster(loop=loop, namespace=ns) as cluster:
                 assert cluster.pod_template.metadata.labels['foo'] == 'bar'
+
+
+def test_repr(cluster):
+    for text in [repr(cluster), str(cluster)]:
+        assert 'Box' not in text
+        assert cluster.scheduler.address in text
+        assert "workers=0" in text