Merge pull request #41 from stackql/feature/updates

fixed tests

Author: jeffreyaven

tests/pystackql_tests.py

@@ -207,8 +207,8 @@ def test_12_execute_with_pandas_output(self):
         self.assertTrue(is_valid_dataframe, f"Result is not a valid DataFrame: {result}")
         # Check datatypes of the columns
         expected_dtypes = {
-            'year': 'int64',
-            'num_buckets': 'int64'
+            'instance_type': 'str',
+            'num_instances': 'int64'
         }
         for col, expected_dtype in expected_dtypes.items():
             actual_dtype = result[col].dtype
@@ -286,8 +286,8 @@ def test_22_execute_server_mode_pandas_output(self):
         self.assertTrue(is_valid_dataframe, f"Result is not a valid DataFrame: {result}")
         # Check datatypes of the columns
         expected_dtypes = {
-            'year': 'int64',
-            'num_buckets': 'int64'
+            'instance_type': 'str',
+            'num_instances': 'int64'
         }
         for col, expected_dtype in expected_dtypes.items():
             actual_dtype = result[col].dtype

tests/test_params.py

@@ -42,10 +42,12 @@ def registry_pull_resp_pattern(provider):
 """
 
 aws_query = f"""
-SELECT 
-SPLIT_PART(CreationDate, '-', 1) as year, count(*) as num_buckets FROM aws.s3_api.buckets
-WHERE region = 'us-east-1'
-GROUP BY year
+SELECT
+instance_type,
+count(*) as num_instances
+FROM aws.ec2.instances
+WHERE region = 'ap-southeast-2'
+GROUP BY instance_type
 """
 
 regions = os.environ.get('AWS_REGIONS').split(',')