fixed df in test_column_names

am-tropin · Nov 17, 2023 · 33df2e0 · 33df2e0
1 parent 22100d6
commit 33df2e0
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 11 deletions.
diff --git a/app/tests/.ipynb_checkpoints/test_function_store-checkpoint.ipynb b/app/tests/.ipynb_checkpoints/test_function_store-checkpoint.ipynb
@@ -76,7 +76,7 @@
     "# )\n",
     "\n",
     "@pytest.mark.mandatory\n",
-    "def test_column_names(df):\n",
+    "def test_column_names(grab_test_file):\n",
     "    required_names = {'floor', 'rooms', 'city', 'district', 'radius'}\n",
     "    dt.validate(grab_test_file.columns, required_names)\n",
     "# https://github.com/shawnbrown/datatest/blob/bf136eab23c2b6ea36c201e1446fca9243c3fba6/docs/intro/automated-testing.rst#L139\n"
@@ -188,7 +188,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f9990793",
+   "id": "59add23f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -202,23 +202,23 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "13c848b0",
+   "id": "d156ca82",
    "metadata": {},
    "outputs": [],
    "source": []
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "5ffd885b",
+   "id": "c2cb977c",
    "metadata": {},
    "outputs": [],
    "source": []
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f55a4b8f",
+   "id": "55ee66be",
    "metadata": {},
    "outputs": [],
    "source": []

diff --git a/app/tests/test_function_store.ipynb b/app/tests/test_function_store.ipynb
@@ -76,7 +76,7 @@
     "# )\n",
     "\n",
     "@pytest.mark.mandatory\n",
-    "def test_column_names(df):\n",
+    "def test_column_names(grab_test_file):\n",
     "    required_names = {'floor', 'rooms', 'city', 'district', 'radius'}\n",
     "    dt.validate(grab_test_file.columns, required_names)\n",
     "# https://github.com/shawnbrown/datatest/blob/bf136eab23c2b6ea36c201e1446fca9243c3fba6/docs/intro/automated-testing.rst#L139\n"
@@ -188,7 +188,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f9990793",
+   "id": "59add23f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -202,23 +202,23 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "13c848b0",
+   "id": "d156ca82",
    "metadata": {},
    "outputs": [],
    "source": []
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "5ffd885b",
+   "id": "c2cb977c",
    "metadata": {},
    "outputs": [],
    "source": []
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f55a4b8f",
+   "id": "55ee66be",
    "metadata": {},
    "outputs": [],
    "source": []

diff --git a/app/tests/test_function_store.py b/app/tests/test_function_store.py
@@ -61,7 +61,7 @@ def grab_test_file():
 # )
 
 @pytest.mark.mandatory
-def test_column_names(df):
+def test_column_names(grab_test_file):
     required_names = {'floor', 'rooms', 'city', 'district', 'radius'}
     dt.validate(grab_test_file.columns, required_names)
 # https://github.com/shawnbrown/datatest/blob/bf136eab23c2b6ea36c201e1446fca9243c3fba6/docs/intro/automated-testing.rst#L139