xtensor-stack
diff --git a/‎include/xtensor/xbuilder.hpp‎
Lines changed: 1 addition & 1 deletion b/‎include/xtensor/xbuilder.hpp‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎test/test_xbuilder.cpp‎
Lines changed: 19 additions & 0 deletions b/‎test/test_xbuilder.cpp‎
Lines changed: 19 additions & 0 deletions
@@ -402,7 +402,7 @@ namespace xt
     inline auto linspace(T start, T stop, std::size_t num_samples = 50, bool endpoint = true) noexcept
     {
         using fp_type = std::common_type_t<T, double>;
-        fp_type step = fp_type(stop - start) / fp_type(num_samples - (endpoint ? 1 : 0));
+        fp_type step = fp_type(stop - start) / std::fmax(fp_type(1), fp_type(num_samples - (endpoint ? 1 : 0)));
         return cast<T>(detail::make_xgenerator(detail::arange_generator<fp_type>(fp_type(start), fp_type(stop), step), {num_samples}));
     }
 
 
@@ -187,6 +187,25 @@ namespace xt
         ASSERT_EQ(m_assigned(3), at_3);
     }
 
+    TEST(xbuilder, linspace_1_point)
+    {
+        xt::xarray<double> a = linspace<double>(0., 0., 1, false);
+        decltype(a)::shape_type expected_shape = {1};
+        EXPECT_EQ(a.dimension(), size_t(1));
+        EXPECT_EQ(a.shape(), expected_shape);
+        EXPECT_EQ(0., a(0));
+
+        xt::xarray<double> b = linspace<double>(0., 0., 1, true);
+        EXPECT_EQ(b.dimension(), size_t(1));
+        EXPECT_EQ(b.shape(), expected_shape);
+        EXPECT_EQ(0., b(0));
+
+        xt::xarray<double> c = linspace<double>(0., 2., 1, true);
+        EXPECT_EQ(c.dimension(), size_t(1));
+        EXPECT_EQ(c.shape(), expected_shape);
+        EXPECT_EQ(0., b(0));
+    }
+
     TEST(xbuilder, linspace_n_samples_endpoint)
     {
         auto ls = linspace<float>(20.f, 50.f, 100, false);
Original file line number	Diff line number	Diff line change
`@@ -402,7 +402,7 @@ namespace xt`
`402`	`402`	`inline auto linspace(T start, T stop, std::size_t num_samples = 50, bool endpoint = true) noexcept`
`403`	`403`	`{`
`404`	`404`	`using fp_type = std::common_type_t<T, double>;`
`405`		`- fp_type step = fp_type(stop - start) / fp_type(num_samples - (endpoint ? 1 : 0));`
	`405`	`+ fp_type step = fp_type(stop - start) / std::fmax(fp_type(1), fp_type(num_samples - (endpoint ? 1 : 0)));`
`406`	`406`	`return cast<T>(detail::make_xgenerator(detail::arange_generator<fp_type>(fp_type(start), fp_type(stop), step), {num_samples}));`
`407`	`407`	`}`
`408`	`408`