gemini.md

Author: GarrettWu

GEMINI.md

@@ -2,12 +2,48 @@
 
 ## Testing
 
-We use `pytest` to instrument our tests.
+We use `nox` to instrument our tests.
 
-- To test your changes, run unit tests with `pytest`:
+- To test your changes, run unit tests with `nox`:
 
   ```bash
-  pytest <test_file>::<test>
+  nox -r -s unit
+  ```
+
+- To run a single unit test:
+
+  ```bash
+  nox -r -s unit-3.13 -- -k <name of test>
+  ```
+
+- Ignore this step if you lack access to Google Cloud resources. To run system
+  tests, you can execute::
+
+   # Run all system tests
+   $ nox -r -s system
+
+   # Run a single system test
+   $ nox -r -s system-3.13 -- -k <name of test>
+
+- The codebase must have better coverage than it had previously after each
+  change. You can test coverage via `nox -s unit system cover` (takes a long
+  time). Omit `system` if you lack access to cloud resources.
+
+## Code Style
+
+- We use the automatic code formatter `black`. You can run it using
+  the nox session `format`. This will eliminate many lint errors. Run via:
+
+  ```bash
+  nox -r -s format
+  ```
+
+- PEP8 compliance is required, with exceptions defined in the linter configuration.
+  If you have ``nox`` installed, you can test that you have not introduced
+  any non-compliant code via:
+
+  ```
+  nox -r -s lint
   ```
 
 - When writing tests, use the idiomatic "pytest" style.

tests/unit/bigquery/test_ai.py

@@ -270,7 +270,7 @@ def test_generate_table_with_options(mock_dataframe, mock_session):
 
 
 @mock.patch("bigframes.pandas.read_pandas")
-def test_generate_table_with_pandas_dataframe(
+def test_generate_text_with_pandas_dataframe(
     read_pandas_mock, mock_dataframe, mock_session
 ):
     # This tests that pandas input path works and calls read_pandas
@@ -281,10 +281,9 @@ def test_generate_table_with_pandas_dataframe(
 
     pandas_df = pd.DataFrame({"content": ["test"]})
 
-    bbq.ai.generate_table(
+    bbq.ai.generate_text(
         model_name,
         pandas_df,
-        output_schema="col1 STRING",
     )
 
     read_pandas_mock.assert_called_once()