Adding probabilistic options and improving docs.

test_sucks.py

@@ -1,9 +1,10 @@
 from xml.etree import ElementTree
 
-from nose.tools import assert_equals
+from nose.tools import *
 
 from sucks import *
 
+
 # There are no tests for the XMPP stuff here because a) it's relatively complicated to test given
 # the library's design and its multithreaded nature, and b) I'm manually testing every change anyhow,
 # as it's not clear how the robot really behaves.
@@ -16,7 +17,6 @@ def test_clean_command():
                   b'<ctl td="Clean"><clean speed="standard" type="auto" /></ctl>')  # protocol has attribs in other order
 
 
-
 def test_edge_command():
     # called Edge because that's what the UI uses, even though the protocol is different
     c = Edge(10)
@@ -39,3 +39,30 @@ def test_stop_command():
     assert_equals(ElementTree.tostring(c.to_xml()),
                   b'<ctl td="Clean"><clean speed="standard" type="stop" /></ctl>')
 
+
+def test_frequency_param_type():
+    t = FREQUENCY
+    assert_equals(t.convert('0', None, None), 0)
+    assert_equals(t.convert('1', None, None), 1.0)
+    assert_equals(t.convert('1/2', None, None), 0.5)
+    assert_equals(t.convert('1/7', None, None), 1.0 / 7.0)
+    with assert_raises(click.exceptions.BadParameter):
+        t.convert('bob', None, None)
+    with assert_raises(click.exceptions.BadParameter):
+        t.convert('2', None, None)
+    with assert_raises(click.exceptions.BadParameter):
+        t.convert('7/5', None, None)
+
+def test_should_run():
+    count = 0
+    for _ in range(10000):
+        if should_run(0.1):
+            count += 1
+    assert_almost_equal(count, 1000, delta=200)
+
+    count = 0
+    for _ in range(10000):
+        if should_run(0.9):
+            count += 1
+    assert_almost_equal(count, 9000, delta=200)
+