Updated NURBS curve export tests for knots and varying widths

Autodesk · May 4, 2022 · b8e70a0 · b8e70a0
1 parent 8cf3d43
commit b8e70a0
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/test/lib/usd/translators/testUsdExportNurbsCurve.py b/test/lib/usd/translators/testUsdExportNurbsCurve.py
@@ -55,10 +55,39 @@ def testExport(self):
         nc = UsdGeom.NurbsCurves.Get(stage, '/curve1')
         self.assertEqual(nc.GetWidthsAttr().Get(), Vt.FloatArray([1.0]))
         self.assertEqual(nc.GetWidthsInterpolation(), UsdGeom.Tokens.constant)
+        self.assertEqual(nc.GetKnotsAttr().Get(), Vt.DoubleArray([0, 0, 0, 1, 2, 3, 4, 5, 5, 5]))
         self.assertEqual(nc.GetRangesAttr().Get(), Vt.Vec2dArray([Gf.Vec2d(0, 5)]))
         self.assertEqual(nc.GetOrderAttr().Get(), Vt.IntArray([4]))
         self.assertEqual(nc.GetCurveVertexCountsAttr().Get(), Vt.IntArray([8]))
 
+    def testExportWidths(self):
+        '''
+        Test export with manually authored widths on the curve.
+        '''
+        expectedWidths = [1., 2., 3., 4., 5., 6., 7., 8.]
+        cmds.select("curveShape1")
+        cmds.addAttr(longName="widths", dt="floatArray")
+        cmds.setAttr("curveShape1.widths", expectedWidths ,type="floatArray")
+        self.assertEqual(cmds.getAttr("curveShape1.widths"), expectedWidths)
+
+        usdFile = os.path.abspath('UsdExportNurbsCurveTest_Widths.usda')
+        cmds.usdExport(mergeTransformAndShape=True, file=usdFile,
+            shadingMode='none')
+
+        stage = Usd.Stage.Open(usdFile)
+
+        nc = UsdGeom.NurbsCurves.Get(stage, '/curve1')
+        self.assertEqual(nc.GetWidthsAttr().Get(), Vt.FloatArray(expectedWidths))
+        self.assertEqual(nc.GetWidthsInterpolation(), UsdGeom.Tokens.vertex)
+        self.assertEqual(nc.GetKnotsAttr().Get(), Vt.DoubleArray([0, 0, 0, 1, 2, 3, 4, 5, 5, 5]))
+        self.assertEqual(nc.GetRangesAttr().Get(), Vt.Vec2dArray([Gf.Vec2d(0, 5)]))
+        self.assertEqual(nc.GetOrderAttr().Get(), Vt.IntArray([4]))
+        self.assertEqual(nc.GetCurveVertexCountsAttr().Get(), Vt.IntArray([8]))
+
+        with self.assertRaises(ValueError):
+            cmds.deleteAttr("widths")
+            cmds.getAttr("curveShape1.widths")
+
     def testNotExportViaPython(self):
         '''
         Test that filtering curves works when invoked via Python: