update UnitTests.hs

2 files changed, 7 insertions, 3 deletions

diff --git a/run-test-suite.sh b/run-test-suite.sh
index 5a7cead1..ded9f41a 100755
--- a/run-test-suite.sh
+++ b/run-test-suite.sh
@@ -1,4 +1,4 @@
-#!/bin/bash -e
+#!/bin/sh -e
 
 if [ "$1" == "--profile" ] ; then
   shift
diff --git a/test/UnitTests.hs b/test/UnitTests.hs
index b6decfff..85485ae9 100644
--- a/test/UnitTests.hs
+++ b/test/UnitTests.hs
@@ -29,10 +29,14 @@ instance Arbitrary SourcePos where
       = [newPos n' l' c' | n' <- shrink n, l' <- shrink l, c' <- shrink c]
   -- TODO: Diagonalize shrink
 
+instance Arbitrary Range where
+  arbitrary = Range <$> arbitrary <*> arbitrary
+  shrink (Range a b) = Range <$> shrink a <*> shrink b
+
 instance Arbitrary SI where
-  arbitrary = oneof [NoSI . Set.fromList <$> arbitrary, Range <$> arbitrary]
+  arbitrary = oneof [NoSI . Set.fromList <$> arbitrary, RangeSI <$> arbitrary]
   shrink (NoSI ds) = []
-  shrink (Range r) = mempty: map Range (shrink r)
+  shrink (RangeSI r) = mempty: map RangeSI (shrink r)
 
 instance Arbitrary Infos where
   arbitrary        = Infos . Map.fromList <$> arbitrary