using System;
using System.Collections.Generic;

using NUnit.Framework;

using Python.Runtime;
using Python.Runtime.Codecs;

namespace Python.EmbeddingTest
{
 public class NumPyTests
 {
 [OneTimeSetUp]
 public void SetUp()
 {
 PythonEngine.Initialize();
 TupleCodec<ValueTuple>.Register();
 }

 [OneTimeTearDown]
 public void Dispose()
 {
 PythonEngine.Shutdown();
 }

 [Test]
 public void TestReadme()
 {
 Assert.AreEqual("1.0", np.cos(np.pi * 2).ToString());

 dynamic sin = np.sin;
 StringAssert.StartsWith("-0.95892", sin(5).ToString());

 double c = (double)(np.cos(5) + sin(5));
 Assert.AreEqual(-0.675262, c, 0.01);

 dynamic a = np.array(new List<float> { 1, 2, 3 });
 Assert.AreEqual("float64", a.dtype.ToString());

 dynamic b = np.array(new List<float> { 6, 5, 4 }, Py.kw("dtype", np.int32));
 Assert.AreEqual("int32", b.dtype.ToString());

 Assert.AreEqual("[ 6. 10. 12.]", (a * b).ToString().Replace(" ", " "));
 }

 [Test]
 public void MultidimensionalNumPyArray()
 {
 var array = new[,] { { 1, 2 }, { 3, 4 } };
 var ndarray = np.InvokeMethod("asarray", array.ToPython());
 Assert.AreEqual((2, 2), ndarray.GetAttr("shape").As<(int, int)>());
 Assert.AreEqual(1, ndarray[(0, 0).ToPython()].InvokeMethod("__int__").As<int>());
 Assert.AreEqual(array[1, 0], ndarray[(1, 0).ToPython()].InvokeMethod("__int__").As<int>());
 }

 [Test]
 public void Int64Array()
 {
 var array = new long[,] { { 1, 2 }, { 3, 4 } };
 var ndarray = np.InvokeMethod("asarray", array.ToPython());
 Assert.AreEqual((2, 2), ndarray.GetAttr("shape").As<(int, int)>());
 Assert.AreEqual(1, ndarray[(0, 0).ToPython()].InvokeMethod("__int__").As<long>());
 Assert.AreEqual(array[1, 0], ndarray[(1, 0).ToPython()].InvokeMethod("__int__").As<long>());
 }

 [Test]
 public void VarArg()
 {
 dynamic zX = np.array(new[,] { { 1, 2, 3 }, { 4, 5, 6 }, { 8, 9, 0 } });
 dynamic grad = np.gradient(zX, 4.0, 5.0);
 dynamic grad2 = np.InvokeMethod("gradient", new PyObject[] {zX, new PyFloat(4.0), new PyFloat(5.0)});

 Assert.AreEqual(4.125, grad[0].sum().__float__().As<double>(), 0.001);
 Assert.AreEqual(-1.2, grad[1].sum().__float__().As<double>(), 0.001);
 Assert.AreEqual(4.125, grad2[0].sum().__float__().As<double>(), 0.001);
 Assert.AreEqual(-1.2, grad2[1].sum().__float__().As<double>(), 0.001);
 }

#pragma warning disable IDE1006
 dynamic np
 {
 get
 {
 try
 {
 return Py.Import("numpy");
 }
 catch (PythonException)
 {
 Assert.Inconclusive("Numpy or dependency not installed");
 return null;
 }
 }
 }

 }
}