frank/new ci

.github/workflows/main.yml

@@ -43,13 +43,18 @@ jobs:
           virtualenvs-in-project: true
           installer-parallel: true
       - name: Install dependencies
-        run: poetry install
+        run: poetry install 
+      - name: Install pytest
         run: poetry run pip install pytest
       - name: Run tests
-        run: bash scripts/ci.sh
+        env:
+          MAINNET_RPC_ENDPOINT: ${{ secrets.MAINNET_RPC_ENDPOINT }}
+          DEVNET_RPC_ENDPOINT: ${{ secrets.DEVNET_RPC_ENDPOINT }}
+        run: poetry run bash scripts/ci.sh
 
   bump-version:
     runs-on: ubicloud
+    needs: [black, tests]
     if: github.event_name == 'push' && github.ref == 'refs/heads/master'
     steps:
       - uses: actions/checkout@v3
@@ -75,7 +80,7 @@ jobs:
 
   release:
     runs-on: ubicloud
-    needs: [black, tests, bump-version]
+    needs: [bump-version]
     if: github.event_name == 'push' && github.ref == 'refs/heads/master'
     steps:
     - name: Checkout

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
 -   repo: https://github.com/psf/black
-    rev: 23.11.0
+    rev: 24.4.2
     hooks:
     - id: black
       language_version: python3.10

README.md

@@ -56,19 +56,6 @@ If you intend to use `AccountSubscriptionConfig("demo)`, you *must* call `get_ma
 
 `bash setup.sh`
 
-
-## Building the docs
-
-Local Docs: `mkdocs serve`
-
-Updating public docs: `poetry run mkdocs gh-deploy --force`
-
-## Releasing a new version of the package
-
-- `python new_release.py`
-- Create a new release at https://github.com/drift-labs/driftpy/releases.
-  - (The CI process will upload a new version of the package to [PyPI](https://pypi.org/project/driftpy/))
-
 # Development
 
 Ensure correct python version (using pyenv is recommended):
@@ -85,10 +72,7 @@ poetry install
 
 Run tests:
 ```
-poetry run bash test.sh
+poetry run bash test-scripts/integration_test.sh
+poetry run bash test-scripts/math_tests.sh
 ```
 
-Run Acceptance Tests
-```
-poetry run bash acceptance_test.sh
-```

examples/fetch_all_markets.py

@@ -4,27 +4,42 @@
 from driftpy.drift_client import DriftClient, AccountSubscriptionConfig
 import asyncio
 
+
 async def get_all_market_names():
-    env = 'mainnet-beta' # 'devnet'
-    rpc = 'https://api.mainnet-beta.solana.com' # todo replace
-    kp = Keypair() # random wallet
+    env = "mainnet-beta"  # 'devnet'
+    rpc = "https://api.mainnet-beta.solana.com"  # todo replace
+    kp = Keypair()  # random wallet
     wallet = Wallet(kp)
     connection = AsyncClient(rpc)
     provider = Provider(connection, wallet)
-    drift_client = DriftClient(provider.connection, provider.wallet, env.split('-')[0], account_subscription=AccountSubscriptionConfig("cached"))
-
-    all_perps_markets = await drift_client.program.account['PerpMarket'].all()
-    sorted_all_perps_markets = sorted(all_perps_markets, key=lambda x: x.account.market_index)
-    result_perp = [bytes(x.account.name).decode('utf-8').strip() for x in sorted_all_perps_markets]
+    drift_client = DriftClient(
+        provider.connection,
+        provider.wallet,
+        env.split("-")[0],
+        account_subscription=AccountSubscriptionConfig("cached"),
+    )
+
+    all_perps_markets = await drift_client.program.account["PerpMarket"].all()
+    sorted_all_perps_markets = sorted(
+        all_perps_markets, key=lambda x: x.account.market_index
+    )
+    result_perp = [
+        bytes(x.account.name).decode("utf-8").strip() for x in sorted_all_perps_markets
+    ]
 
-    all_spot_markets = await drift_client.program.account['SpotMarket'].all()
-    sorted_all_spot_markets = sorted(all_spot_markets, key=lambda x: x.account.market_index)
-    result_spot = [bytes(x.account.name).decode('utf-8').strip() for x in sorted_all_spot_markets]
+    all_spot_markets = await drift_client.program.account["SpotMarket"].all()
+    sorted_all_spot_markets = sorted(
+        all_spot_markets, key=lambda x: x.account.market_index
+    )
+    result_spot = [
+        bytes(x.account.name).decode("utf-8").strip() for x in sorted_all_spot_markets
+    ]
 
-    result = result_perp + result_spot[1:] # ignore quote spot market index
+    result = result_perp + result_spot[1:]  # ignore quote spot market index
     return result
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
     loop = asyncio.new_event_loop()
     answer = loop.run_until_complete(get_all_market_names())
     print(answer)