{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "c1c21480",
   "metadata": {},
   "source": [
    "# Test well elements"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6671c6f9-ba3b-45ec-8d9d-0a43b7cf0e73",
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "\n",
    "import timflow.steady as tfs"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "da461528-d94d-42e7-a2d1-6f2bdc95e66b",
   "metadata": {},
   "source": [
    "### Well with specified $Q$ screened in multiple layers"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6f2d1c20-3027-488d-9ba6-6bcb2e298f51",
   "metadata": {},
   "outputs": [],
   "source": [
    "ml = tfs.ModelMaq(\n",
    "    kaq=[10, 20, 5],\n",
    "    z=[21, 20, 12, 10, 2, 0, -10],\n",
    "    c=[100, 100, 200],\n",
    "    hstar=10,\n",
    "    topboundary=\"semi\",\n",
    ")\n",
    "w = tfs.Well(model=ml, xw=20, yw=40, rw=0.3, Qw=100, layers=[0, 1])\n",
    "ml.solve()\n",
    "hinside = w.headinside()\n",
    "assert np.allclose(hinside[1:], hinside[0])\n",
    "print(\"head inside: \", hinside)\n",
    "print(\"discharge: \", w.discharge())"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "e5743104-ebcd-4505-b022-13037ba0a603",
   "metadata": {},
   "source": [
    "### Well with specified $Q$ screened in multiple layers with entry resistance"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "b426762f-6480-4118-9f61-25f8cf3abe2c",
   "metadata": {},
   "outputs": [],
   "source": [
    "ml = tfs.ModelMaq(\n",
    "    kaq=[10, 20, 5],\n",
    "    z=[21, 20, 12, 10, 2, 0, -10],\n",
    "    c=[100, 100, 200],\n",
    "    hstar=10,\n",
    "    topboundary=\"semi\",\n",
    ")\n",
    "w = tfs.Well(model=ml, xw=20, yw=40, rw=0.3, Qw=100, res=0.1, layers=[0, 1])\n",
    "ml.solve()\n",
    "hinside = w.headinside()\n",
    "assert np.allclose(hinside[1:], hinside[0])\n",
    "print(\"head inside: \", hinside)\n",
    "print(\"discharge: \", w.discharge())"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "4c4f6d9a-d5fa-4be2-94ea-c2ba29cbd413",
   "metadata": {},
   "source": [
    "### Well with specified head screened in multiple layers with entry resistance"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "bc5f5125-f0b2-4eed-9fe6-e92ecfd58155",
   "metadata": {},
   "outputs": [],
   "source": [
    "ml = tfs.ModelMaq(\n",
    "    kaq=[10, 20, 5],\n",
    "    z=[21, 20, 12, 10, 5, 0, -10],\n",
    "    c=[100, 100, 200],\n",
    "    hstar=10,\n",
    "    topboundary=\"semi\",\n",
    ")\n",
    "w = tfs.HeadWell(model=ml, xw=20, yw=40, rw=0.3, res=0.1, hw=6, layers=[0, 1])\n",
    "ml.solve()\n",
    "hinside = w.headinside()\n",
    "assert np.allclose(hinside[1:], hinside[0])\n",
    "print(\"head inside: \", hinside)\n",
    "print(\"discharge: \", w.discharge())"
   ]
  }
 ],
 "metadata": {
  "language_info": {
   "name": "python"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}