diff options
-rwxr-xr-x | tests/units/client/solver.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/tests/units/client/solver.py b/tests/units/client/solver.py new file mode 100755 index 0000000..62f8f28 --- /dev/null +++ b/tests/units/client/solver.py @@ -0,0 +1,25 @@ +#!/usr/bin/env python +# sugar-lint: disable + +import os + +from __init__ import tests + +from sugar_network.client import solver + + +class SolverTest(tests.Test): + + def test_select_architecture(self): + host_arch = os.uname()[-1] + + self.assertEqual(host_arch, solver.select_architecture( + sorted(solver.machine_ranks.keys(), cmp=lambda x, y: cmp(solver.machine_ranks[x], solver.machine_ranks[y])))) + self.assertEqual(host_arch, solver.select_architecture( + sorted(solver.machine_ranks.keys(), cmp=lambda x, y: cmp(solver.machine_ranks[y], solver.machine_ranks[x])))) + self.assertEqual(host_arch, solver.select_architecture([host_arch])) + self.assertEqual(host_arch, solver.select_architecture(['foo', host_arch, 'bar'])) + + +if __name__ == '__main__': + tests.main() |