fixing distinct test

Author: dantownsend

aragorm/query.py

@@ -83,7 +83,7 @@ async def run(self, as_dict=True, credentials=None):
         # Might try and merge them.
         raw = self.response_handler(raw)
 
-        output = getattr(self, '_output')
+        output = getattr(self, '_output', None)
         if output and type(raw) is list:
             if output.as_list:
                 if len(raw[0].keys()) != 1:

run-tests.sh

@@ -1,2 +1,6 @@
 #!/bin/bash
+# To run all in a folder tests/
+# To run all in a file tests/test_foo.py
+# To run all in a class tests/test_foo.py::TestFoo
+# To run a single test tests/test_foo.py::TestFoo::test_foo
 python -m pytest -s $1

tests/table/test_select.py

@@ -189,6 +189,8 @@ def test_limit(self):
 
         response = Pokemon.select(
             'name'
+        ).order_by(
+            'name'
         ).limit(
             1
         ).run_sync()
@@ -203,10 +205,10 @@ def test_limit(self):
     def test_first(self):
         self.insert_rows()
 
-        import ipdb; ipdb.set_trace()
-
         response = Pokemon.select(
             'name'
+        ).order_by(
+            'name'
         ).first().run_sync()
 
         print(f'response = {response}')
@@ -266,6 +268,8 @@ def test_distinct(self):
         self.insert_rows()
         self.insert_rows()
 
-        import ipdb; ipdb.set_trace()
+        response = Pokemon.select('name').where(
+            Pokemon.name == 'pikachu'
+        ).distinct().run_sync()
 
-        response = Pokemon.select('name').distinct().run_sync()
+        self.assertTrue(response == [{'name': 'pikachu'}])