public void TestDTypes() { var requestsSchema = new StructType(new List<StructField> { new StructField("test", new StringType(), false), }); var x = requestsSchema.JsonValue.ToString(); Mock<IStructTypeProxy> mockStructTypeProxy = new Mock<IStructTypeProxy>(); mockStructTypeProxy.Setup(m => m.ToJson()).Returns(x); Mock<IStructFieldProxy> mockStructFieldProxy = new Mock<IStructFieldProxy>(); mockStructFieldProxy.Setup(m => m.GetStructFieldName()).Returns("testcol"); Mock<IStructDataTypeProxy> mockStructDataTypeProxy = new Mock<IStructDataTypeProxy>(); mockStructDataTypeProxy.Setup(m => m.GetDataTypeSimpleString()).Returns("ss"); mockStructFieldProxy.Setup(m => m.GetStructFieldDataType()).Returns(mockStructDataTypeProxy.Object); mockStructTypeProxy.Setup(m => m.GetStructTypeFields()) .Returns(new List<IStructFieldProxy>() { mockStructFieldProxy.Object }); Mock<IDataFrameProxy> mockDataFrameProxy = new Mock<IDataFrameProxy>(); mockDataFrameProxy.Setup(m => m.GetSchema()).Returns(mockStructTypeProxy.Object); mockDatasetProxy.Setup(m => m.ToDF()).Returns(mockDataFrameProxy.Object); var dataset = new Dataset(mockDatasetProxy.Object); var dTypes = dataset.DTypes(); Assert.AreEqual(1, dTypes.Count()); var first = dTypes.First(); Assert.AreEqual("testcol", first.Item1); Assert.AreEqual("ss", first.Item2); }
public void TestExplain() { Mock<IDataFrameProxy> mockDataFrameProxy = new Mock<IDataFrameProxy>(); mockDataFrameProxy.Setup(m => m.GetQueryExecution()).Returns("Execution Plan"); mockDataFrameProxy.Setup(m => m.GetExecutedPlan()).Returns("Execution Plan"); mockDatasetProxy.Setup(m => m.ToDF()).Returns(mockDataFrameProxy.Object); var dataset = new Dataset(mockDatasetProxy.Object); dataset.Explain(); mockDataFrameProxy.Verify(m => m.GetQueryExecution(), Times.Once); dataset.Explain(true); mockDataFrameProxy.Verify(m => m.GetExecutedPlan(), Times.Once); }
public void TestShow() { Mock<IDataFrameProxy> mockDataFrameProxy = new Mock<IDataFrameProxy>(); mockDataFrameProxy.Setup(m => m.GetShowString(It.IsAny<int>(), It.IsAny<bool>())).Returns("Show"); mockDatasetProxy.Setup(m => m.ToDF()).Returns(mockDataFrameProxy.Object); var dataset = new Dataset(mockDatasetProxy.Object); dataset.Show(); mockDataFrameProxy.Verify(m => m.GetShowString(20, true), Times.Once); }
public void TestSchema(bool usePrintSchema) { var requestsSchema = new StructType(new List<StructField> { new StructField("test", new StringType(), false), }); var jsonValue = requestsSchema.JsonValue.ToString(); Mock<IStructTypeProxy> mockStructTypeProxy = new Mock<IStructTypeProxy>(); mockStructTypeProxy.Setup(m => m.ToJson()).Returns(jsonValue); Mock<IDataFrameProxy> mockDataFrameProxy = new Mock<IDataFrameProxy>(); mockDataFrameProxy.Setup(m => m.GetSchema()).Returns(mockStructTypeProxy.Object); mockDatasetProxy.Setup(m => m.ToDF()).Returns(mockDataFrameProxy.Object); var dataset = new Dataset(mockDatasetProxy.Object); if (usePrintSchema) dataset.PrintSchema(); else dataset.ShowSchema(); mockDataFrameProxy.Verify(m => m.GetSchema(), Times.Once); mockStructTypeProxy.Verify(m => m.ToJson(), Times.Once()); }