pypsbuilder 2.5.1__py3-none-any.whl

pypsbuilder/tests/test_pclasses.py

@@ -0,0 +1,279 @@
+import pytest
+from pypsbuilder import InvPoint, UniLine, PTsection
+from pypsbuilder.tcapi import TC35API
+
+pytest.ps = PTsection(trange=(400.0, 700.0), prange=(7.0, 16.0))
+
+
+@pytest.fixture
+def mock_tc():
+    return TC35API("./examples/outputs", None, None)
+
+
+def test_parse_ini1(mock_tc):
+    test = "inv1"
+    ofile = mock_tc.workdir / "{}-log.txt".format(test)
+    icfile = mock_tc.workdir / "{}-ic.txt".format(test)
+
+    with ofile.open("r", encoding=mock_tc.TCenc) as f:
+        output = f.read()
+
+    with icfile.open("r", encoding=mock_tc.TCenc) as f:
+        resic = f.read()
+
+    status, res, output = mock_tc.parse_logfile(output=output, resic=resic)
+
+    inv = InvPoint(
+        phases={"bi", "mu", "chl", "H2O", "ep", "q", "g", "sph", "pa"},
+        out={"ep", "chl"},
+        variance=res.variance,
+        x=res.x,
+        y=res.y,
+        results=res,
+        output=output,
+    )
+    pytest.ps.add_inv(1, inv)
+
+    assert status == "ok", "Wrong status"
+    assert res.variance == 4, "Wrong variance"
+    assert res[0].p == 12.2438, "Wrong pressure"
+    assert res[0].T == 530.136, "Wrong temperature"
+    assert len(res) == 1, "Wrong results length"
+    assert type(res[0].data) == dict, "Wrong data type data"
+    assert type(res[0].ptguess) == list, "Wrong data type of ptguess"
+
+
+def test_parse_ini2(mock_tc):
+    test = "inv2"
+    ofile = mock_tc.workdir / "{}-log.txt".format(test)
+    icfile = mock_tc.workdir / "{}-ic.txt".format(test)
+
+    with ofile.open("r", encoding=mock_tc.TCenc) as f:
+        output = f.read()
+
+    with icfile.open("r", encoding=mock_tc.TCenc) as f:
+        resic = f.read()
+
+    status, res, output = mock_tc.parse_logfile(output=output, resic=resic)
+
+    inv = InvPoint(
+        phases={"ep", "pa", "sph", "q", "H2O", "mu", "chl", "g", "ab", "bi"},
+        out={"ab", "chl"},
+        variance=res.variance,
+        x=res.x,
+        y=res.y,
+        results=res,
+        output=output,
+    )
+    pytest.ps.add_inv(2, inv)
+
+    assert status == "ok", "Wrong status"
+    assert res.variance == 3, "Wrong variance"
+    assert res[0].p == 7.0359, "Wrong pressure"
+    assert res[0].T == 504.062, "Wrong temperature"
+    assert len(res) == 1, "Wrong results length"
+    assert type(res[0].data) == dict, "Wrong data type data"
+    assert type(res[0].ptguess) == list, "Wrong data type of ptguess"
+
+
+def test_parse_ini3(mock_tc):
+    test = "inv3"
+    ofile = mock_tc.workdir / "{}-log.txt".format(test)
+    icfile = mock_tc.workdir / "{}-ic.txt".format(test)
+
+    with ofile.open("r", encoding=mock_tc.TCenc) as f:
+        output = f.read()
+
+    with icfile.open("r", encoding=mock_tc.TCenc) as f:
+        resic = f.read()
+
+    status, res, output = mock_tc.parse_logfile(output=output, resic=resic)
+
+    inv = InvPoint(
+        phases={"pa", "H2O", "sph", "g", "mu", "bi", "q", "ep", "ab"},
+        out={"ab", "ep"},
+        variance=res.variance,
+        x=res.x,
+        y=res.y,
+        results=res,
+        output=output,
+    )
+    pytest.ps.add_inv(3, inv)
+
+    assert status == "ok", "Wrong status"
+    assert res.variance == 4, "Wrong variance"
+    assert res[0].p == 11.908, "Wrong pressure"
+    assert res[0].T == 561.425, "Wrong temperature"
+    assert len(res) == 1, "Wrong results length"
+    assert type(res[0].data) == dict, "Wrong data type data"
+    assert type(res[0].ptguess) == list, "Wrong data type of ptguess"
+
+
+def test_parse_uni1(mock_tc):
+    test = "uni1"
+    ofile = mock_tc.workdir / "{}-log.txt".format(test)
+    icfile = mock_tc.workdir / "{}-ic.txt".format(test)
+
+    with ofile.open("r", encoding=mock_tc.TCenc) as f:
+        output = f.read()
+
+    with icfile.open("r", encoding=mock_tc.TCenc) as f:
+        resic = f.read()
+
+    status, res, output = mock_tc.parse_logfile(output=output, resic=resic)
+
+    uni = UniLine(
+        phases={"bi", "mu", "chl", "H2O", "ep", "q", "g", "sph", "pa"},
+        out={"chl"},
+        variance=res.variance,
+        x=res.x,
+        y=res.y,
+        begin=2,
+        end=1,
+        results=res,
+        output=output,
+    )
+    pytest.ps.add_uni(1, uni)
+
+    assert status == "ok", "Wrong status"
+    assert res.variance == 4, "Wrong variance"
+    assert res[15].p == 8.3, "Wrong pressure"
+    assert res[15].T == 516.896, "Wrong temperature"
+    assert len(res) == 33, "Wrong results length"
+    assert type(res[0].data) == dict, "Wrong data type data"
+    assert type(res[0].ptguess) == list, "Wrong data type of ptguess"
+
+
+def test_parse_uni2(mock_tc):
+    test = "uni2"
+    ofile = mock_tc.workdir / "{}-log.txt".format(test)
+    icfile = mock_tc.workdir / "{}-ic.txt".format(test)
+
+    with ofile.open("r", encoding=mock_tc.TCenc) as f:
+        output = f.read()
+
+    with icfile.open("r", encoding=mock_tc.TCenc) as f:
+        resic = f.read()
+
+    status, res, output = mock_tc.parse_logfile(output=output, resic=resic)
+
+    uni = UniLine(
+        phases={"pa", "H2O", "sph", "g", "mu", "bi", "q", "ep"},
+        out={"ep"},
+        variance=res.variance,
+        x=res.x,
+        y=res.y,
+        begin=1,
+        end=3,
+        results=res,
+        output=output,
+    )
+    pytest.ps.add_uni(2, uni)
+
+    assert status == "ok", "Wrong status"
+    assert res.variance == 5, "Wrong variance"
+    assert res[25].p == 12.0698, "Wrong pressure"
+    assert res[25].T == 546.413, "Wrong temperature"
+    assert len(res) == 51, "Wrong results length"
+    assert type(res[0].data) == dict, "Wrong data type data"
+    assert type(res[0].ptguess) == list, "Wrong data type of ptguess"
+
+
+def test_parse_uni3(mock_tc):
+    test = "uni3"
+    ofile = mock_tc.workdir / "{}-log.txt".format(test)
+    icfile = mock_tc.workdir / "{}-ic.txt".format(test)
+
+    with ofile.open("r", encoding=mock_tc.TCenc) as f:
+        output = f.read()
+
+    with icfile.open("r", encoding=mock_tc.TCenc) as f:
+        resic = f.read()
+
+    status, res, output = mock_tc.parse_logfile(output=output, resic=resic)
+
+    uni = UniLine(
+        phases={"pa", "H2O", "sph", "g", "mu", "bi", "q", "ep", "ab"},
+        out={"ab"},
+        variance=res.variance,
+        x=res.x,
+        y=res.y,
+        begin=2,
+        end=3,
+        results=res,
+        output=output,
+    )
+    pytest.ps.add_uni(3, uni)
+
+    assert status == "ok", "Wrong status"
+    assert res.variance == 4, "Wrong variance"
+    assert res[14].p == 8.08, "Wrong pressure"
+    assert res[14].T == 523.539, "Wrong temperature"
+    assert len(res) == 31, "Wrong results length"
+    assert type(res[0].data) == dict, "Wrong data type data"
+    assert type(res[0].ptguess) == list, "Wrong data type of ptguess"
+
+
+def test_contains_inv():
+    for uni in pytest.ps.unilines.values():
+        inv1 = pytest.ps.invpoints[uni.begin]
+        inv2 = pytest.ps.invpoints[uni.end]
+        assert uni.contains_inv(inv1), "Error in UniLine.contains_inv method"
+        assert uni.contains_inv(inv2), "Error in UniLine.contains_inv method"
+
+
+def test_getidinv():
+    for key, inv in pytest.ps.invpoints.items():
+        is_new, id_found = pytest.ps.getidinv(inv)
+        assert is_new is False, "Error chcecking existing invariant point"
+        assert key == id_found, "Error chcecking existing inv id"
+
+
+def test_getiduni():
+    for key, uni in pytest.ps.unilines.items():
+        is_new, id_found = pytest.ps.getiduni(uni)
+        assert is_new is False, "Error chcecking existing univariant line"
+        assert key == id_found, "Error chcecking existing uni id"
+
+
+def test_auto_connect():
+    for uni in pytest.ps.unilines.values():
+        candidates = [
+            inv for inv in pytest.ps.invpoints.values() if uni.contains_inv(inv)
+        ]
+        assert len(candidates) == 2, "Error to detect auto_connect possibility"
+
+
+def test_remaining_uni():
+    for key, inv in pytest.ps.invpoints.items():
+        n = 0
+        for phases, out in inv.all_unilines():
+            tmp_uni = UniLine(phases=phases, out=out)
+            is_new, id_found = pytest.ps.getiduni(tmp_uni)
+            if is_new:
+                n += 1
+        assert n == 2, "Error in detection of remaining univariant lines"
+
+
+def test_trim_uni():
+    uni = pytest.ps.unilines[1]
+    assert uni.used == slice(0, 33), "Wrong used slice before trimming uni 1"
+    pytest.ps.trim_uni(1)
+    assert uni.used == slice(10, 33), "Wrong used slice after trimming uni 1"
+    #
+    uni = pytest.ps.unilines[2]
+    assert uni.used == slice(0, 51), "Wrong used slice before trimming uni 2"
+    pytest.ps.trim_uni(2)
+    assert uni.used == slice(20, 31), "Wrong used slice after trimming uni 2"
+    #
+    uni = pytest.ps.unilines[3]
+    assert uni.used == slice(0, 31), "Wrong used slice before trimming uni 3"
+    pytest.ps.trim_uni(3)
+    assert uni.used == slice(10, 31), "Wrong used slice after trimming uni 3"
+
+
+def test_create_shapes():
+    shapes, shape_edges, log = pytest.ps.create_shapes()
+    akey = frozenset({"pa", "ep", "g", "q", "bi", "mu", "H2O", "sph"})
+    assert len(shapes) == 1, "Wrong number of areas created"
+    assert akey in shapes, "Wrong key for constructed area"

pypsbuilder/ui/addinv.ui

@@ -0,0 +1,113 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
+ <class>AddInv</class>
+ <widget class="QDialog" name="AddInv">
+  <property name="geometry">
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>300</width>
+    <height>104</height>
+   </rect>
+  </property>
+  <property name="windowTitle">
+   <string>Add invariant point</string>
+  </property>
+  <layout class="QVBoxLayout" name="verticalLayout">
+   <item>
+    <layout class="QHBoxLayout" name="horizontalLayout_2">
+     <item>
+      <widget class="QLabel" name="label">
+       <property name="text">
+        <string>Label</string>
+       </property>
+      </widget>
+     </item>
+     <item>
+      <widget class="QLineEdit" name="labelEdit">
+       <property name="readOnly">
+        <bool>true</bool>
+       </property>
+      </widget>
+     </item>
+     <item>
+      <widget class="QCheckBox" name="checkKeep">
+       <property name="text">
+        <string>Keep results</string>
+       </property>
+      </widget>
+     </item>
+    </layout>
+   </item>
+   <item>
+    <layout class="QHBoxLayout" name="horizontalLayout">
+     <item>
+      <widget class="QLabel" name="x_label">
+       <property name="text">
+        <string>X</string>
+       </property>
+      </widget>
+     </item>
+     <item>
+      <widget class="QLineEdit" name="xEdit"/>
+     </item>
+     <item>
+      <widget class="QLabel" name="y_label">
+       <property name="text">
+        <string>Y</string>
+       </property>
+      </widget>
+     </item>
+     <item>
+      <widget class="QLineEdit" name="yEdit"/>
+     </item>
+    </layout>
+   </item>
+   <item>
+    <widget class="QDialogButtonBox" name="buttonBox">
+     <property name="orientation">
+      <enum>Qt::Horizontal</enum>
+     </property>
+     <property name="standardButtons">
+      <set>QDialogButtonBox::Cancel|QDialogButtonBox::Ok</set>
+     </property>
+    </widget>
+   </item>
+  </layout>
+ </widget>
+ <resources/>
+ <connections>
+  <connection>
+   <sender>buttonBox</sender>
+   <signal>accepted()</signal>
+   <receiver>AddInv</receiver>
+   <slot>accept()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>248</x>
+     <y>254</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>157</x>
+     <y>274</y>
+    </hint>
+   </hints>
+  </connection>
+  <connection>
+   <sender>buttonBox</sender>
+   <signal>rejected()</signal>
+   <receiver>AddInv</receiver>
+   <slot>reject()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>316</x>
+     <y>260</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>286</x>
+     <y>274</y>
+    </hint>
+   </hints>
+  </connection>
+ </connections>
+</ui>

pypsbuilder/ui/adduni.ui

@@ -0,0 +1,119 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
+ <class>AddUni</class>
+ <widget class="QDialog" name="AddUni">
+  <property name="geometry">
+   <rect>
+    <x>0</x>
+    <y>0</y>
+    <width>300</width>
+    <height>100</height>
+   </rect>
+  </property>
+  <property name="windowTitle">
+   <string>Add univariant line</string>
+  </property>
+  <layout class="QVBoxLayout" name="verticalLayout">
+   <item>
+    <layout class="QHBoxLayout" name="horizontalLayout_2">
+     <item>
+      <widget class="QLabel" name="label">
+       <property name="text">
+        <string>Label</string>
+       </property>
+      </widget>
+     </item>
+     <item>
+      <widget class="QLineEdit" name="labelEdit">
+       <property name="readOnly">
+        <bool>true</bool>
+       </property>
+      </widget>
+     </item>
+    </layout>
+   </item>
+   <item>
+    <layout class="QHBoxLayout" name="horizontalLayout">
+     <item>
+      <widget class="QLabel" name="label_2">
+       <property name="text">
+        <string>Begin:</string>
+       </property>
+      </widget>
+     </item>
+     <item>
+      <widget class="QComboBox" name="comboBegin"/>
+     </item>
+     <item>
+      <spacer name="horizontalSpacer">
+       <property name="orientation">
+        <enum>Qt::Horizontal</enum>
+       </property>
+       <property name="sizeHint" stdset="0">
+        <size>
+         <width>40</width>
+         <height>20</height>
+        </size>
+       </property>
+      </spacer>
+     </item>
+     <item>
+      <widget class="QLabel" name="label_3">
+       <property name="text">
+        <string>End:</string>
+       </property>
+      </widget>
+     </item>
+     <item>
+      <widget class="QComboBox" name="comboEnd"/>
+     </item>
+    </layout>
+   </item>
+   <item>
+    <widget class="QDialogButtonBox" name="buttonBox">
+     <property name="orientation">
+      <enum>Qt::Horizontal</enum>
+     </property>
+     <property name="standardButtons">
+      <set>QDialogButtonBox::Cancel|QDialogButtonBox::Ok</set>
+     </property>
+    </widget>
+   </item>
+  </layout>
+ </widget>
+ <resources/>
+ <connections>
+  <connection>
+   <sender>buttonBox</sender>
+   <signal>accepted()</signal>
+   <receiver>AddUni</receiver>
+   <slot>accept()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>248</x>
+     <y>254</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>157</x>
+     <y>274</y>
+    </hint>
+   </hints>
+  </connection>
+  <connection>
+   <sender>buttonBox</sender>
+   <signal>rejected()</signal>
+   <receiver>AddUni</receiver>
+   <slot>reject()</slot>
+   <hints>
+    <hint type="sourcelabel">
+     <x>316</x>
+     <y>260</y>
+    </hint>
+    <hint type="destinationlabel">
+     <x>286</x>
+     <y>274</y>
+    </hint>
+   </hints>
+  </connection>
+ </connections>
+</ui>

pypsbuilder/ui/compile_ui.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+
+pyuic5 ptbuilder.ui > ../ui_ptbuilder.py
+pyuic5 txbuilder.ui > ../ui_txbuilder.py
+pyuic5 pxbuilder.ui > ../ui_pxbuilder.py
+pyuic5 adduni.ui > ../ui_adduni.py
+pyuic5 addinv.ui > ../ui_addinv.py
+pyuic5 uniguess.ui > ../ui_uniguess.py
+