diff --git a/tests/test_azure_generator.py b/tests/test_azure_generator.py index 7db0f70f..e4817b77 100644 --- a/tests/test_azure_generator.py +++ b/tests/test_azure_generator.py @@ -432,8 +432,10 @@ def test_update_data_with_attributes(self): if generator.azure_columns == AZURE_COLUMNS_V2_SUBSCRIPTION: self.assertEqual(row["SubscriptionId"], self.payer_account) self.assertEqual(row["ResourceId"], self.instance_id) + row["BillingCurrency"] = self.currency else: self.assertEqual(row["SubscriptionGuid"], self.payer_account) + row["BillingCurrencyCode"] = self.currency class TestDTGenerator(AzureGeneratorTestCase):